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

被引:1
作者
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 comprehensive review of EEG-based brain-computer interface paradigms [J].
Abiri, Reza ;
Borhani, Soheil ;
Sellers, Eric W. ;
Jiang, Yang ;
Zhao, Xiaopeng .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (01)
[2]   EEG feature fusion for motor imagery: A new robust framework towards stroke patients rehabilitation [J].
Al-Qazzaz, Noor Kamal ;
Alyasseri, Zaid Abdi Alkareem ;
Abdulkareem, Karrar Hameed ;
Ali, Nabeel Salih ;
Al-Mhiqani, Mohammed Nasser ;
Guger, Christoph .
COMPUTERS IN BIOLOGY AND MEDICINE, 2021, 137
[3]   Deep learning for motor imagery EEG-based classification: A review [J].
Al-Saegh, Ali ;
Dawwd, Shefa A. ;
Abdul-Jabbar, Jassim M. .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 63
[4]   Physics-Informed Attention Temporal Convolutional Network for EEG-Based Motor Imagery Classification [J].
Altaheri, Hamdi ;
Muhammad, Ghulam ;
Alsulaiman, Mansour .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2023, 19 (02) :2249-2258
[5]   Deep learning techniques for classification of electroencephalogram (EEG) motor imagery (MI) signals: a review [J].
Altaheri, Hamdi ;
Muhammad, Ghulam ;
Alsulaiman, Mansour ;
Amin, Syed Umar ;
Altuwaijri, Ghadir Ali ;
Abdul, Wadood ;
Bencherif, Mohamed A. ;
Faisal, Mohammed .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (20) :14681-14722
[6]   Electroencephalogram-Based Motor Imagery Signals Classification Using a Multi-Branch Convolutional Neural Network Model with Attention Blocks [J].
Altuwaijri, Ghadir Ali ;
Muhammad, Ghulam .
BIOENGINEERING-BASEL, 2022, 9 (07)
[7]   A Multi-Branch Convolutional Neural Network with Squeeze-and-Excitation Attention Blocks for EEG-Based Motor Imagery Signals Classification [J].
Altuwaijri, Ghadir Ali ;
Muhammad, Ghulam ;
Altaheri, Hamdi ;
Alsulaiman, Mansour .
DIAGNOSTICS, 2022, 12 (04)
[8]  
Arora S, 2018, PMLR, P1455, DOI DOI 10.48550/ARXIV.1803.01768
[9]   Random Forest and Filter Bank Common Spatial Patterns for EEG-Based Motor Imagery Classification [J].
Bentlemsan, Maouia ;
Zemouri, ET-Tahir ;
Bouchaffra, Djamel ;
Yahya-Zoubir, Bahia ;
Ferroudji, Karim .
PROCEEDINGS FIFTH INTERNATIONAL CONFERENCE ON INTELLIGENT SYSTEMS, MODELLING AND SIMULATION, 2014, :235-238
[10]   TSPNet: a time-spatial parallel network for classification of EEG-based multiclass upper limb motor imagery BCI [J].
Bi, Jingfeng ;
Chu, Ming ;
Wang, Gang ;
Gao, Xiaoshan .
FRONTIERS IN NEUROSCIENCE, 2023, 17