MFC-GAN: class-imbalanced dataset classification using multiple fake class generative adversarial network.
Ali-Gombe, Adamu; Elyan, Eyad
Professor Eyad Elyan firstname.lastname@example.org
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.
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|
|Peer Reviewed||Peer Reviewed|
|Keywords||Image classification; Imbalanced data; Deep learning|
ALI-GOMBE 2019 MFC-GAN
Publisher Licence URL
You might also like
A data-driven decision support tool for offshore oil and gas decommissioning.
A deep learning digitisation framework to mark up corrosion circuits in piping and instrumentation diagrams.
Artificial intelligence surgery: how do we get to autonomous actions in surgery?
Class-decomposition and augmentation for imbalanced data sentiment analysis.
Face detection with YOLO on edge.