Imputation for prediction: beware of diminishing returns
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
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
- 2021Deep Neural Networks and Tabular Data: A Surveycited by 489
- 2018GAIN: Missing Data Imputation using Generative Adversarial Netscited by 349
- 2021SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Trainingcited by 310
- 2020Missing Data Imputation using Optimal Transportcited by 99
- 2022HyperImpute: Generalized Iterative Imputation with Automatic Model Selectioncited by 81
- 2021What's a good imputation to predict with missing values?cited by 61
- 2019On the consistency of supervised learning with missing valuescited by 52
- 2022Benchmarking missing-values approaches for predictive models on health databasescited by 23
- 2022Classification of datasets with imputed missing values: does imputation quality matter?cited by 21
- 2022The Missing Indicator Method: From Low to High Dimensionscited by 19
4 more
- 2020Does imputation matter? Benchmark for predictive modelscited by 7
- 2024Random features models: a way to study the success of naive imputationcited by 7
- 2021On the Relation between Prediction and Imputation Accuracy under Missing Covariatescited by 6
- 2016Missing Data Imputation for Supervised Learningcited by 5
Built on by
Citing papers per year: 2025: 4; 2026: 7.
11 papers.
- 2025Missing Data in Signal Processing and Machine Learning: Models, Methods and Modern Approachescited by 2
- 2025Missing Data Multiple Imputation for Tabular Q-Learning in Online RLcited by 1
- 2026A Multimodal Deep Learning Framework for Predicting ICU Deterioration: Integrating ECG Waveforms with Clinical Data and Clinician Benchmarkingcited by 1
- 2025Imputation-free Learning of Tabular Data with Missing Values using Incremental Feature Partitions in Transformercited by 0
- 2025Optimal Transport with Heterogeneously Missing Datacited by 0
- 2026On the Inherent Privacy Amplification of Missing Datacited by 0
- 2026Handling Covariate Mismatch in Collaborative Linear Predictioncited by 0
- 2026Characterizing the Predictive Impact of Modalities with Supervised Latent-Variable Modelingcited by 0
- 2026Hypernetworks for Dynamic Feature Selectioncited by 0
- 2026Distributionally Faithful Imputation via Positive Semi-Definite Kernel Density Estimationcited by 0
- 2026Hybrid Methods for Robust Tabular Data Imputationcited by 0
- 2026Hybrid Methods for Robust Tabular Data Imputation30 Sep 2026
- 2026Distributionally Faithful Imputation via Positive Semi-Definite Kernel Density Estimation8 Jul 2026
- 2026Hypernetworks for Dynamic Feature Selection12 May 2026
- 2026Characterizing the Predictive Impact of Modalities with Supervised Latent-Variable Modeling19 Feb 2026
- 2026Handling Covariate Mismatch in Collaborative Linear Prediction2 Feb 2026
- 2026On the Inherent Privacy Amplification of Missing Data2 Feb 2026
- 2026A Multimodal Deep Learning Framework for Predicting ICU Deterioration: Integrating ECG Waveforms with Clinical Data and Clinician Benchmarking10 Jan 2026
- 2025Missing Data Multiple Imputation for Tabular Q-Learning in Online RL12 Oct 2025
- 2025Missing Data in Signal Processing and Machine Learning: Models, Methods and Modern Approaches2 Jun 2025
- 2025Optimal Transport with Heterogeneously Missing Data22 May 2025
- 2025Imputation-free Learning of Tabular Data with Missing Values using Incremental Feature Partitions in Transformer20 Apr 2025