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

被引:180
作者
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
相关论文
共 42 条
[1]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[2]  
[Anonymous], 2014, UNSUPERVISED DOMAIN
[3]  
[Anonymous], 2016, ADV NEURAL INFORM PR
[4]  
[Anonymous], 2019, ARXIV PREPRINT ARXIV
[5]   EEG Data Space Adaptation to Reduce Intersession Nonstationarity in Brain-Computer Interface [J].
Arvaneh, Mahnaz ;
Guan, Cuntai ;
Ang, Kai Keng ;
Quek, Chai .
NEURAL COMPUTATION, 2013, 25 (08) :2146-2171
[6]   Dynamic time warping-based transfer learning for improving common spatial patterns in brain-computer interface [J].
Azab, Ahmed M. ;
Ahmadi, Hamed ;
Mihaylova, Lyudmila ;
Arvaneh, Mahnaz .
JOURNAL OF NEURAL ENGINEERING, 2020, 17 (01)
[7]   Optimizing spatial filters for robust EEG single-trial analysis [J].
Blankertz, Benjamin ;
Tomioka, Ryota ;
Lemm, Steven ;
Kawanabe, Motoaki ;
Mueller, Klaus-Robert .
IEEE SIGNAL PROCESSING MAGAZINE, 2008, 25 (01) :41-56
[8]   Deep learning for electroencephalogram (EEG) classification tasks: a review [J].
Craik, Alexander ;
He, Yongtian ;
Contreras-Vidal, Jose L. .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (03)
[9]   Inter-subject transfer learning with an end-to-end deep convolutional neural network for EEG-based BCI [J].
Fahimi, Fatemeh ;
Zhang, Zhuo ;
Goh, Wooi Boon ;
Lee, Tih-Shi ;
Ang, Kai Keng ;
Guan, Cuntai .
JOURNAL OF NEURAL ENGINEERING, 2019, 16 (02)
[10]   Deep Reconstruction-Classification Networks for Unsupervised Domain Adaptation [J].
Ghifary, Muhammad ;
Kleijn, W. Bastiaan ;
Zhang, Mengjie ;
Balduzzi, David ;
Li, Wen .
COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 :597-613