Missingness-Aware Conformal Prediction Under Cross-Hospital Distribution Shift
Not on the map. Why?
Abstract (arXiv)
Clinical measurements are recorded for some patients but not others, at rates that differ across hospitals, and marginal conformal coverage does not ensure coverage within groups defined by missingness. We propose a missingness-aware conformal calibration procedure for mortality prediction under cross-hospital distribution shift. It selects a measurement on an independent sample, groups patients by whether that measurement is recorded, and applies Mondrian calibration within each group, so no calibration outcome is reused. We evaluate the procedure across hospitals in eICU and across care units within one MIMIC-IV hospital, using three predictors. Relative to pooled calibration, it reduces the average worst-group coverage gap on its selected groups in all six settings, with a median reduction of 1.9 percentage points; paired site-bootstrap intervals exclude zero in five. These gains do not extend uniformly. Calibration by predicted risk achieves smaller gaps on a broader panel of missingness groups, and when eICU hospitals are evaluated separately, the gain shrinks for all three predictors and reverses in sign for one. We explain this discrepancy with a hospital-level decomposition. Pooling reweights hospitals through a covariance between group shares and coverage errors, and lets errors of opposite sign cancel: weighting explains the reversal, and cancellation accounts for most of the attenuation for the other two predictors. Constructed population distributions show that pooled and within-hospital evaluations can rank calibration methods oppositely even without sampling noise. Pooled improvement alone therefore cannot establish better coverage within hospitals, even when the calibration groups are fixed.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 33 references that are in the map. First posted on arXiv 25 Sep 2026.
Builds on
19 of 33 references.
References · 19
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2016Distribution-Free Predictive Inference For Regressioncited by 668
- 2019Conformalized Quantile Regressioncited by 629
- 2016Recurrent Neural Networks for Multivariate Time Series with Missing Valuescited by 554
- 2019Conformal Prediction Under Covariate Shiftcited by 519
- 2020Classification with Valid and Adaptive Coveragecited by 396
- 2021Adaptive Conformal Inference Under Distribution Shiftcited by 375
- 2020Uncertainty Sets for Image Classifiers using Conformal Predictioncited by 366
- 2022Conformal prediction beyond exchangeabilitycited by 355
- 2016Least Ambiguous Set-Valued Classifiers with Bounded Error Levelscited by 272
- 2023Conformal Prediction With Conditional Guaranteescited by 152
- 2021Localized Conformal Prediction: A Generalized Inference Framework for Conformal Predictioncited by 114
7 more
- 2023Conformal Prediction with Missing Valuescited by 25
- 2023Distribution-free inference with hierarchical datacited by 15
- 2024On Temperature Scaling and Conformal Prediction of Deep Classifierscited by 13
- 2025Conditional Coverage Diagnostics for Conformal Predictioncited by 8
- 2025CLEAR: Calibrated Learning for Epistemic and Aleatoric Riskcited by 6
- 2026Conformal Prediction Assessment: A Framework for Conditional Coverage Evaluation and Selectioncited by 2
- 2026Conformal Calibration for Multi-Modal Regression with Missing Modalitiescited by 1
Built on by
Not cited yet.