mlsanityData to 14 Oct 2026

No imputation without representation

arXiv 2206.14254 ↗First posted 28 Jun 2022cs.LG stat.MLcited by 2

Not on the map. Why?

Abstract (arXiv)

By filling in missing values in datasets, imputation allows these datasets to be used with algorithms that cannot handle missing values by themselves. However, missing values may in principle contribute useful information that is lost through imputation. The missing-indicator approach can be used in combination with imputation to instead represent this information as a part of the dataset. There are several theoretical considerations why missing-indicators may or may not be beneficial, but there has not been any large-scale practical experiment on real-life datasets to test this question for machine learning predictions. We perform this experiment for three imputation strategies and a range of different classification algorithms, on the basis of twenty real-life datasets. In a follow-up experiment, we determine attribute-specific missingness thresholds for each classifier above which missing-indicators are more likely than not to increase classification performance. And in a second follow-up experiment, we evaluate numerical imputation of one-hot encoded categorical attributes. We reach the following conclusions. Firstly, missing-indicators generally increase classification performance. Secondly, with missing-indicators, nearest neighbour and iterative imputation do not lead to better performance than simple mean/mode imputation. Thirdly, for decision trees, pruning is necessary to prevent overfitting. Fourthly, the thresholds above which missing-indicators are more likely than not to improve performance are lower for categorical attributes than for numerical attributes. Lastly, mean imputation of numerical attributes preserves some of the information from missing values. Consequently, when not using missing-indicators it can be advantageous to apply mean imputation to one-hot encoded categorical attributes instead of mode imputation.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 6 of its 89 references that are in the map. First posted on arXiv 28 Jun 2022.

Builds on

6 of 89 references.

References · 6

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2021What's a good imputation to predict with missing values?cited by 62
  3. 2019On the consistency of supervised learning with missing valuescited by 53
  4. 2018Processing of missing data by neural networkscited by 37
  5. 2022Benchmarking missing-values approaches for predictive models on health databasescited by 23
  6. 2020Trajectories, bifurcations and pseudotime in large clinical datasets: applications to myocardial infarction and diabetes datacited by 13

Built on by

Cited by year

Citing papers per year: 2022: 1; 2023: 0; 2024: 1.

2 papers.