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 条
  • [31] Multiclass classification of motor imagery tasks based on multi-branch convolutional neural network and temporal convolutional network model
    Yu, Shiqi
    Wang, Zedong
    Wang, Fei
    Chen, Kai
    Yao, Dezhong
    Xu, Peng
    Zhang, Yong
    Wang, Hesong
    Zhang, Tao
    CEREBRAL CORTEX, 2024, 34 (02)
  • [32] Recognition of single upper limb motor imagery tasks from EEG using multi-branch fusion convolutional neural network
    Zhang, Rui
    Chen, Yadi
    Xu, Zongxin
    Zhang, Lipeng
    Hu, Yuxia
    Chen, Mingming
    FRONTIERS IN NEUROSCIENCE, 2023, 17
  • [33] Subject adaptation convolutional neural network for EEG-based motor imagery classification
    Liu, Siwei
    Zhang, Jia
    Wang, Andong
    Wu, Hanrui
    Zhao, Qibin
    Long, Jinyi
    JOURNAL OF NEURAL ENGINEERING, 2022, 19 (06)
  • [34] Improving EEG-Based Motor Imagery Classification Using Hybrid Neural Network
    Li, Cong
    Yang, Honghong
    Wu, Xia
    Zhang, Yumei
    2021 IEEE 9TH INTERNATIONAL CONFERENCE ON INFORMATION, COMMUNICATION AND NETWORKS (ICICN 2021), 2021, : 486 - 489
  • [35] EEG-Based Motor Imagery Classification Using Multilayer Perceptron Neural Network
    Ferreira, S. K. S.
    Silveira, A. S.
    Pereira, A.
    XXVII BRAZILIAN CONGRESS ON BIOMEDICAL ENGINEERING, CBEB 2020, 2022, : 1873 - 1878
  • [36] A novel multi-scale fusion convolutional neural network for EEG-based motor imagery classification
    Yang, Guangyu
    Liu, Jinguo
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 96
  • [37] ETCNet: An EEG-based motor imagery classification model combining efficient channel attention and temporal convolutional network
    Qin, Yuxin
    Li, Baojiang
    Wang, Wenlong
    Shi, Xingbin
    Wang, Haiyan
    Wang, Xichao
    BRAIN RESEARCH, 2024, 1823
  • [38] Motor Imagery EEG Classification Based on a Weighted Multi-Branch Structure Suitable for Multisubject Data
    Wang, Huiyang
    Jiang, Jiuchuan
    Gan, John Q.
    Wang, Haixian
    IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2023, 70 (11) : 3040 - 3051
  • [39] Enhancing Cross-Subject Motor Imagery Classification in EEG-Based Brain-Computer Interfaces by Using Multi-Branch CNN
    Chowdhury, Radia Rayan
    Muhammad, Yar
    Adeel, Usman
    SENSORS, 2023, 23 (18)
  • [40] SincNet-Based Hybrid Neural Network for Motor Imagery EEG Decoding
    Liu, Chang
    Jin, Jing
    Daly, Ian
    Li, Shurui
    Sun, Hao
    Huang, Yitao
    Wang, Xingyu
    Cichocki, Andrzej
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2022, 30 : 540 - 549