MFC-GAN: class-imbalanced dataset classification using multiple fake class generative adversarial network.

Ali-Gombe, Adamu; Elyan, Eyad

Abstract

Class-imbalanced datasets are common across different domains such as health, banking, security and others. With such datasets, the learning algorithms are often biased toward the majority class-instances. Data Augmentation is a common approach that aims at rebalancing a dataset by injecting more data samples of the minority class instances. In this paper, a new data augmentation approach is proposed using a Generative Adversarial Networks (GAN) to handle the class imbalance problem. Unlike common GAN models, which use a single fake class, the proposed method uses multiple fake classes to ensure a fine-grained generation and classification of the minority class instances. Moreover, the proposed GAN model is conditioned to generate minority class instances aiming at rebalancing the dataset. Extensive experiments were carried out using public datasets, where synthetic samples generated using our model were added to the imbalanced dataset, followed by performing classification using Convolutional Neural Network. Experiment results show that our model can generate diverse minority class instances, even in extreme cases where the number of minority class instances is relatively low. Additionally, superior performance of our model over other common augmentation and oversampling methods was achieved in terms of classification accuracy and quality of the generated samples.

Citation

ALI-GOMBE, A. and ELYAN, E. 2019. MFC-GAN: class-imbalanced dataset classification using multiple fake class generative adversarial network. Neurocomputing [online], 361, pages 212-221. Available from: https://doi.org/10.1016/j.neucom.2019.06.043

• Journal Article Type: Article
• Acceptance Date: Jun 18, 2019
• Online Publication Date: Jul 18, 2019
• Publication Date: Oct 7, 2019
• Deposit Date: Jul 19, 2019
• Publicly Available Date: Jul 19, 2020
• Journal: Neurocomputing
• Print ISSN: 0925-2312
• Electronic ISSN: 1872-8286
• Publisher: Elsevier
• Peer Reviewed: Peer Reviewed
• Volume: 361
• Pages: 212-221
• DOI: https://doi.org/10.1016/j.neucom.2019.06.043
• Keywords: Image classification; Imbalanced data; Deep learning
• Public URL: https://rgu-repository.worktribe.com/output/323565
• Contract Date: Jul 19, 2019