WGAN Domain Adaptation for EEG-Based Emotion Recognition

• Published in: Neural Information Processing Vol. 11305; pp. 275 - 286
• Main Authors: Luo, Yun, Zhang, Si-Yang, Zheng, Wei-Long, Lu, Bao-Liang
• Format: Book Chapter
• Language: English
• Published: Switzerland Springer International Publishing AG 2018; Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: Domain adaptation; EEG; Emotion recognition; GAN
• ISBN: 3030042200; 9783030042202
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-030-04221-9_25

In this paper, we propose a novel Wasserstein generative adversarial network domain adaptation (WGANDA) framework for building cross-subject electroencephalography (EEG)-based emotion recognition models. The proposed framework consists of GANs-like components and a two-step training procedure with pre-training and adversarial training. Pre-training is to map source domain and target domain to a common feature space, and adversarial-training is to narrow down the gap between the mappings of the source and target domains on the common feature space. A Wasserstein GAN gradient penalty loss is applied to adversarial-training to guarantee the stability and convergence of the framework. We evaluate the framework on two public EEG datasets for emotion recognition, SEED and DEAP. The experimental results demonstrate that our WGANDA framework successfully handles the domain shift problem in cross-subject EEG-based emotion recognition and significantly outperforms the state-of-the-art domain adaptation methods.