MLLGMEMay 18

Conformal Prediction via Transported Beta Laws

arXiv:2605.1902434.2
Predicted impact top 50% in ML · last 90 daysOriginality Incremental advance
AI Analysis

Provides a theoretical framework to assess calibration-conditional coverage in conformal prediction, useful for practitioners needing guarantees beyond marginal coverage.

The paper studies the calibration-conditional coverage distribution of split conformal prediction, showing it follows a Beta law under i.i.d. data and quantifying deviations from this reference via Wasserstein distances. The framework provides bounds on marginal coverage gaps and bad-calibration probabilities, with explicit characterizations for non-i.i.d. settings.

Split conformal prediction provides finite-sample marginal coverage under exchangeability, but this guarantee averages over the random calibration sample. We study instead the law of the calibration-conditional coverage induced by a realized conformal threshold. In the continuous i.i.d. setting this law is exactly $Beta(k,n+1-k)$, so the usual marginal guarantee corresponds to its mean. We take this beta law as a finite-sample reference object and quantify departures from it using Wasserstein distances on $[0,1]$. The framework yields direct bounds on marginal coverage gaps and on bad-calibration probabilities, and separates different sources of non-i.i.d. behavior according to how they deform the beta reference: test-side shift acts through a transport map on the coverage scale, while calibration dependence changes the order-statistic law itself. We instantiate the framework in scale-shift, clustered, and stationary mixing settings, where the induced deformations can be characterized explicitly or through Berry-Esseen approximations. Simulations on dependent processes confirm that the first-order approximation tracks the empirical Wasserstein distance even at moderate sample sizes.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes