GAIN swMATH ID: 33872 Software Authors: Jinsung Yoon, James Jordon, Mihaela van der Schaar Description: GAIN: Missing Data Imputation using Generative Adversarial Nets. We propose a novel method for imputing missing data by adapting the well-known Generative Adversarial Nets (GAN) framework. Accordingly, we call our method Generative Adversarial Imputation Nets (GAIN). The generator (G) observes some components of a real data vector, imputes the missing components conditioned on what is actually observed, and outputs a completed vector. The discriminator (D) then takes a completed vector and attempts to determine which components were actually observed and which were imputed. To ensure that D forces G to learn the desired distribution, we provide D with some additional information in the form of a hint vector. The hint reveals to D partial information about the missingness of the original sample, which is used by D to focus its attention on the imputation quality of particular components. This hint ensures that G does in fact learn to generate according to the true data distribution. We tested our method on various datasets and found that GAIN significantly outperforms state-of-the-art imputation methods. Homepage: https://arxiv.org/abs/1806.02920 Related Software: missForest; MICE; MisGAN; Wasserstein GAN; Adam; CollaGAN; LambdaMART; LightGBM; BRITS; MaskGAN; SeqGAN; AmbientGAN; pix2pix; CycleGAN; Auto-WEKA; XGBoost; Spearmint; randomForest; KnockoffGAN; GitHub Cited in: 8 Documents all top 5 Cited by 38 Authors 2 Alaa, Ahmed M. 2 van der Schaar, Mihaela 1 Biessmann, Felix 1 Cai, Xiangrui 1 Delong, Łukasz 1 Deng, Grace 1 Ding, Xiaoke 1 Ercole, Ari 1 Floto, Andres 1 Gallinari, Patrick 1 Gimson, Alexander 1 Guo, Wenya 1 Han, Cuize 1 Jarrett, Daniel P. 1 Kirchmeyer, Matthieu 1 Kozak, Anna 1 Lange, Dustin 1 Liang, Jiahui 1 Liò, Pietro 1 Mantrach, Amin 1 Matteson, David S. 1 McKinney, Eoin 1 Naidu, Prathik 1 Qian, Zhaozhi 1 Rakotomamonjy, Alain 1 Rukat, Tammo 1 Salinas, David 1 Schelter, Sebastian 1 Schmidt, Phillipp 1 Scholtes, Stefan 1 Song, Xianmin 1 Tao, Pengfei 1 Taptunov, Andrey 1 Tian, Jing 1 Wood, Angela M. 1 Yuan, Xiaojie 1 Zhang, Ying 1 Zhou, Baohang all top 5 Cited in 6 Serials 2 Machine Learning 2 Data Mining and Knowledge Discovery 1 Physica A 1 Information Sciences 1 Journal of Machine Learning Research (JMLR) 1 ASTIN Bulletin Cited in 4 Fields 6 Computer science (68-XX) 4 Statistics (62-XX) 1 Statistical mechanics, structure of matter (82-XX) 1 Game theory, economics, finance, and other social and behavioral sciences (91-XX) Citations by Year