vix.ing · top · new · best · stats

Data Augmentation in Emotion Classification Using Generative Adversarial Networks

2017/11/02 by Xinyue Zhu, Yifan Liu, Zhu, Xinyue +5 · 6 citations
Computer Science · #Anomaly Detection Techniques and Applications #Computer Vision and Pattern Recognition (cs.CV) #Digital Media Forensic Detection #FOS: Computer and information sciences #Generative Adversarial Networks and Image Synthesis #cs.CV

paper · pdf · doi:10.48550/arxiv.1711.00648

openalex publication_date 2017/11/02 · openalex created_date 2017/11/10 · arxiv created 2017/12/14 · arxiv updated 2017/12/15 · openalex updated_date 2026/07/28

Abstract

It is a difficult task to classify images with multiple class labels using only a small number of labeled examples, especially when the label (class) distribution is imbalanced. Emotion classification is such an example of imbalanced label distribution, because some classes of emotions like disgusted are relatively rare comparing to other labels like \it happy or sad. In this paper, we propose a data augmentation method using generative adversarial networks (GAN). It can complement and complete the data manifold and find better margins between neighboring classes. Specifically, we design a framework with a CNN model as the classifier and a cycle-consistent adversarial networks (CycleGAN) as the generator. In order to avoid gradient vanishing problem, we employ the least-squared loss as adversarial loss. We also propose several evaluation methods on three benchmark datasets to validate GAN's performance. Empirical results show that we can obtain 5%~10% increase in the classification accuracy after employing the GAN-based data augmentation techniques.

Citations

Cited by

Related