MFC-GAN: Class-imbalanced dataset classification using Multiple Fake Class Generative Adversarial Network

被引:123
作者
Ali-Gombe, Adamu [1 ]
Elyan, Eyad [1 ,2 ]
机构
[1] Robert Gordon Univ, Sch Comp Sci & Digital Media, Aberdeen, Scotland
[2] Robert Gordon Univ, Higher Educ Acad, Aberdeen, Scotland
关键词
Image classification; Imbalanced data; Deep learning;
D O I
10.1016/j.neucom.2019.06.043
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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. (C) 2019 Elsevier B.V. All rights reserved.
引用
收藏
页码:212 / 221
页数:10
相关论文
共 34 条
  • [1] Fish Classification in Context of Noisy Images
    Ali-Gombe, Adamu
    Elyan, Eyad
    Jayne, Chrisina
    [J]. ENGINEERING APPLICATIONS OF NEURAL NETWORKS, EANN 2017, 2017, 744 : 216 - 226
  • [2] [Anonymous], 2017, IEEE C COMP VIS PATT, DOI DOI 10.1109/CVPR.2017.525
  • [3] [Anonymous], ARXIV171104340
  • [4] [Anonymous], COMMUN ACM, DOI [DOI 10.1145/3065386, 10.1145/3065386]
  • [5] [Anonymous], ARXIV180102385
  • [6] [Anonymous], 2011, NEURAL INFORM PROCES
  • [7] [Anonymous], P IEEE INT C COMP VI
  • [8] [Anonymous], ICLR2018
  • [9] [Anonymous], ARXIV171100648
  • [10] [Anonymous], 2015, P NIPS