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 条
  • [21] BrainGridNet: A two-branch depthwise CNN for decoding EEG-based multi-class motor imagery
    Wang, Xingfu
    Wang, Yu
    Qi, Wenxia
    Kong, Delin
    Wang, Wei
    NEURAL NETWORKS, 2024, 170 : 312 - 324
  • [22] Improved Decoding of EEG-Based Motor Imagery Using Convolutional Neural Network and Data Space Adaptation
    Chua, Shawn
    Tao, Yang
    So, Rosa Q.
    2019 IEEE EMBS INTERNATIONAL CONFERENCE ON BIOMEDICAL & HEALTH INFORMATICS (BHI), 2019,
  • [23] Electroencephalogram-Based Motor Imagery Signals Classification Using a Multi-Branch Convolutional Neural Network Model with Attention Blocks
    Altuwaijri, Ghadir Ali
    Muhammad, Ghulam
    BIOENGINEERING-BASEL, 2022, 9 (07):
  • [24] Deep Neural Network with Attention Mechanism for Classification of Motor Imagery EEG
    Huang, Yen-Cheng
    Chang, Jia-Ren
    Chen, Li-Fen
    Chen, Yong-Sheng
    2019 9TH INTERNATIONAL IEEE/EMBS CONFERENCE ON NEURAL ENGINEERING (NER), 2019, : 1130 - 1133
  • [25] MMCNN: A Multi-branch Multi-scale Convolutional Neural Network for Motor Imagery Classification
    Jia, Ziyu
    Lin, Youfang
    Wang, Jing
    Yang, Kaixin
    Liu, Tianhang
    Zhang, Xinwang
    MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2020, PT III, 2021, 12459 : 736 - 751
  • [26] Multi-Scale Convolutional Attention and Riemannian Geometry Network for EEG-Based Motor Imagery Classification
    Zhou, Ben
    Wang, Lei
    Xu, Wenchang
    Jiang, Chenyu
    IEEE ACCESS, 2024, 12 : 79731 - 79740
  • [27] Data augmentation strategies for EEG-based motor imagery decoding
    George, Olawunmi
    Smith, Roger
    Madiraju, Praveen
    Yahyasoltani, Nasim
    Ahamed, Sheikh Iqbal
    HELIYON, 2022, 8 (08)
  • [28] Attention-based convolutional neural network with multi-modal temporal information fusion for motor imagery EEG decoding
    Ma X.
    Chen W.
    Pei Z.
    Zhang Y.
    Chen J.
    Computers in Biology and Medicine, 2024, 175
  • [29] Decoding of Motor Coordination Imagery Involving the Lower Limbs by the EEG-Based Brain Network
    Fu, Yunfa
    Zhou, Zhouzhou
    Gong, Anmin
    Qian, Qian
    Su, Lei
    Zhao, Lei
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2021, 2021
  • [30] EMBANet: A flexible efficient multi-branch attention network
    Zu, Keke
    Zhang, Hu
    Zhang, Lei
    Lu, Jian
    Xu, Chen
    Chen, Hongyang
    Zheng, Yu
    NEURAL NETWORKS, 2025, 185