1. A novel multi-scale fusion convolutional neural network for EEG-based motor imagery classification.
- Author
-
Yang, Guangyu and Liu, Jinguo
- Subjects
CONVOLUTIONAL neural networks ,ARTIFICIAL neural networks ,MOTOR imagery (Cognition) ,BRAIN-computer interfaces ,FILTER banks ,DEEP learning - Abstract
Brain-computer interfaces based on motor imagery have played important roles in motor rehabilitation, brain function regulation, disease monitoring, etc. However, due to the low signal-to-noise ratio and spatial resolution of the EEG, their decoding performance still needs to be further improved. In this paper, we propose a multi-scale fusion convolutional neural network model (MSFCNNet) for four classification tasks involving motor imagery EEG signals. Based on EEGNet, we add an attention module and a two-dimensional dilated convolution layer to construct networks of different scales and carry out network fusion. The attention module utilizes the multi-head self-attention mechanism to highlight the most valuable features. The two-dimensional dilated convolution layer recognizes features effectively and increases the receptive field of the model. In addition, the multi-scale network fusion strategy further improves the decoding performance of the model. Our proposed model is tested on the BCI Competition IV-2a dataset and the High-Gamma dataset. Under subject-dependent and subject-independent models, the classification accuracy of the proposed model on the BCI Competition IV-2a dataset reaches 87.16% and 71.03%, respectively. And in the High-Gamma dataset, the classification accuracy of the proposed model reached 94.43% in the subject-dependent model. Compared with the filter bank common spatial pattern (FBCSP) algorithm and other deep network models, our model achieves a certain improvement in classification accuracy. The experimental results show that MSFCNNet has better decoding performance and stronger robustness. • Propose a multi-scale fusion convolutional neural network for EEG classification. • Apply the multi-scale model feature fusion strategy for model training. • Apply the multi-head attention mechanism to improve decoding performance. • Achieve 87.16% EEG classification accuracy in subject-dependent mode. • Achieve 71.03% EEG classification accuracy in subject-independent mode. [ABSTRACT FROM AUTHOR]
- Published
- 2024
- Full Text
- View/download PDF