Cited by 162Emotion and Mood RecognitionEEG and Brain-Computer InterfacesBiometric Identification and Security

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

Open access
Luo, Yun, Zhu, Li-Zhen, Wan, Zi-Yu, Lu, Bao-Liang·2020·Journal of Neural Engineering

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.

Integrity signals

Registry

Crossref · 10.1088/1741-2552/abb580

Cites

49 of 49 references verified0 citation errors · 0 not checkable

Statistics

Recomputing p-values needs the paper’s full text.

Claims the paper makes

We couldn’t retrieve this paper’s full text, so its claims and self-check weren’t run. The registry and citation checks still apply.

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • cWGAN and sVAE DA with power spectral density or differential entropy representations enhanced classifier performance in emotion recognition.

    “Experiments show that two representations of EEG signals were suitable for the requirement of the artificial datasets that enhances the performance of the classifier.”

Cites

Flagged first, then by citations

2 of the 51 references couldn’t be matched to a known paper.

Cited by

Sorted by citations

Authors
Luo, Y., Zhu, L.-Z., Wan, Z.-Y., & Lu, B.-L.
Affiliations
Shanghai Jiao Tong University (CN)
Corresponding author
Lu, Bao-Liang
Venue
Journal of Neural Engineering
Published
2020-10-01
Volume
17
Issue
5
Pages
056021
ISSN
  • 1741-2552 (ISSN-L)
  • 1741-2560
OpenAlex
W3083255269
Cited by
162
References
51
Citations by year
2026: 15 · 2025: 28 · 2024: 39 · 2023: 37 · 2022: 33 · 2021: 8 · 2020: 2
FWCI
19.32
Type
journal article
Language
en
Open access
Locations
  • Journal of Neural Engineering (publishedVersion)
  • PubMed (publishedVersion)
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
Funders
  • National Natural Science Foundation of China (61673266)
  • 111 Project
  • National Key Research and Development Program of China (2017YFB1002501)
  • SJTU Trans-med Awards Research (WF540162605)
  • Fundamental Research Funds for the Central Universities
Topics
  • Social Sciences › Psychology › Experimental and Cognitive Psychology › Emotion and Mood Recognition
  • Life Sciences › Neuroscience › Cognitive Neuroscience › EEG and Brain-Computer Interfaces
  • Physical Sciences › Computer Science › Signal Processing › Biometric Identification and Security
Keywords
EEG-based emotion recognition, data augmentation, deep generative models, variational autoencoder, generative adversarial networks, conditional Wasserstein GAN, data scarcity, power spectral density, differential entropy, affective model, SEED dataset, DEAP dataset
MeSH terms
Machine Learning, Algorithms, Electroencephalography, Emotions, Neural Networks, Computer

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review