Conditional Wasserstein GAN-based Oversampling of Tabular Data for Imbalanced Learning

08/20/2020
by   Justin Engelmann, et al.
0

Class imbalance is a common problem in supervised learning and impedes the predictive performance of classification models. Popular countermeasures include oversampling the minority class. Standard methods like SMOTE rely on finding nearest neighbours and linear interpolations which are problematic in case of high-dimensional, complex data distributions. Generative Adversarial Networks (GANs) have been proposed as an alternative method for generating artificial minority examples as they can model complex distributions. However, prior research on GAN-based oversampling does not incorporate recent advancements from the literature on generating realistic tabular data with GANs. Previous studies also focus on numerical variables whereas categorical features are common in many business applications of classification methods such as credit scoring. The paper propoes an oversampling method based on a conditional Wasserstein GAN that can effectively model tabular datasets with numerical and categorical variables and pays special attention to the down-stream classification task through an auxiliary classifier loss. We benchmark our method against standard oversampling methods and the imbalanced baseline on seven real-world datasets. Empirical results evidence the competitiveness of GAN-based oversampling.

READ FULL TEXT
research
05/02/2018

Text to Image Synthesis Using Generative Adversarial Networks

Generating images from natural language is one of the primary applicatio...
research
06/28/2022

Improving Correlation Capture in Generating Imbalanced Data using Differentially Private Conditional GANs

Despite the remarkable success of Generative Adversarial Networks (GANs)...
research
07/17/2018

Airline Passenger Name Record Generation using Generative Adversarial Networks

Passenger Name Records (PNRs) are at the heart of the travel industry. C...
research
10/05/2022

ciDATGAN: Conditional Inputs for Tabular GANs

Conditionality has become a core component for Generative Adversarial Ne...
research
11/02/2017

Oversampling for Imbalanced Learning Based on K-Means and SMOTE

Learning from class-imbalanced data continues to be a common and challen...
research
02/16/2021

CTAB-GAN: Effective Table Data Synthesizing

While data sharing is crucial for knowledge development, privacy concern...

Please sign up or login with your details

Forgot password? Click here to reset