MBCNN-EATCFNet: A multi-branch neural network with efficient attention mechanism for decoding EEG-based motor imagery

被引:0
|
作者
Xiong, Shiming [1 ]
Wang, Li [1 ]
Xia, Guoxian [1 ]
Deng, Jiaxian [1 ]
机构
[1] Guangzhou Univ, Sch Elect & Commun Engn, Guangzhou 510006, Peoples R China
关键词
Brain-computer interface (BCI); Electroencephalography (EEG); Motor imagery (MI); Deep learning (DL); CLASSIFICATION;
D O I
10.1016/j.robot.2024.104899
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The decoding performance of motor imagery (MI) based on electroencephalogram (EEG) limits the practical applications of brain-computer interface (BCI). In this paper, we propose a decoding approach for MI classification tasks based on a multi-branch convolutional neural network (MBCNN) and a time convolution fusion network with efficient attention mechanism (EATCFN). In MBCNN-EATCFNet, the combination of multi-branch and multi-scale structures is employed to capture spatiotemporal features at different scales. Additionally, to extract more discriminative temporal features from EEG signals, by integrating ECA into TCN, the module captures integrated information of bidirectional cross-channel interactions and long-term dependencies in the temporal sequence. Finally, to improve the adaptability of the model, a novel adaptive feature fusion method is proposed to distinguish the importance of bidirectional features. Our proposed model achieves classification results of 81.34 % (subject-dependent) and 69.46 % (subject-independent) on the BCI Competition IV dataset 2a, and 87.45 % (subject-dependent) and 83.63 % (subject-independent) on the BCI Competition IV dataset 2b, respectively. On dataset 2a, compared to eight baseline models, our approach achieves an average improvement of 10.15 % (subject-dependent) and 4.34 % (subject-independent), respectively. On dataset 2b, it achieves an average improvement of 2.76 % (subject-dependent) and 1.55 % (subject-independent). Furthermore, ablation experiments have been conducted to validate the effectiveness of each module. This model has significant potential in the clinical and practical application of MI-based BCI systems, thus promoting the further development of BCI technology.
引用
收藏
页数:19
相关论文
共 50 条
  • [1] A Multi-Branch 3D Convolutional Neural Network for EEG-Based Motor Imagery Classification
    Zhao, Xinqiao
    Zhang, Hongmiao
    Zhu, Guilin
    You, Fengxiang
    Kuang, Shaolong
    Sun, Lining
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2019, 27 (10) : 2164 - 2177
  • [2] A Multi-Branch Convolutional Neural Network with Squeeze-and-Excitation Attention Blocks for EEG-Based Motor Imagery Signals Classification
    Altuwaijri, Ghadir Ali
    Muhammad, Ghulam
    Altaheri, Hamdi
    Alsulaiman, Mansour
    DIAGNOSTICS, 2022, 12 (04)
  • [3] A compact multi-branch 1D convolutional neural network for EEG-based motor imagery classification
    Liu, Xiaoguang
    Xiong, Shicheng
    Wang, Xiaodong
    Liang, Tie
    Wang, Hongrui
    Liu, Xiuling
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 81
  • [4] A Multi-Domain Convolutional Neural Network for EEG-Based Motor Imagery Decoding
    Zhi, Hongyi
    Yu, Zhuliang
    Yu, Tianyou
    Gu, Zhenghui
    Yang, Jian
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2023, 31 : 3988 - 3998
  • [5] EEG-FMCNN: A fusion multi-branch 1D convolutional neural network for EEG-based motor imagery classification
    Wang, Wenlong
    Li, Baojiang
    Wang, Haiyan
    Wang, Xichao
    Qin, Yuxin
    Shi, Xingbin
    Liu, Shuxin
    MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2024, 62 (01) : 107 - 120
  • [6] EEG-FMCNN: A fusion multi-branch 1D convolutional neural network for EEG-based motor imagery classification
    Wenlong Wang
    Baojiang Li
    Haiyan Wang
    Xichao Wang
    Yuxin Qin
    Xingbin Shi
    Shuxin Liu
    Medical & Biological Engineering & Computing, 2024, 62 (1) : 107 - 120
  • [7] A robust multi-branch multi-attention-mechanism EEGNet for motor imagery BCI decoding
    Deng, Haodong
    Li, Mengfan
    Li, Jundi
    Guo, Miaomiao
    Xu, Guizhi
    JOURNAL OF NEUROSCIENCE METHODS, 2024, 405
  • [8] One-Dimensional Convolutional Multi-branch Fusion Network for EEG-Based Motor Imagery Classification
    Liu, Xiaoguang
    Zhang, Mingjin
    Xiong, Shicheng
    Wang, Xiaodong
    Liang, Tie
    Li, Jun
    Xiong, Peng
    Wang, Hongrui
    Liu, Xiuling
    IRBM, 2023, 44 (06)
  • [9] A Densely Connected Multi-Branch 3D Convolutional Neural Network for Motor Imagery EEG Decoding
    Liu, Tianjun
    Yang, Deling
    BRAIN SCIENCES, 2021, 11 (02) : 1 - 24
  • [10] Dual-Branch Convolution Network With Efficient Channel Attention for EEG-Based Motor Imagery Classification
    Zhou, Kai
    Haimudula, Aierken
    Tang, Wanying
    IEEE ACCESS, 2024, 12 : 74930 - 74943