Data augmentation for intra-class imbalance with generative adversarial network

Abstract

In classification tasks, the accuracy of classifiers depends on training data. It is known that inter-class imbalanced data degrade the classification accuracy. Previous approaches tend to use data augmentation to solve inter-class imbalance, but the possibility of intra-class imbalance has been ignored. In this paper, we propose a novel method to solve the intra-class imbalance with Generative Adversarial Networks (GAN). The key idea is to examine the distribution of training data in latent space. We experimentally demonstrate that the proposed method generates diverse images and improves classification accuracy on the CIFAR-10 dataset.