Research Repository

See what's under the surface

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

Ali-Gombe, Adamu; Elyan, Eyad

Authors

Adamu Ali-Gombe

Eyad Elyan

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.

Journal Article Type Article
Publication Date Oct 7, 2019
Journal Neurocomputing
Print ISSN 0925-2312
Publisher Elsevier
Peer Reviewed Peer Reviewed
Volume 361
Pages 212-221
Institution 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
DOI https://doi.org/10.1016/j.neucom.2019.06.043
Keywords Image classification; Imbalanced data; Deep learning

Files

This file is under embargo until Jul 19, 2020 due to copyright reasons.

Contact publications@rgu.ac.uk to request a copy for personal use.



Downloadable Citations