A transfer learning framework based on motor imagery rehabilitation for stroke
- PMID: 34611209
- PMCID: PMC8492790
- DOI: 10.1038/s41598-021-99114-1
A transfer learning framework based on motor imagery rehabilitation for stroke
Abstract
Deep learning networks have been successfully applied to transfer functions so that the models can be adapted from the source domain to different target domains. This study uses multiple convolutional neural networks to decode the electroencephalogram (EEG) of stroke patients to design effective motor imagery (MI) brain-computer interface (BCI) system. This study has introduced 'fine-tune' to transfer model parameters and reduced training time. The performance of the proposed framework is evaluated by the abilities of the models for two-class MI recognition. The results show that the best framework is the combination of the EEGNet and 'fine-tune' transferred model. The average classification accuracy of the proposed model for 11 subjects is 66.36%, and the algorithm complexity is much lower than other models.These good performance indicate that the EEGNet model has great potential for MI stroke rehabilitation based on BCI system. It also successfully demonstrated the efficiency of transfer learning for improving the performance of EEG-based stroke rehabilitation for the BCI system.
© 2021. The Author(s).
Conflict of interest statement
Author Jiali Xu was employed by the company Shandong Energy Group Co Ltd. The remaining authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.
Figures
References
-
- Graimann B, Allison B, Pfurtscheller G. Brain–Computer Interfaces: A Gentle Introduction Brain-Computer Interfaces. Springer; 2009. pp. 1–27.
-
- Liu, C., Wang, H., Pu, H., et al. EEG feature extraction and pattern recognition during right and left hands motor imagery in brain-computer interface. In 2012 5th International Conference on BioMedical Engineering and Informatics. 506–510 (2012), 10.1109/BMEI.2012.6513023
-
- Bakhshi A. A comparison among classification accuracy of neural network, flda and blda in p300-based BCI system. Int. J. Comput. Appl. 2012;46(46):11–15.
Publication types
MeSH terms
LinkOut - more resources
Full Text Sources
Medical
