mlsanityData to 14 Oct 2026

Adaptive Conformal Prediction via Bayesian Uncertainty Weighting for Hierarchical Healthcare Data

arXiv 2601.01223 ↗First posted 3 Jan 2026cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Clinical decision-making demands uncertainty quantification that provides both distribution-free coverage guarantees and risk-adaptive precision, requirements that existing methods fail to jointly satisfy. We present a hybrid Bayesian-conformal framework that addresses this fundamental limitation in healthcare predictions. Our approach integrates Bayesian hierarchical random forests with group-aware conformal calibration, using posterior uncertainties to weight conformity scores while maintaining rigorous coverage validity. Evaluated on 61,538 admissions across 3,793 U.S. hospitals and 4 regions, our method achieves target coverage (94.3% vs 95% target) with adaptive precision: 21% narrower intervals for low-uncertainty cases while appropriately widening for high-risk predictions. Critically, we demonstrate that well-calibrated Bayesian uncertainties alone severely under-cover (14.1%), highlighting the necessity of our hybrid approach. This framework enables risk-stratified clinical protocols, efficient resource planning for high-confidence predictions, and conservative allocation with enhanced oversight for uncertain cases, providing uncertainty-aware decision support across diverse healthcare settings.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 62 references that are in the map. First posted on arXiv 3 Jan 2026.

Builds on

15 of 62 references.

References · 15

  1. 2017On Calibration of Modern Neural Networkscited by 4,960
  2. 2016Concrete Problems in AI Safetycited by 1,642
  3. 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
  4. 2016Distribution-Free Predictive Inference For Regressioncited by 668
  5. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
  6. 2019Conformal Prediction Under Covariate Shiftcited by 519
  7. 2018Scalable and accurate deep learning for electronic health recordscited by 280
  8. 2017A Conceptual Introduction to Hamiltonian Monte Carlocited by 209
  9. 2015Practical Bayesian model evaluation using leave-one-out cross-validation and WAICcited by 110
  10. 2021Conformal Bayesian Computationcited by 39
  11. 2018Distribution-Free Prediction Sets for Two-Layer Hierarchical Modelscited by 35
  12. 2022A review of predictive uncertainty estimation with machine learningcited by 29
3 more
  1. 2016Engineering Safety in Machine Learningcited by 24
  2. 2023Bayesian Optimization with Formal Safety Guarantees via Online Conformal Predictioncited by 10
  3. 2025Conformalized Generative Bayesian Imaging: An Uncertainty Quantification Framework for Computational Imagingcited by 5

Built on by

Not cited yet.