EISATC-Fusion: Inception Self-Attention Temporal Convolutional Network Fusion for Motor Imagery EEG Decoding

被引:0
|
作者
Liang, Guangjin [1 ]
Cao, Dianguo [1 ]
Wang, Jinqiang [1 ]
Zhang, Zhongcai [1 ]
Wu, Yuqiang [1 ]
机构
[1] Qufu Normal Univ, Coll Engn, Rizhao 276826, Peoples R China
关键词
Brain-computer interface (BCI); motor imagery (MI); attention collapse; temporal convolution network (TCN); transfer learning;
D O I
10.1109/TNSRE.2024.3382226
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
The motor imagery brain-computer interface (MI-BCI) based on electroencephalography (EEG) is a widely used human-machine interface paradigm. However, due to the non-stationarity and individual differences among subjects in EEG signals, the decoding accuracy is limited, affecting the application of the MI-BCI. In this paper, we propose the EISATC-Fusion model for MI EEG decoding, consisting of inception block, multi-head self-attention (MSA), temporal convolutional network (TCN), and layer fusion. Specifically, we design a DS Inception block to extract multi-scale frequency band information. And design a new cnnCosMSA module based on CNN and cos attention to solve the attention collapse and improve the interpretability of the model. The TCN module is improved by the depthwise separable convolution to reduces the parameters of the model. The layer fusion consists of feature fusion and decision fusion, fully utilizing the features output by the model and enhances the robustness of the model. We improve the two-stage training strategy for model training. Early stopping is used to prevent model overfitting, and the accuracy and loss of the validation set are used as indicators for early stopping. The proposed model achieves within-subject classification accuracies of 84.57% and 87.58% on BCI Competition IV Datasets 2a and 2b, respectively. And the model achieves cross-subject classification accuracies of 67.42% and 71.23% (by transfer learning) when training the model with two sessions and one session of Dataset 2a, respectively. The interpretability of the model is demonstrated through weight visualization method.
引用
收藏
页码:1535 / 1545
页数:11
相关论文
共 50 条
  • [1] Attention-based convolutional neural network with multi-modal temporal information fusion for motor imagery EEG decoding
    Ma, Xinzhi
    Chen, Weihai
    Pei, Zhongcai
    Zhang, Yue
    Chen, Jianer
    [J]. Computers in Biology and Medicine, 2024, 175
  • [2] A multiscale feature fusion network based on attention mechanism for motor imagery EEG decoding
    Gao, Dongrui
    Yang, Wen
    Li, Pengrui
    Liu, Shihong
    Liu, Tiejun
    Wang, Manqing
    Zhang, Yongqing
    [J]. APPLIED SOFT COMPUTING, 2024, 151
  • [3] Attention-Based Multiscale Spatial-Temporal Convolutional Network for Motor Imagery EEG Decoding
    Zhang, Yu
    Li, Penghai
    Cheng, Longlong
    Li, Mingji
    Li, Hongji
    [J]. IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2024, 70 (01) : 2423 - 2434
  • [4] EEG Motor Imagery Classification using Fusion Convolutional Neural Network
    Zouch, Wassim
    Echtioui, Amira
    [J]. ICAART: PROCEEDINGS OF THE 14TH INTERNATIONAL CONFERENCE ON AGENTS AND ARTIFICIAL INTELLIGENCE - VOL 1, 2022, : 548 - 553
  • [5] EEG-ITNet: An Explainable Inception Temporal Convolutional Network for Motor Imagery Classification
    Salami, Abbas
    Andreu-Perez, Javier
    Gillmeister, Helge
    [J]. IEEE ACCESS, 2022, 10 : 36672 - 36685
  • [6] Filter bank sinc-convolutional network with channel self-attention for high performance motor imagery decoding
    Chen, Jiaming
    Wang, Dan
    Yi, Weibo
    Xu, Meng
    Tan, Xiyue
    [J]. JOURNAL OF NEURAL ENGINEERING, 2023, 20 (02)
  • [7] Prediction of tool wear in sculpture surface by a new fusion method of temporal convolutional network and self-attention
    Jia, Wenjie
    Wang, Wei
    Li, Ziwei
    Li, Hai
    [J]. INTERNATIONAL JOURNAL OF ADVANCED MANUFACTURING TECHNOLOGY, 2022, 121 (3-4): : 2565 - 2583
  • [8] A Spatial Filter Temporal Graph Convolutional Network for decoding motor imagery EEG signals
    Tang, Xianlun
    Zhang, Jing
    Qi, Yidan
    Liu, Ke
    Li, Rui
    Wang, Huiming
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2024, 238
  • [9] Prediction of tool wear in sculpture surface by a new fusion method of temporal convolutional network and self-attention
    Jia, Wenjie
    Wang, Wei
    Li, Ziwei
    Li, Hai
    [J]. International Journal of Advanced Manufacturing Technology, 2022, 121 (3-4): : 2565 - 2583
  • [10] Prediction of tool wear in sculpture surface by a new fusion method of temporal convolutional network and self-attention
    Wenjie Jia
    Wei Wang
    Ziwei Li
    Hai Li
    [J]. The International Journal of Advanced Manufacturing Technology, 2022, 121 : 2565 - 2583