Conformal Prediction with Missing Values - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

Conformal Prediction with Missing Values

Résumé

Conformal prediction is a theoretically grounded framework for constructing predictive intervals. We study conformal prediction with missing values in the covariates -- a setting that brings new challenges to uncertainty quantification. We first show that the marginal coverage guarantee of conformal prediction holds on imputed data for any missingness distribution and almost all imputation functions. However, we emphasize that the average coverage varies depending on the pattern of missing values: conformal methods tend to construct prediction intervals that undercover the response conditionally to some missing patterns. This motivates our novel generalized conformalized quantile regression framework, missing data augmentation, which yields prediction intervals that are valid conditionally to the patterns of missing values, despite their exponential number. We then show that a universally consistent quantile regression algorithm trained on the imputed data is Bayes optimal for the pinball risk, thus achieving valid coverage conditionally to any given data point. Moreover, we examine the case of a linear model, which demonstrates the importance of our proposal in overcoming the heteroskedasticity induced by missing values. Using synthetic and data from critical care, we corroborate our theory and report improved performance of our methods.
Fichier principal
Vignette du fichier
zaffran23a.pdf (1.49 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03896384 , version 1 (13-12-2022)
hal-03896384 , version 2 (07-03-2023)
hal-03896384 , version 3 (06-06-2023)
hal-03896384 , version 4 (09-11-2023)

Identifiants

  • HAL Id : hal-03896384 , version 4

Citer

Margaux Zaffran, Aymeric Dieuleveut, Julie Josse, Yaniv Romano. Conformal Prediction with Missing Values. ICML 2023 - 40 th International Conference on Machine Learning, Jul 2023, Honolulu (Hawai), United States. pp.40578. ⟨hal-03896384v4⟩
417 Consultations
323 Téléchargements

Partager

More