PC-GAIN: Pseudo-label Conditional Generative Adversarial Imputation Networks for Incomplete Data

Abstract

Datasets with missing values are very common in real world applications. GAIN, a recently proposed deep generative model for missing data imputation, has been proved to outperform many state-of-the-art methods. But GAIN only uses a reconstruction loss in the generator to minimize the imputation error of the non-missing part, ignoring the potential category information which can reflect the relationship between samples. In this paper, we propose a novel unsupervised missing data imputation method named PC-GAIN, which utilizes potential category information to further enhance the imputation power. Specifically, we first propose a pre-training procedure to learn potential category information contained in a subset of low-missing-rate data. Then an auxiliary classifier is determined using the synthetic pseudo-labels. Further, this classifier is incorporated into the generative adversarial framework to help the generator to yield higher quality imputation results. The proposed method can improve the imputation quality of GAIN significantly. Experimental results on various benchmark datasets show that our method is also superior to other baseline approaches. Our code is available at \url{https://github.com/WYu-Feng/pc-gain}.

arXiv comment18pages

References in corpus (12)

• Conditional Generative Adversarial Nets
• MissForest - nonparametric missing value imputation for mixed-type data
• GANs Trained by a Two Time-Scale Update Rule Converge to a Local Nash Equilibrium
• Conditional Image Synthesis With Auxiliary Classifier GANs
• Unsupervised and Semi-supervised Learning with Categorical Generative Adversarial Networks
• The Cramer Distance as a Solution to Biased Wasserstein Gradients
• Synthesizing Tabular Data using Generative Adversarial Networks
• Missing Data Imputation with Adversarially-trained Graph Convolutional Networks
• Establishing strong imputation performance of a denoising autoencoder in a wide range of missing data problems
• MisGAN: Learning from Incomplete Data with Generative Adversarial Networks
• Airline Passenger Name Record Generation using Generative Adversarial Networks
• MIDA: Multiple Imputation using Denoising Autoencoders