Subject-independent meta-learning framework towards optimal training of EEG-based classifiers

Han Wei Ng,Cuntai Guan

DOI: https://doi.org/10.1016/j.neunet.2024.106108

IF: 7.8

2024-04-01

Neural Networks

Abstract:Advances in deep learning have shown great promise towards the application of performing high-accuracy Electroencephalography (EEG) signal classification in a variety of tasks. However, many EEG-based datasets are often plagued by the issue of high inter-subject signal variability. Robust deep learning models are notoriously difficult to train under such scenarios, often leading to subpar or widely varying performance across subjects under the leave-one-subject-out paradigm. Recently, the model agnostic meta-learning framework was introduced as a way to increase the model's ability to generalize towards new tasks. While the original framework focused on task-based meta-learning, this research aims to show that the meta-learning methodology can be modified towards subject-based signal classification while maintaining the same task objectives and achieve state-of-the-art performance. Namely, we propose the novel implementation of a few/zero-shot subject-independent meta-learning framework towards multi-class inner speech and binary class motor imagery classification. Compared to current subject-adaptive methods which utilize large number of labels from the target, the proposed framework shows its effectiveness in training zero-calibration and few-shot models for subject-independent EEG classification. The proposed few/zero-shot subject-independent meta-learning mechanism performs well on both small and large datasets and achieves robust, generalized performance across subjects. The results obtained shows a significant improvement over the current state-of-the-art, with the binary class motor imagery achieving 88.70% and the accuracy of multi-class inner speech achieving an average of 31.15%. Codes will be made available to public upon publication.

computer science, artificial intelligence,neurosciences

What problem does this paper attempt to address?

The paper attempts to address the problem of how to improve the generalization ability of EEG signal-based classification models in brain-computer interface (BCI) applications through a meta-learning framework, especially in the face of inter-individual signal variability and limited data resources. Specifically, the study aims at the following aspects:

1. **Propose a new meta-learning framework**: This framework can train models with stronger generalization ability in the case of few or no target individual samples, to cope with the differences in EEG signals between different individuals.

2. **Validate the effectiveness of the framework in different EEG classification tasks**: By validating on two different EEG classification tasks—inner speech recognition and motor imagery classification, the proposed meta-learning method is shown to effectively overcome the problem of data resource scarcity.

3. **Achieve state-of-the-art classification accuracy**: For smaller datasets (such as inner speech tasks), it is demonstrated that only a few target individual samples are needed to achieve good classification results; while in motor imagery classification tasks, the best performance can be obtained without target individual data.

Through these contributions, the study aims to advance EEG signal classification technology, particularly by reducing the need for large amounts of calibration data in practical application scenarios, thereby enhancing the practicality and user experience of brain-computer interface systems.