Feature Augmentation for Imbalanced Classification with Conditional Mixture WGANs.

Yinghui Zhang,Bo Sun,Yongkang Xiao,Rong Xiao,YunGang Wei
DOI: https://doi.org/10.1016/j.image.2019.03.010
IF: 3.453
2019-01-01
Signal Processing Image Communication
Abstract:Heterogeneity of class distribution is an intrinsic property of a real-world dataset. Therefore, imbalanced classification is a popular but challenging task. Several methods exist to address this problem. Notably, the adversarial-based data augmentation method, which aims to directly learn the distribution of minority classes unlike simple data modification, has been applied to the challenging task. While effective, the method focuses on a certain domain and lacks universality, and the generated samples lack diversity due to the mode collapse of Generative Adversarial Networks (GANs). In this paper, we propose a general framework of data augmentation using GANs in feature space for imbalanced classification. The core of the framework comprises conditional mixture WGANs (cMWGANs), which are used to approximate true feature distribution and generate label preserved and diverse features for the minority class of various datasets. We conduct three experiments on SVHN, FER2013, and Amazon Review of Instant Video to demonstrate the versatility of the framework and better performance of our cMWGANs in single feature learning. The results show significant improvement with feature augmentation of cMWGANs.
What problem does this paper attempt to address?