EEG Data Augmentation for Emotion Recognition Using a Conditional Wasserstein GAN.

Abstract

Due to the lack of electroencephalography (EEG) data, it is hard to build an emotion recognition model with high accuracy from EEG signals using machine learning approach. Inspired by generative adversarial networks (GANs), we introduce a Conditional Wasserstein GAN (CWGAN) framework for EEG data augmentation to enhance EEG-based emotion recognition. A Wasserstein GAN with gradient penalty is adopted to generate realistic-like EEG data in differential entropy (DE) form. Three indicators are used to judge the qualities of the generated high-dimensional EEG data, and only high quality data are appended to supplement the data manifold, which leads to better classification of different emotions. We evaluate the proposed CWGAN framework on two public EEG datasets for emotion recognition, namely SEED and DEAP. The experimental results demonstrate that using the EEG data generated by CWGAN significantly improves the accuracies of emotion recognition models.