Comparison of Imputation Strategies for Incomplete Electronic Health Data
Abstract
Abstract Missing data is a persistent challenge in electronic health records (EHRs), often compromising data integrity and limiting the effectiveness of predictive models in healthcare. This study systematically evaluates five widely used imputation strategies—GAIN, MICE, Median, MissForest, and MIWAE—across three real-world clinical datasets under varying missingness mechanisms (MCAR, MAR, and MNAR) and missingness rates (10%–90%). We assessed imputation quality using multiple statistical measures and examined the relationship between imputation accuracy and downstream classification performance. Our results show that MICE and MissForest consistently outperform other methods across most scenarios, while deep learning-based approaches such as GAIN exhibit high instability under MAR and MNAR, particularly at higher missingness levels. Furthermore, imputation quality does not always align with classification performance, underscoring the need to consider task-specific goals when selecting imputation strategies. We also provide a practical framework summarizing method recommendations based on missingness type and rate, aiming to support robust data preprocessing decisions in clinical AI applications.
Fields of Study & Subjects
Machine Learning in Healthcare
Imbalanced Data Classification Techniques
Artificial Intelligence in Healthcare
Publication details
- Journal / source
- Health Data Science
- Publication date
- 24 June 2026
- Language
- EN