mlsanityData to 7 Oct 2026

Imputation for prediction: beware of diminishing returns

arXiv 2407.19804 ↗First posted 29 Jul 2024cs.AI cs.LG stat.MLcited by 11

Not on the map. Why?

Abstract (arXiv)

Missing values are prevalent across various fields, posing challenges for training and deploying predictive models. In this context, imputation is a common practice, driven by the hope that accurate imputations will enhance predictions. However, recent theoretical and empirical studies indicate that simple constant imputation can be consistent and competitive. This empirical study aims at clarifying if and when investing in advanced imputation methods yields significantly better predictions. Relating imputation and predictive accuracies across combinations of imputation and predictive models on 19 datasets, we show that imputation accuracy matters less i) when using expressive models, ii) when incorporating missingness indicators as complementary inputs, iii) matters much more for generated linear outcomes than for real-data outcomes. Interestingly, we also show that the use of the missingness indicator is beneficial to the prediction performance, even in MCAR scenarios. Overall, on real-data with powerful models, improving imputation only has a minor effect on prediction performance. Thus, investing in better imputations for improved predictions often offers limited benefits.

Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 16 of its 35 references that are in the map. First posted on arXiv 29 Jul 2024.

Builds on

16 of 35 references.

References · 16

  1. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
  2. 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
  3. 2021Deep Neural Networks and Tabular Data: A Surveycited by 489
  4. 2018GAIN: Missing Data Imputation using Generative Adversarial Netscited by 349
  5. 2021SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Trainingcited by 310
  6. 2020Missing Data Imputation using Optimal Transportcited by 99
  7. 2022HyperImpute: Generalized Iterative Imputation with Automatic Model Selectioncited by 81
  8. 2021What's a good imputation to predict with missing values?cited by 61
  9. 2019On the consistency of supervised learning with missing valuescited by 52
  10. 2022Benchmarking missing-values approaches for predictive models on health databasescited by 23
  11. 2022Classification of datasets with imputed missing values: does imputation quality matter?cited by 21
  12. 2022The Missing Indicator Method: From Low to High Dimensionscited by 19
4 more
  1. 2020Does imputation matter? Benchmark for predictive modelscited by 7
  2. 2024Random features models: a way to study the success of naive imputationcited by 7
  3. 2021On the Relation between Prediction and Imputation Accuracy under Missing Covariatescited by 6
  4. 2016Missing Data Imputation for Supervised Learningcited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 4; 2026: 7.

11 papers.