SOTAVerified

Modeling Tabular data using Conditional GAN

2019-07-01NeurIPS 2019Code Available1· sign in to hype

Lei Xu, Maria Skoularidou, Alfredo Cuesta-Infante, Kalyan Veeramachaneni

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

Modeling the probability distribution of rows in tabular data and generating realistic synthetic data is a non-trivial task. Tabular data usually contains a mix of discrete and continuous columns. Continuous columns may have multiple modes whereas discrete columns are sometimes imbalanced making the modeling difficult. Existing statistical and deep neural network models fail to properly model this type of data. We design TGAN, which uses a conditional generative adversarial network to address these challenges. To aid in a fair and thorough comparison, we design a benchmark with 7 simulated and 8 real datasets and several Bayesian network baselines. TGAN outperforms Bayesian methods on most of the real datasets whereas other deep learning methods could not.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
Adult Census IncomeCopulaGANDT Accuracy76.29Unverified
Adult Census IncomeTVAEDT Accuracy82.8Unverified
Adult Census IncomeCTGANDT Accuracy81.32Unverified
California Housing PricesTVAEParameters(M)0.05Unverified
California Housing PricesCTGANParameters(M)0.2Unverified
California Housing PricesCopulaGANParameters(M)0.2Unverified
DiabetesTVAEDT Accuracy0.53Unverified
DiabetesCopulaGANDT Accuracy0.39Unverified
DiabetesCTGANDT Accuracy0.5Unverified
HELOCTVAEDT Accuracy76.39Unverified
HELOCCTGANDT Accuracy61.34Unverified
HELOCCopulaGANDT Accuracy42.36Unverified
SICKTVAEDT Accuracy95.39Unverified
SICKCopulaGANDT Accuracy93.77Unverified
SICKCTGANDT Accuracy92.05Unverified
TravelCTGANDT Accuracy73.3Unverified
TravelCopulaGANDT Accuracy73.61Unverified
TravelTVAEDT Accuracy81.68Unverified

Reproductions