Benchmarking missing-values approaches for predictive models on health databases

Alexandre Perez-Lebel (1, 2, 3, 4) , Gaël Varoquaux (1, 2, 3, 4) , Marine Le Morvan (4) , Julie Josse (5, 6) , Jean-Baptiste Poline (1, 2)

1 MNI - McConnell Brain Imaging Centre
2 McGill University = Université McGill [Montréal, Canada]
3 MILA - Montreal Institute for Learning Algorithms [Montréal]
4 PARIETAL - Modelling brain structure, function and variability based on high-field MRI data
5 IDESP - Institut Desbrest de santé publique
6 PREMEDICAL - Médecine de précision par intégration de données et inférence causale

BACKGROUND: As databases grow larger, it becomes harder to fully control their collection, and they frequently come with missing values: incomplete observations. These large databases are well suited to train machine-learning models, for instance for forecasting or to extract biomarkers in biomedical settings. Such predictive approaches can use discriminative -rather than generative- modeling, and thus open the door to new missing-values strategies. Yet existing empirical evaluations of strategies to handle missing values have focused on inferential statistics. RESULTS: Here we conduct a systematic benchmark of missing-values strategies in predictive models with a focus on large health databases: four electronic health record datasets, a population brain imaging one, a health survey and two intensive care ones. Using gradient-boosted trees, we compare native support for missing values with simple and state-of-the-art imputation prior to learning. We investigate prediction accuracy and computational time. For prediction after imputation, we find that adding an indicator to express which values have been imputed is important, suggesting that the data are missing not at random. Elaborate missing values imputation can improve prediction compared to simple strategies but requires longer computational time on large data. Learning trees that model missing values-with missing incorporated attribute-leads to robust, fast, and well-performing predictive modeling. CONCLUSIONS: Native support for missing values in supervised machine learning predicts better than state-of-the-art imputation with much less computational cost. When using imputation, it is important to add indicator columns expressing which values have been imputed.

Mots clés: Missing values; Machine learning; Supervised learning; Benchmark; Imputation; Multiple imputation; Bagging

Domaines: Machine Learning [stat.ML]; Apprentissage [cs.LG]; Intelligence artificielle [cs.AI]

Alexandre Perez-Lebel, Gaël Varoquaux, Marine Le Morvan, Julie Josse, Jean-Baptiste Poline. Benchmarking missing-values approaches for predictive models on health databases. GigaScience, inPress, ⟨10.1093/gigascience/giac013⟩. ⟨hal-03526292v2⟩