Research on EEG Recognition Based on Deep Neural Networks with Attention Mechanism

Abstract

Electroencephalogram(EEG) is widely applied in brain-computer interface systems. Due to its high time resolution, portability, and non-invasive characteristics, it has been applied for the rehabilitation of disabled patients. At present, deep learning has also been applied in EEG feature extraction and pattern classification. However, most EEG features in deep neural networks lack discriminative ability, which limits the representation ability of these methods. To overcome this issue, two methods were proposed, named ConvNet_S_A and ConvNet_S_CA, which are based on the backbone network Shallow ConvNet. The proposed methods combine the channel attention mechanism (SE module) and the convolutional block attention module (CBAM module) to learn discriminative EEG features, respectively. Experimental results on the public EEG datasets and self-collected EEG datasets demonstrate the effectiveness.