Explainable cross-task adaptive transfer learning for motor imagery EEG classification

被引:1
|
作者
Miao, Minmin [1 ,2 ]
Yang, Zhong [1 ]
Zeng, Hong [3 ]
Zhang, Wenbin [4 ]
Xu, Baoguo [3 ]
Hu, Wenjun [1 ,2 ]
机构
[1] Huzhou Univ, Sch Informat Engn, Huzhou, Peoples R China
[2] Huzhou Univ, Zhejiang Prov Key Lab Smart Management & Applicat, Huzhou, Peoples R China
[3] Southeast Univ, Sch Instrument Sci & Engn, Nanjing, Peoples R China
[4] Hohai Univ, Coll Comp & Informat, Nanjing, Peoples R China
基金
中国国家自然科学基金;
关键词
brain-computer interface; motor imagery; transfer learning; cross-task; explainability; EVENT-RELATED DESYNCHRONIZATION; CONVOLUTIONAL NEURAL-NETWORK; BRAIN-COMPUTER INTERFACES; DOMAIN ADAPTATION; POTENTIALS; ACTIVATION; EXECUTION; PATTERNS;
D O I
10.1088/1741-2552/ad0c61
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. In the field of motor imagery (MI) electroencephalography (EEG)-based brain-computer interfaces, deep transfer learning (TL) has proven to be an effective tool for solving the problem of limited availability in subject-specific data for the training of robust deep learning (DL) models. Although considerable progress has been made in the cross-subject/session and cross-device scenarios, the more challenging problem of cross-task deep TL remains largely unexplored. Approach. We propose a novel explainable cross-task adaptive TL method for MI EEG decoding. Firstly, similarity analysis and data alignment are performed for EEG data of motor execution (ME) and MI tasks. Afterwards, the MI EEG decoding model is obtained via pre-training with extensive ME EEG data and fine-tuning with partial MI EEG data. Finally, expected gradient-based post-hoc explainability analysis is conducted for the visualization of important temporal-spatial features. Main results. Extensive experiments are conducted on one large ME EEG High-Gamma dataset and two large MI EEG datasets (openBMI and GIST). The best average classification accuracy of our method reaches 80.00% and 72.73% for OpenBMI and GIST respectively, which outperforms several state-of-the-art algorithms. In addition, the results of the explainability analysis further validate the correlation between ME and MI EEG data and the effectiveness of ME/MI cross-task adaptation. Significance. This paper confirms that the decoding of MI EEG can be well facilitated by pre-existing ME EEG data, which largely relaxes the constraint of training samples for MI EEG decoding and is important in a practical sense.
引用
收藏
页数:25
相关论文
共 50 条
  • [1] Task Transfer Learning for EEG Classification in Motor Imagery-Based BCI System
    Zheng, Xuanci
    Li, Jie
    Ji, Hongfei
    Duan, Lili
    Li, Maozhen
    Pang, Zilong
    Zhuang, Jie
    Rongrong, Lu
    Tianhao, Gao
    [J]. COMPUTATIONAL AND MATHEMATICAL METHODS IN MEDICINE, 2020, 2020
  • [2] Adaptive transfer learning for EEG motor imagery classification with deep Convolutional Neural Network
    Zhang, Kaishuo
    Robinson, Neethu
    Lee, Seong-Whan
    Guan, Cuntai
    [J]. NEURAL NETWORKS, 2021, 136 : 1 - 10
  • [3] Excellent fine-tuning: From specific-subject classification to cross-task classification for motor imagery
    Jia, Xueyu
    Song, Yonghao
    Xie, Longhan
    [J]. BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 79
  • [4] Cross-Task Mental Workload Recognition Based on EEG Tensor Representation and Transfer Learning
    Guan, Kai
    Zhang, Zhimin
    Liu, Tao
    Niu, Haijun
    [J]. IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2023, 31 : 2632 - 2639
  • [5] Empowering EEG motor imagery classification with deep transfer learning approach
    Mishra, Awanish Kumar
    Gupta, Indresh Kumar
    Srivastava, Swati
    Alfarhood, Sultan
    Safran, Mejdl
    [J]. EXPERT SYSTEMS, 2024, 41 (04)
  • [6] Motor Imagery EEG Signal Classification based on Deep Transfer Learning
    Wei, Mingnan
    Yang, Rui
    Huang, Mengjie
    [J]. 2021 IEEE 34TH INTERNATIONAL SYMPOSIUM ON COMPUTER-BASED MEDICAL SYSTEMS (CBMS), 2021, : 85 - 90
  • [7] Classification of motor imagery electroencephalogram signals by using adaptive cross-subject transfer learning
    Feng, Jin
    Li, Yunde
    Jiang, Chengliang
    Liu, Yu
    Li, Mingxin
    Hu, Qinghui
    [J]. FRONTIERS IN HUMAN NEUROSCIENCE, 2022, 16
  • [8] Dual selections based knowledge transfer learning for cross-subject motor imagery EEG classification
    Luo, Tian-jian
    [J]. FRONTIERS IN NEUROSCIENCE, 2023, 17
  • [9] Classification and Transfer Learning of EEG during a Kinesthetic Motor Imagery Task using Deep Convolutional Neural Networks
    Craik, Alexander
    Kilicarslan, Atilla
    Contreras-Vidal, Jose L.
    [J]. 2019 41ST ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2019, : 3046 - 3049
  • [10] EEG-Based Motor Imagery Classification with Deep Multi-Task Learning
    Song, Yaguang
    Wang, Danli
    Yue, Kang
    Zheng, Nan
    Shen, Zuo-Jun Max
    [J]. 2019 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2019,