Abstract:Abstract Objective. The data scarcity problem in emotion recognition from electroencephalography (EEG) leads to difficulty in building an affective model with high accuracy using machine learning algorithms or deep neural networks. Inspired by emerging deep generative models, we propose three methods for augmenting EEG training data to enhance the performance of emotion recognition models. Approach. Our proposed methods are based on two deep generative models, variational autoencoder (VAE) and generative adversarial network (GAN), and two data augmentation ways, full and partial usage strategies. For the full usage strategy, all of the generated data are augmented to the training dataset without judging the quality of the generated data, while for the partial usage, only high-quality data are selected and appended to the training dataset. These three methods are called conditional Wasserstein GAN (cWGAN), selective VAE (sVAE), and selective WGAN (sWGAN). Main results. To evaluate the effectiveness of these proposed methods, we perform a systematic experimental study on two public EEG datasets for emotion recognition, namely, SEED and DEAP. We first generate realistic-like EEG training data in two forms: power spectral density and differential entropy. Then, we augment the original training datasets with a different number of generated realistic-like EEG data. Finally, we train support vector machines and deep neural networks with shortcut layers to build affective models using the original and augmented training datasets. The experimental results demonstrate that our proposed data augmentation methods based on generative models outperform the existing data augmentation approaches such as conditional VAE, Gaussian noise, and rotational data augmentation. We also observe that the number of generated data should be less than 10 times of the original training dataset to achieve the best performance. Significance. The augmented training datasets produced by our proposed sWGAN method significantly enhance the performance of EEG-based emotion recognition models.

EEG2Vec: Learning Affective EEG Representations via Variational Autoencoders

Multi-view Domain-Adaptive Representation Learning for EEG-based Emotion Recognition

EEG Emotion Recognition Based on Self-attention Dynamic Graph Neural Networks

Variational Autoencoder Based Latent Factor Decoding of Multichannel EEG for Emotion Recognition

Electroencephalogram Emotion Recognition Using Variational Modal Decomposition Based Dispersion Entropy Feature Extraction

Dual-Encoder VAE-GAN With Spatiotemporal Features for Emotional EEG Data Augmentation

Learning Generative Factors of EEG Data with Variational auto-encoders

Latent Factor Decoding of Multi-Channel EEG for Emotion Recognition Through Autoencoder-Like Neural Networks

Data Augmentation for EEG-Based Emotion Recognition Using Generative Adversarial Networks

Electroencephalogram Emotion Recognition Based on Empirical Mode Decomposition and Optimal Feature Selection.

Variational Instance-Adaptive Graph for EEG Emotion Recognition

Data augmentation for enhancing EEG-based emotion recognition with deep generative models

Variational Pathway Reasoning for EEG Emotion Recognition

Multi-Modal Domain Adaptation Variational Autoencoder for EEG-Based Emotion Recognition

Multi-Domain Encoding of Spatiotemporal Dynamics in EEG for Emotion Recognition

hvEEGNet: exploiting hierarchical VAEs on EEG data for neuroscience applications

Disentangled Variational Autoencoder for Emotion Recognition in Conversations

Deep Sparse Autoencoder and Recursive Neural Network for EEG Emotion Recognition

Towards emotion recognition for virtual environments: an evaluation of EEG features on benchmark dataset

VAEEG: Variational Auto-encoder for Extracting EEG Representation

Minimizing EEG Human Interference: A Study of an Adaptive EEG Spatial Feature Extraction with Deep Convolutional Neural Networks