ERTNet: an interpretable transformer-based framework for EEG emotion recognition

Bibliographic Details
Title: ERTNet: an interpretable transformer-based framework for EEG emotion recognition
Authors: Ruixiang Liu, Yihu Chao, Xuerui Ma, Xianzheng Sha, Limin Sun, Shuo Li, Shijie Chang
Source: Frontiers in Neuroscience, Vol 18 (2024)
Publisher Information: Frontiers Media S.A., 2024.
Publication Year: 2024
Collection: LCC:Neurosciences. Biological psychiatry. Neuropsychiatry
Subject Terms: EEG, emotion recognition, deep learning, transformer, interpretability, Neurosciences. Biological psychiatry. Neuropsychiatry, RC321-571
More Details: BackgroundEmotion recognition using EEG signals enables clinicians to assess patients’ emotional states with precision and immediacy. However, the complexity of EEG signal data poses challenges for traditional recognition methods. Deep learning techniques effectively capture the nuanced emotional cues within these signals by leveraging extensive data. Nonetheless, most deep learning techniques lack interpretability while maintaining accuracy.MethodsWe developed an interpretable end-to-end EEG emotion recognition framework rooted in the hybrid CNN and transformer architecture. Specifically, temporal convolution isolates salient information from EEG signals while filtering out potential high-frequency noise. Spatial convolution discerns the topological connections between channels. Subsequently, the transformer module processes the feature maps to integrate high-level spatiotemporal features, enabling the identification of the prevailing emotional state.ResultsExperiments’ results demonstrated that our model excels in diverse emotion classification, achieving an accuracy of 74.23% ± 2.59% on the dimensional model (DEAP) and 67.17% ± 1.70% on the discrete model (SEED-V). These results surpass the performances of both CNN and LSTM-based counterparts. Through interpretive analysis, we ascertained that the beta and gamma bands in the EEG signals exert the most significant impact on emotion recognition performance. Notably, our model can independently tailor a Gaussian-like convolution kernel, effectively filtering high-frequency noise from the input EEG data.DiscussionGiven its robust performance and interpretative capabilities, our proposed framework is a promising tool for EEG-driven emotion brain-computer interface.
Document Type: article
File Description: electronic resource
Language: English
ISSN: 1662-453X
Relation: https://www.frontiersin.org/articles/10.3389/fnins.2024.1320645/full; https://doaj.org/toc/1662-453X
DOI: 10.3389/fnins.2024.1320645
Access URL: https://doaj.org/article/354a2f30bd17473d9916feafa046f623
Accession Number: edsdoj.354a2f30bd17473d9916feafa046f623
Database: Directory of Open Access Journals
More Details
ISSN:1662453X
DOI:10.3389/fnins.2024.1320645
Published in:Frontiers in Neuroscience
Language:English