Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Multivariate Standardized Residuals for Conformal Prediction

About

While split conformal prediction guarantees marginal coverage, approaching the stronger property of conditional coverage is essential for reliable uncertainty quantification. Naive conformal scores, however, suffer from poor conditional coverage in heteroskedastic settings. In univariate regression, this is commonly addressed by normalizing non-conformity scores using an estimated local score variance. In this work, we propose a natural extension of this normalization to the multivariate setting, effectively whitening the residuals to decouple output correlations and standardize local variance. Furthermore, we derive a sufficient condition characterizing a broad class of distributions for which standardized residuals yield asymptotic conditional coverage. We demonstrate that using the Mahalanobis distance induced by a learned local covariance as a non-conformity score provides a closed-form, computationally efficient mechanism for capturing inter-output correlations and heteroskedasticity, avoiding the expensive sampling required by previous methods based on cumulative distribution functions. This structure unlocks several practical extensions, including the handling of missing output values, the refinement of conformal sets when partial information is revealed, and the construction of valid conformal sets for transformations of the output. Finally, we provide extensive empirical evidence on both synthetic and real-world datasets showing that our approach yields conformal sets that improve upon the conditional coverage of existing multivariate baselines.

Sacha Braun, Eug\`ene Berta, Michael I. Jordan, Francis Bach• 2025

Related benchmarks

TaskDatasetResultRank
Conformal PredictionBias
Volume1.64
23
Conformal PredictionHouse
Volume0.0234
23
Conformal PredictionCASP
Volume2.33
23
Conformal PredictionRF2
Volume42
22
Conformal PredictionRF1
Volume32
22
Conditional Coverage for Partially Revealed Outputstaxi
ERT (%)2.09
11
Conditional Coverage for Partially Revealed OutputsHouse
ERT1.8
11
Conditional Coverage for Partially Revealed OutputsCASP
ERT2.31
11
Conformal Predictiontaxi
Volume4.49
11
Conditional Coverage for Partially Revealed OutputsRF1
ERT (%)1.82
10
Showing 10 of 26 rows

Other info

Follow for update