期刊
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING
卷 31, 期 -, 页码 710-719出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TNSRE.2022.3230250
关键词
Electroencephalography; Convolution; Transformers; Feature extraction; Decoding; Convolutional neural networks; Task analysis; EEG classification; self-attention; transformer; brain-computer interface (BCI); motor imagery
In this paper, a compact Convolutional Transformer called EEG Conformer is proposed to capture both local and global features for EEG classification. The model consists of convolution module, self-attention module, and a simple classifier module, and a visualization strategy is devised for interpretability. Extensive experiments demonstrate that the proposed method achieves state-of-the-art performance on multiple public datasets.
Due to the limited perceptual field, convolutional neural networks (CNN) only extract local temporal features and may fail to capture long-term dependencies for EEG decoding. In this paper, we propose a compact Convolutional Transformer, named EEG Conformer, to encapsulate local and global features in a unified EEG classification framework. Specifically, the convolution module learns the low-level local features throughout the one-dimensional temporal and spatial convolution layers. The self-attention module is straightforwardly connected to extract the global correlation within the local temporal features. Subsequently, the simple classifier module based on fully-connected layers is followed to predict the categories for EEG signals. To enhance interpretability, we also devise a visualization strategy to project the class activation mapping onto the brain topography. Finally, we have conducted extensive experiments to evaluate our method on three public datasets in EEG-based motor imagery and emotion recognition paradigms. The experimental results show that our method achieves state-of-the-art performance and has great potential to be a new baseline for general EEG decoding. The code has been released in https://github.com/eeyhsong/EEG-Conformer.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据