Few-Shot Class-Incremental Learning With Dynamic Prototype Refinement for Brain Activity Classification

The brain-computer interface (BCI) system facilitates efficient communication and control, with Electroencephalography (EEG) signals as a vital component. Traditional EEG signal classification, based on static deep-learning models, presents a challenge when new classes of the subject’s brain activity emerge. The goal is to develop a model that can recognize new few-shot classes while preserving its ability to discriminate between existing ones. This scenario is referred to as Few-Shot Class-Incremental Learning (FSCIL). This work introduces IncrementEEG, a novel framework meticulously designed to tackle the distinct challenges of FSCIL in EEG-based brain activity classification, focusing specifically on emotion recognition and steady-state visual evoked potential (SSVEP). Our work analyzes the role of additive angular margin loss in improving the model’s discrimination capabilities. The proposed method is designed to demonstrate robustness in open-world conditions and adaptability to new tasks. Furthermore, we introduce a prototype refinement module comprising a prototype augmentation block and an update block. The prototype augmentation block in the deep feature space preserves the decision boundary for prior tasks, and the prototype update block utilizes a shared embedding space to compute the relation matrix for bootstrapping prototype updates. Extensive experiments conducted across multiple datasets show the superior performance of the IncrementEEG framework compared to state-of-the-art methods. The proposed method advances FSCIL brain activity classification, offering promising potential for applications in Brain-Computer Interface systems.

Authors

• Lei Cao
• Hao Li Shanghai Artificial Intelligence Laboratory & Northwestern Polytechnical University
• Yilin Dong
• Tianyu Liu
• Jie Li Shanghai Jiao Tong University Shanghai Jiao Tong University (Wuxi) Blockchain Advanced Research Center

Keywords

• Brain modeling
• Power capacitors
• Electroencephalography
• Training
• Prototypes
• Emotion recognition
• Brain
• Incremental learning
• Adaptation models
• Training data
• Brain Activity
• Few-shot Learning
• Class-incremental Learning
• Classification Of Brain Activity
• Prototype Refinement
• Few-shot Class-incremental Learning
• Feature Space
• Extensive Experiments
• Relationship Matrix
• Latent Space
• EEG Signals
• Additional Loss
• Brain-computer Interface System
• Steady-state Visual Evoked Potential
• Additional Margin
• Deep Neural Network
• Scaling Factor
• Benchmark Datasets
• Catastrophic Forgetting
• Base Classes
• Classification Of Samples
• Class Prototypes
• Emotion Recognition Task
• Incremental Model
• Earth Mover’s Distance
• Joint Learning
• SSVEP
• BCI
• brain activity classification
• Humans
• Brain-Computer Interfaces
• Signal Processing, Computer-Assisted
• Evoked Potentials, Visual
• Emotions
• Adult
• Deep Learning
• Algorithms
• Male