PC-GAIN: Pseudo-label Conditional Generative Adversarial Imputation Networks for Incomplete Data
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}.
Code (1)
Tasks
ImputationMissing ValuesPseudo LabelMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Generative Pseudo-label Refinement for Unsupervised Domain Adaptation
We investigate and characterize the inherent resilience of conditional Generative Adversarial Networks (cGANs) against noise in their conditioning labels, and exploit this fact in the context of Unsupervised Domain Adapt…
Domain AdaptationPseudo LabelUnsupervised Domain AdaptationPseudo Label-Guided Model Inversion Attack via Conditional Generative Adversarial Network
Model inversion (MI) attacks have raised increasing concerns about privacy, which can reconstruct training data from public models. Indeed, MI attacks can be formalized as an optimization problem that seeks private data …
Generative Adversarial NetworkPseudo LabelAn Improved Self-supervised GAN via Adversarial Training
We propose to improve unconditional Generative Adversarial Networks (GAN) by training the self-supervised learning with the adversarial process. In particular, we apply self-supervised learning via the geometric transfor…
ClassificationGeneral ClassificationSelf-Supervised LearningAre Generative Classifiers More Robust to Adversarial Attacks?
There is a rising interest in studying the robustness of deep neural network classifiers against adversaries, with both advanced attack and defence techniques being actively developed. However, most recent work focuses o…
Adversarial DefenseAdversarial RobustnessAdversarial Self-Training Improves Robustness and Generalization for Gradual Domain Adaptation
Gradual Domain Adaptation (GDA), in which the learner is provided with additional intermediate domains, has been theoretically and empirically studied in many contexts. Despite its vital role in security-critical scenari…