Adaptive transfer learning for EEG motor imagery classification with deep Convolutional Neural Network

被引:146
|
作者
Zhang, Kaishuo [1 ]
Robinson, Neethu [1 ]
Lee, Seong-Whan [2 ]
Guan, Cuntai [1 ]
机构
[1] Nanyang Technol Univ, Sch Comp Sci & Engn, Singapore 639798, Singapore
[2] Korea Univ, Dept Artificial Intelligence, Seoul 02841, South Korea
关键词
Transfer learning; Brain-computer interface (BCI); Electroencephalography (EEG); Convolutional Neural Network (CNN); BRAIN-COMPUTER INTERFACES;
D O I
10.1016/j.neunet.2020.12.013
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent years, deep learning has emerged as a powerful tool for developing Brain-Computer Interface (BCI) systems. However, for deep learning models trained entirely on the data from a specific individual, the performance increase has only been marginal owing to the limited availability of subject-specific data. To overcome this, many transfer-based approaches have been proposed, in which deep networks are trained using pre-existing data from other subjects and evaluated on new target subjects. This mode of transfer learning however faces the challenge of substantial inter-subject variability in brain data. Addressing this, in this paper, we propose 5 schemes for adaptation of a deep convolutional neural network (CNN) based electroencephalography (EEG)-BCI system for decoding hand motor imagery (MI). Each scheme fine-tunes an extensively trained, pre-trained model and adapt it to enhance the evaluation performance on a target subject. We report the highest subject independent performance with an average (N = 54) accuracy of 84.19% (+/- 9.98%) for two-class motor imagery, while the best accuracy on this dataset is 74.15% (+/- 15.83%) in the literature. Further, we obtain a statistically significant improvement (p = 0.005) in classification using the proposed adaptation schemes compared to the baseline subject-independent model. (C) 2020 Elsevier Ltd. All rights reserved.
引用
收藏
页码:1 / 10
页数:10
相关论文
共 50 条
  • [1] 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
  • [2] A Novel Simplified Convolutional Neural Network Classification Algorithm of Motor Imagery EEG Signals Based on Deep Learning
    Li, Feng
    He, Fan
    Wang, Fei
    Zhang, Dengyong
    Xia, Yi
    Li, Xiaoyu
    [J]. APPLIED SCIENCES-BASEL, 2020, 10 (05):
  • [3] A Novel Convolutional Neural Network Classification Approach of Motor-Imagery EEG Recording Based on Deep Learning
    Echtioui, Amira
    Mlaouah, Ayoub
    Zouch, Wassim
    Ghorbel, Mohamed
    Mhiri, Chokri
    Hamam, Habib
    [J]. APPLIED SCIENCES-BASEL, 2021, 11 (21):
  • [4] EEG Representation in Deep Convolutional Neural Networks for Classification of Motor Imagery
    Robinson, Neethu
    Lee, Seong-Whan
    Guan, Cuntai
    [J]. 2019 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN AND CYBERNETICS (SMC), 2019, : 1322 - 1326
  • [5] Classification of Motor Imagery EEG Signals Based on Deep Autoencoder and Convolutional Neural Network Approach
    Hwaidi, Jamal F.
    Chen, Thomas M.
    [J]. IEEE ACCESS, 2022, 10 : 48071 - 48081
  • [6] 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
  • [7] Hybrid deep neural network using transfer learning for EEG motor imagery decoding
    Zhang, Ruilong
    Zong, Qun
    Dou, Liqian
    Zhao, Xinyi
    Tang, Yifan
    Li, Zhiyu
    [J]. BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 63
  • [8] A convolutional spiking neural network with adaptive coding for motor imagery classification
    Liao, Xiaojian
    Wu, Yuli
    Wang, Zi
    Wang, Deheng
    Zhang, Hongmiao
    [J]. NEUROCOMPUTING, 2023, 549
  • [9] Motor Imagery EEG Signal Classification Using Optimized Convolutional Neural Network
    Thiyam, Deepa Beeta
    Raymond, Shelishiyah
    Avasarala, Padmanabha Sarma
    [J]. PRZEGLAD ELEKTROTECHNICZNY, 2024, 100 (08): : 273 - 279
  • [10] Deep Neural Network with Attention Mechanism for Classification of Motor Imagery EEG
    Huang, Yen-Cheng
    Chang, Jia-Ren
    Chen, Li-Fen
    Chen, Yong-Sheng
    [J]. 2019 9TH INTERNATIONAL IEEE/EMBS CONFERENCE ON NEURAL ENGINEERING (NER), 2019, : 1130 - 1133