arXiv AI

Stochastic Grouping Conformal Prediction for Effective Subgroup Reliability

Stochastic Grouping Conformal Prediction (SGCP) is a new framework that provides distribution‑free coverage guarantees while addressing coverage disparities across clinically important subgroups. SGCP learns a stochastic grouping map so that each sample can draw calibration data from others with similar calibration behavior, creating a local score law that improves reliability for all subpopulations. The authors prove that SGCP maintains the standard coverage guarantee and demonstrate through synthetic and real‑world experiments that it reduces subgroup coverage gaps while keeping prediction set sizes small or comparable to existing methods.

arXiv Machine Learning
Jul 21

Isotonic Conformal Prediction

arXiv:2607. 16675v1 Announce Type: cross Abstract: A point prediction that is well calibrated on average can still be systematically biased conditional on its own value, undermining its use in downstream decision-making.

By Daniel Bensimon, Sean Xiang Yu, Eric D. Kolaczyk, Archer Y. Yang
arXiv Statistics ML
Oct 27

Isotonic Conformal Inference

arXiv:2607.16675v2 Announce Type: replace Abstract: A predictor with small overall bias can still exhibit systematic bias conditional on its predicted value, limiting its reliability for downstream u...

By Daniel Bensimon, Zhiquan Liu, Sean X. Yu, Eric D. Kolaczyk, Archer Y. Yang
arXiv Machine Learning
Sep 29

Missingness-Aware Conformal Prediction Under Cross-Hospital Distribution Shift

The paper introduces a missingness‑aware conformal calibration method for mortality prediction that accounts for cross‑hospital distribution shifts. By selecting a measurement on an independent sample, grouping patients by whether that measurement is recorded, and applying Mondrian calibration within each group, the method avoids reusing calibration outcomes. Experiments on eICU and MIMIC‑IV data show that, compared to pooled calibration, it reduces the worst‑group coverage gap by a median of 1.9 percentage points across six settings, though the benefit varies with predictor and hospital.

By Liang You, Dongwen Ou, Hengyu Shi, Siyuan Dai
arXiv Machine Learning
Sep 15

Noise-Adaptive Conformal Classification with Marginal Coverage

arXiv:2501.18060v2 Announce Type: replace-cross Abstract: Conformal inference provides a rigorous statistical framework for uncertainty quantification in machine learning, enabling well-calibrated pr...

By Teresa Bortolotti, Y. X. Rachel Wang, Xin Tong, Alessandra Menafoglio, Simone Vantini, Matteo Sesia
arXiv AI
Oct 28

When uncertainty quantification fails: structural vulnerabilities and corrections for selective prediction under clinical class imbalance

The paper investigates how uncertainty quantification fails in selective prediction systems when clinical data exhibit severe class imbalance, leading to systematic exclusion of minority patients. Through experiments on 56 synthetic scenarios and three neurodegenerative disease cohorts, the authors show that standard uncertainty measures become proxies for class membership rather than true ambiguity, and that combining class‑aware correction (CCR) with threshold‑aware metrics restores proper monotonicity and protects minority patients. The study proposes adopting threshold‑aligned uncertainty measures and class‑stratified evaluation to mitigate this bias and improve safety in clinical AI pipelines.

By Walter Endrizzi, Flavio Ragni, Raffaele Marchesi, Stefano Bovo, Monica Moroni, Giuseppe Jurman, Venet Osmani