Imbalanced Data Classification via Cooperative Interaction Between Classifier and Generator
Open Access
- 2 February 2021
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Neural Networks and Learning Systems
- Vol. 33 (8) , 3343-3356
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.Keywords
This publication has 52 references indexed in Scilit:
- Analysing the classification of imbalanced data-sets with multiple classes: Binarization techniques and ad-hoc approachesKnowledge-Based Systems, 2013
- Predicting disease risks from highly imbalanced data using random forestBMC Medical Informatics and Decision Making, 2011
- Cluster-based under-sampling approaches for imbalanced data distributionsExpert Systems with Applications, 2009
- Index of Balanced Accuracy: A Performance Measure for Skewed Class Distributions2009
- Protein classification with imbalanced dataProteins: Structure, Function, and Bioinformatics, 2007
- Data mining for improved cardiac careACM SIGKDD Explorations Newsletter, 2006
- Minority report in fraud detectionACM SIGKDD Explorations Newsletter, 2004
- A study of the behavior of several methods for balancing machine learning training dataACM SIGKDD Explorations Newsletter, 2004
- An instance-weighting method to induce cost-sensitive treesIEEE Transactions on Knowledge and Data Engineering, 2002
- Distributed data mining in credit card fraud detectionIEEE Intelligent Systems and their Applications, 1999