Using Bidirectional GAN with Improved Training Architecture for Imbalanced Tasks

Wenjia Zhang,Peng Peng,Heming Zhang
DOI: https://doi.org/10.1109/CSCWD49262.2021.9437750
2021-01-01
Abstract:Classification with an imbalanced dataset has been a hard task for a long time. As generative models become more and more attractive, many researchers try to use these methods to address the imbalance problem. In this paper, we employ Bidirectional Generative Adversarial Networks (BIGAN) with improved training architecture as an oversampling method for minority class. We introduce an additional feedback by making use of the encoder E and further enhance the training of generator G. We conduct the qualitative and quantitative assessment on generated samples, and it shows that under the premise of ensuring diversity, the generated data tend to have less noise. Using it to perform data augmentation in training process, our method has been proved to get a better performance than original GAN-based method on multi-class imbalanced classification task.
What problem does this paper attempt to address?