Attention-Based Multi-dimensional Feature Graph Convolutional Network for Motor Imagery Classification

LI Zhenqi,WANG Jing,JIA Ziyu,LIN Youfang
DOI: https://doi.org/10.3778/j.issn.1673-9418.2103079
2022-01-01
Abstract:Motor imagery (MI), an important application of brain-computer interface (BCI), is critical for sports rehabilitation. Since the spatial position of electrodes is not Euclidean, accurate classification of MI is extremely challenging. Existing methods only consider the information of a single dimension or two dimensions in electroence-phalogram (EEG) signals, and cannot fully capture the inherent features of EEG signals in three dimensions of temporal, spectral and spatial. Meanwhile, dynamic correlations of EEG in each dimension affect the robustness of classification. To solve above problems, a novel model, named attention-based multi-dimensional feature graph convolutional network (AMFGCN), is proposed. Firstly, to model non-Euclidean spatial positions of electrodes and fully show spatial correlation between electrodes, a dedicated graph structure is designed. Secondly, a dual-branch framework of time-space and frequency-space dimensions is proposed, simultaneously decoding EEG signals from temporal, spectral and spatial domains. Thirdly, via fusing attention mechanisms into graph convolution and temporal/ spectral convolution, temporal, spectral and spatial correlations are then dynamically captured respectively. Experiments on four public BCI datasets illustrate superior performance of AMFGCN, which outperforms state-of-the-art competitive methods.
What problem does this paper attempt to address?