Imbalanced Data Classification via Cooperative Interaction Between Classifier and Generator

Hyun Soo Choi, Dahuin Jung, Siwon Kim, Sungroh Yoon

Research output: Contribution to journalArticlepeer-review

20 Scopus citations

Abstract

Learning classifiers with imbalanced data can be strongly biased toward the majority class. To address this issue, several methods have been proposed using generative adversarial networks (GANs). Existing GAN-based methods, however, do not effectively utilize the relationship between a classifier and a generator. This article proposes a novel three-player structure consisting of a discriminator, a generator, and a classifier, along with decision boundary regularization. Our method is distinctive in which the generator is trained in cooperation with the classifier to provide minority samples that gradually expand the minority decision region, improving performance for imbalanced data classification. The proposed method outperforms the existing methods on real data sets as well as synthetic imbalanced data sets.

Original languageEnglish
Pages (from-to)3343-3356
Number of pages14
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume33
Issue number8
DOIs
StatePublished - 1 Aug 2022

Keywords

  • Classification
  • decision boundary
  • deep learning
  • generative adversarial networks (GANs)
  • imbalanced data
  • supervised learning

Fingerprint

Dive into the research topics of 'Imbalanced Data Classification via Cooperative Interaction Between Classifier and Generator'. Together they form a unique fingerprint.

Cite this