A diagonal masking self-attention-based multi-scale network for motor imagery classification

Kaijun Yang,Jihong Wang,Liantao Yang,Lifeng Bian,Zi Jiang Luo,Chen Yang
DOI: https://doi.org/10.1088/1741-2552/ad5405
IF: 5.043
2024-06-05
Journal of Neural Engineering
Abstract:Electroencephalography (EEG)-based motor imagery (MI) is a promising paradigm for brain-computer interface (BCI), but the non-stationarity and low signal-to-noise ratio of EEG signals make it a challenging task. To achieve high-precision motor imagery classification, we propose a Diagonal Masking Self-Attention-based Multi-Scale Network (DMSA-MSNet) to fully develop, extract, and emphasize features from different scales. First, for local features, a multi-scale temporal-spatial block is proposed to extract features from different receptive fields. Second, an adaptive branch fusion block is specifically designed to bridge the semantic gap between these coded features from different scales. Finally, in order to analyze global information over long ranges, adiagonal masking self-attentionblock is introduced, whichhighlightsthe most valuable features in the data. The proposed DMSA-MSNet outperforms state-of-the-art models on the BCI Competition IV 2a and the BCI Competition IV 2b datasets. Our study achieves rich information extraction from EEG signals and provides an effective solution for motor imagery classification. The code is available at https://github.com/HyperSystemAndImageProc/A-diagonal-masking-self-attention-based-Multi-Scale-Network-for-motor-imagery-classification.
engineering, biomedical,neurosciences
What problem does this paper attempt to address?