Beyond Accuracy: An Empirical Study of Uncertainty Estimation in Imputation
Not on the map. Why?
Abstract (arXiv)
Handling missing data is a central challenge in data-driven analysis. Modern imputation methods not only aim for accurate reconstruction but also differ in how they represent and quantify uncertainty. Yet, the reliability and calibration of these uncertainty estimates remain poorly understood. This paper presents a systematic empirical study of uncertainty in imputation, comparing representative methods from three major families: statistical (MICE, SoftImpute), distribution alignment (OT-Impute), and deep generative (GAIN, MIWAE, TabCSDI). Experiments span multiple datasets, missingness mechanisms (MCAR, MAR, MNAR), and missingness rates. Uncertainty is estimated through three complementary routes: multi-run variability, conditional sampling, and predictive-distribution modeling, and evaluated using calibration curves and the Expected Calibration Error (ECE). Results show that accuracy and calibration are often misaligned: models with high reconstruction accuracy do not necessarily yield reliable uncertainty. We analyze method-specific trade-offs among accuracy, calibration, and runtime, identify stable configurations, and offer guidelines for selecting uncertainty-aware imputers in data cleaning and downstream machine learning pipelines.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 17 of its 46 references that are in the map. First posted on arXiv 26 Nov 2025.
Builds on
17 of 46 references.
References · 17
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,277
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 777
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 577
- 2015Importance Weighted Autoencoderscited by 454
- 2022TabDDPM: Modelling Tabular Data with Diffusion Modelscited by 373
- 2018GAIN: Missing Data Imputation using Generative Adversarial Netscited by 349
- 2022Quantifying Aleatoric and Epistemic Uncertainty in Machine Learning: Are Conditional Entropy and Mutual Information Appropriate Measures?cited by 124
- 2020Missing Data Imputation using Optimal Transportcited by 99
- 2022Diffusion models for missing value imputation in tabular datacited by 77
- 2018Variational Autoencoder with Arbitrary Conditioningcited by 68
5 more
- 2024Rethinking Aleatoric and Epistemic Uncertaintycited by 40
- 2020Uncertainty-Aware Variational-Recurrent Imputation Network for Clinical Time Seriescited by 33
- 2024OTClean: Data Cleaning for Conditional Independence Violations using Optimal Transportcited by 8
- 2024Threshold-Independent Fair Matching through Score Calibrationcited by 4
- 2022Leveraging variational autoencoders for multiple data imputationcited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.