Using WGAN for Improving Imbalanced Classification Performance. Bhatia, S. & Dahyot, R. In Curry, E., Keane, M., Ojo, A., & Salwala, D., editors, 27th Irish Conference on Artificial Intelligence and Cognitive Science, pages 365-375, Galway, Ireland, 2019.
Using WGAN for Improving Imbalanced Classification Performance [pdf]Paper  abstract   bibtex   
This paper investigates data synthesis with a Generative Adversarial Network (GAN) for augmenting the amount of data used for training classifiers (in supervised learning) to compensate for class imbalance (when the classes are not represented equally by the same number of training samples). Our data synthesis approach with GAN is compared with data augmentation in the context of image classification. Our experimental results show encouraging results in comparison to standard data augmentation schemes based on image transforms.

Downloads: 0