Meta-Learning for EEG Motor Imagery Classification

Authors

  • Jian Yu College of Electronic and Information Engineering, Tongji University, Shanghai 201804, China
  • Lili Duan College of Electronic and Information Engineering, Tongji University, Shanghai 201804, China
  • Hongfei Ji College of Electronic and Information Engineering, Tongji University, Shanghai 201804, China
  • Jie Li College of Electronic and Information Engineering, Tongji University, Shanghai 201804, China & Translational Research Center, Shanghai Yangzhi Rehabilitation Hospital, Shanghai Sunshine Rehabilitation Center, Tongji University, Shanghai, China
  • Zilong Pang School of Software, Intelligent Data Processing Engineering Research Center of Henan Province, Henan University, Kaifeng, 475004, China

DOI:

https://doi.org/10.31577/cai_2024_3_735

Keywords:

Brain-computer interface, motor imagery, pre-training, fine-tuning

Abstract

A brain-computer interface (BCI) based on motor imagery (MI) enables users to communicate with the computer directly using brain signals. However, due to the low signal-to-noise ratio and significant inter-subject variations, its long calibration time hinders the development of brain-computer interfaces. Meta-learning, as the current popular framework in few-shot learning, enables the model to quickly adapt to new few-shot tasks by learning a series of similar tasks. In this paper, we propose a novel algorithm that combines pre-training and meta-learning. It can reduce the number of training samples required for the target subject and meanwhile ensure stable and reliable BCI performance. This is achieved by first learning general feature representations from a large number of other subjects' data through pre-training, and then further optimizing the model by task scale using meta-learning based on the pre-training stage. Applications to a dataset confirm the effectiveness of the combination of pre-training and meta-learning. Results indicate that the proposed algorithm outperforms the considered comparable baseline algorithm, early meta-learning frameworks without pre-training, and fine-tuning in transfer learning. In addition, experiments show that the selection of source subjects can further improve the overall performance of the algorithm, and the pre-training stage is crucial for the model to achieve good performance. This study has significant instruction for the application of BCI in the field of medical rehabilitation. It can improve the classification performance of the disabled with poor motor imagination characteristics by using the data of healthy typical subjects and greatly reducing their training time.

Downloads

Download data is not yet available.

Downloads

Published

2024-06-24

How to Cite

Yu, J., Duan, L., Ji, H., Li, J., & Pang, Z. (2024). Meta-Learning for EEG Motor Imagery Classification. Computing and Informatics, 43(3), 735–755. https://doi.org/10.31577/cai_2024_3_735