Get our free extension to see links to code for papers anywhere online!

 Add to Chrome

 Add to Firefox

CatalyzeX Code Finder - Browser extension linking code for ML papers across the web! | Product Hunt Embed

IFGAN: Missing Value Imputation using Feature-specific Generative Adversarial Networks

Dec 23, 2020
Wei Qiu, Yangsibo Huang, Quanzheng Li



Missing value imputation is a challenging and well-researched topic in data mining. In this paper, we propose IFGAN, a missing value imputation algorithm based on Feature-specific Generative Adversarial Networks (GAN). Our idea is intuitive yet effective: a feature-specific generator is trained to impute missing values, while a discriminator is expected to distinguish the imputed values from observed ones. The proposed architecture is capable of handling different data types, data distributions, missing mechanisms, and missing rates. It also improves post-imputation analysis by preserving inter-feature correlations. We empirically show on several real-life datasets that IFGAN outperforms current state-of-the-art algorithm under various missing conditions.

* Wei Qiu and Yangsibo Huang contribute equally to this work 


Share this with someone who'll enjoy it:

   Access Paper Source



Share this with someone who'll enjoy it: