EEGNet-based multi-source domain filter for BCI transfer learning

被引:2
作者
Li, Mengfan [1 ,2 ,3 ]
Li, Jundi [1 ,2 ,3 ]
Song, Zhiyong [1 ,2 ,3 ]
Deng, Haodong [1 ,2 ,3 ]
Xu, Jiaming [4 ,5 ]
Xu, Guizhi [1 ,2 ,3 ]
Liao, Wenzhe [6 ]
机构
[1] Hebei Univ Technol, Sch Elect Engn, State Key Lab Reliabil & Intelligence Elect Equip, Tianjin 300401, Peoples R China
[2] Hebei Key Lab Bioelectromagnet & Neuroengn, Tianjin 300132, Peoples R China
[3] Hebei Univ Technol, Tianjin Key Lab Bioelect & Intelligent Hlth, Tianjin 300130, Peoples R China
[4] Chinese Acad Sci, Inst Automat, Beijing, Peoples R China
[5] Univ Chinese Acad Sci, Sch Artificial Intelligence, Beijing, Peoples R China
[6] Hebei Univ Technol, Sch Artificial Intelligence, Tianjin, Peoples R China
基金
中国国家自然科学基金;
关键词
Brain-computer interface; Multi-source domain filter; Transfer learning; Ensemble learning; EEGNet; NEURAL-NETWORK; SYSTEM;
D O I
10.1007/s11517-023-02967-z
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Deep learning has great potential on decoding EEG in brain-computer interface. While common deep learning algorithms cannot directly train models with data from multiple individuals because of the inter-individual differences in EEG. Collecting enough data for each subject to satisfy the training of deep learning would result in an increase in training cost. This study proposes a novel transfer learning, EEGNet-based multi-source domain filter for transfer learning (EEGNet-MDFTL), to reduce the amount of training data and improve the performance of BCI. The EEGNet-MDFTL uses bagging ensemble learning to learn domain-invariant features from the multi-source domain and utilizes model loss value to filter the multi-source domain. Compared with baseline methods, the accuracy of the EEGNet-MDFTL reaches 91.96%, higher than two state-of-the-art methods, which demonstrates source domain filter can select similar source domains to improve the accuracy of the model, and remains a high level even when the data amount is reduced to 1/8, proving that ensemble learning learns enough domain invariant features from the multi-source domain to make the model insensitive to data amount. The proposed EEGNet-MDFTL is effective in improving the decoding performance with a small amount of data, which is helpful to save the BCI training cost.
引用
收藏
页码:675 / 686
页数:12
相关论文
共 42 条
[21]   A review of rapid serial visual presentation-based brain-computer interfaces [J].
Lees, Stephanie ;
Dayan, Natalie ;
Cecotti, Hubert ;
McCullagh, Paul ;
Maguire, Liam ;
Lotte, Fabien ;
Coyle, Damien .
JOURNAL OF NEURAL ENGINEERING, 2018, 15 (02)
[22]   Transfer Learning Algorithm of P300-EEG Signal Based on XDAWN Spatial Filter and Riemannian Geometry Classifier [J].
Li, Feng ;
Xia, Yi ;
Wang, Fei ;
Zhang, Dengyong ;
Li, Xiaoyu ;
He, Fan .
APPLIED SCIENCES-BASEL, 2020, 10 (05)
[23]   A TrAdaBoost Method for Detecting Multiple Subjects' N200 and P300 Potentials Based on Cross-Validation and an Adaptive Threshold [J].
Li, Mengfan ;
Lin, Fang ;
Xu, Guizhi .
INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2020, 30 (03)
[24]   Automatic feature extraction and fusion recognition of motor imagery EEG using multilevel multiscale CNN [J].
Li, Ming-ai ;
Han, Jian-fu ;
Yang, Jin-fu .
MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2021, 59 (10) :2037-2050
[25]   A Dual Stimuli Approach Combined with Convolutional Neural Network to Improve Information Transfer Rate of Event-Related Potential-Based Brain-Computer Interface [J].
Li, Wei ;
Li, Mengfan ;
Zhou, Huihui ;
Chen, Genshe ;
Jin, Jing ;
Duan, Feng .
INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2018, 28 (10)
[26]   Transfer Feature Learning with Joint Distribution Adaptation [J].
Long, Mingsheng ;
Wang, Jianmin ;
Ding, Guiguang ;
Sun, Jiaguang ;
Yu, Philip S. .
2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2013, :2200-2207
[27]   A review of classification algorithms for EEG-based brain-computer interfaces: a 10 year update [J].
Lotte, F. ;
Bougrain, L. ;
Cichocki, A. ;
Clerc, M. ;
Congedo, M. ;
Rakotomamonjy, A. ;
Yger, F. .
JOURNAL OF NEURAL ENGINEERING, 2018, 15 (03)
[28]   Feature Weighting and Regularization of Common Spatial Patterns in EEG-Based Motor Imagery BCI [J].
Mishuhina, Vasilisa ;
Jiang, Xudong .
IEEE SIGNAL PROCESSING LETTERS, 2018, 25 (06) :783-787
[29]   Improving Cross-State and Cross-Subject Visual ERP-Based BCI With Temporal Modeling and Adversarial Training [J].
Ni, Ziyi ;
Xu, Jiaming ;
Wu, Yuwei ;
Li, Mengfan ;
Xu, Guizhi ;
Xu, Bo .
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2022, 30 :369-379
[30]   A Survey on Transfer Learning [J].
Pan, Sinno Jialin ;
Yang, Qiang .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2010, 22 (10) :1345-1359