A cross-session non-stationary attention-based motor imagery classification method with critic-free domain adaptation

被引:1
作者
Guo, Shuai [1 ,3 ]
Wang, Yi [1 ,3 ]
Zhang, Xin [2 ]
Tang, Baoping [1 ,3 ]
机构
[1] Chongqing Univ, Coll Mech & Vehicle Engn, Chongqing 400044, Peoples R China
[2] Chongqing Univ, Bioengn Coll, Chongqing 400044, Peoples R China
[3] Chongqing Univ, State Key Lab Mech Transmiss Adv Equipment, Chongqing 400044, Peoples R China
关键词
Brain-computer interface; Motor imagery; Domain adaptation; Attention mechanism; NETWORK; TRANSFORMER; INFORMATION;
D O I
10.1016/j.bspc.2024.107122
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Recent studies increasingly employ deep learning to decode electroencephalogram (EEG) signals. While deep learning has improved the performance of motor imagery (MI) classification to some extent, challenges remain due to significant variances in EEG data across sessions and the limitations of convolutional neural networks (CNNs). EEG signals are inherently non-stationary, traditional multi-head attention typically uses normalization methods to reduce non-stationarity and improve performance. However, non-stationary factors are crucial inherent properties of EEG signals and provide valuable guidance for decoding temporal dependencies in EEG signals. In this paper, we introduce a novel CNN combined with the Non-stationary Attention (NSA) and Critic-free Domain Adaptation Network (NSDANet), tailored for decoding MI signals. This network starts with temporal-spatial convolution devised to extract spatial-temporal features from EEG signals. It then obtains multi-modal information from average and variance perspectives. We devise anew self-attention module, the Non-stationary Attention (NSA), to capture the non-stationary temporal dependencies of MI-EEG signals. Furthermore, to align feature distributions between the source and target domains, we propose a critic-free domain adaptation network that uses the Nuclear-norm Wasserstein discrepancy (NWD) to minimize the inter- domain differences. NWD complements the original classifier by acting as a critic without a gradient penalty. This integration leverages discriminative information for feature alignment, thus enhancing EEG decoding performance. We conducted extensive cross-session experiments on both BCIC IV 2a and BCIC IV 2b dataset. Results demonstrate that the proposed method outperforms some existing approaches.
引用
收藏
页数:12
相关论文
共 50 条
[1]   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
[2]  
Arjovsky M, 2017, PR MACH LEARN RES, V70
[3]   Easy Domain Adaptation for cross-subject multi-view emotion recognition [J].
Chen, Chuangquan ;
Vong, Chi-Man ;
Wang, Shitong ;
Wang, Hongtao ;
Pang, Miaoqi .
KNOWLEDGE-BASED SYSTEMS, 2022, 239
[4]   FB-CGANet: filter bank channel group attention network for multi-class motor imagery classification [J].
Chen, Jiaming ;
Yi, Weibo ;
Wang, Dan ;
Du, Jinlian ;
Fu, Lihua ;
Li, Tong .
JOURNAL OF NEURAL ENGINEERING, 2022, 19 (01)
[5]  
Cui SH, 2021, Arxiv, DOI arXiv:2107.06154
[6]   Towards Discriminability and Diversity: Batch Nuclear-norm Maximization under Label Insufficient Situations [J].
Cui, Shuhao ;
Wang, Shuhui ;
Zhuo, Junbao ;
Li, Liang ;
Huang, Qingming ;
Tian, Qi .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :3940-3949
[7]  
Farahani A., 2021, Advances in data science and information engineering, P877
[8]  
Ganin Y, 2016, J MACH LEARN RES, V17
[9]   Multiclass common spatial patterns and information theoretic feature extraction [J].
Grosse-Wentrup, Moritz ;
Buss, Martin .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2008, 55 (08) :1991-2000
[10]   Unsupervised domain adaptation for cross-modality liver segmentation via joint adversarial learning and self-learning [J].
Hong, Jin ;
Yu, Simon Chun-Ho ;
Chen, Weitian .
APPLIED SOFT COMPUTING, 2022, 121