GANSER: A Self-supervised Data Augmentation Framework for EEG-based Emotion Recognition
This work addresses the data scarcity problem in EEG-based affective computing, which is crucial for building accurate and stable deep learning models in this domain, though it appears incremental as it builds on existing data augmentation and adversarial training techniques.
The paper tackles data scarcity in EEG-based emotion recognition by proposing GANSER, a novel data augmentation framework that combines adversarial training with self-supervised learning to generate high-quality and diverse simulated EEG samples, achieving state-of-the-art results in performance gain.
The data scarcity problem in Electroencephalography (EEG) based affective computing results into difficulty in building an effective model with high accuracy and stability using machine learning algorithms especially deep learning models. Data augmentation has recently achieved considerable performance improvement for deep learning models: increased accuracy, stability, and reduced over-fitting. In this paper, we propose a novel data augmentation framework, namely Generative Adversarial Network-based Self-supervised Data Augmentation (GANSER). As the first to combine adversarial training with self-supervised learning for EEG-based emotion recognition, the proposed framework can generate high-quality and high-diversity simulated EEG samples. In particular, we utilize adversarial training to learn an EEG generator and force the generated EEG signals to approximate the distribution of real samples, ensuring the quality of augmented samples. A transformation function is employed to mask parts of EEG signals and force the generator to synthesize potential EEG signals based on the remaining parts, to produce a wide variety of samples. The masking possibility during transformation is introduced as prior knowledge to guide to extract distinguishable features for simulated EEG signals and generalize the classifier to the augmented sample space. Finally, extensive experiments demonstrate our proposed method can help emotion recognition for performance gain and achieve state-of-the-art results.