Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Conformal Prediction for Ensembles: Improving Efficiency via Score-Based Aggregation

About

Distribution-free uncertainty estimation for ensemble methods is increasingly desirable due to the widening deployment of multi-modal black-box predictive models. Conformal prediction is one approach that avoids such distributional assumptions. Methods for conformal aggregation have in turn been proposed for ensembled prediction, where the prediction regions of individual models are merged as to retain coverage guarantees while minimizing conservatism. Merging the prediction regions directly, however, sacrifices structures present in the conformal scores that can further reduce conservatism. We, therefore, propose a novel framework that extends the standard scalar formulation of a score function to a multivariate score that produces more efficient prediction regions. We then demonstrate that such a framework can be efficiently leveraged in both classification and predict-then-optimize regression settings downstream and empirically show the advantage over alternate conformal aggregation methods.

Eduardo Ochoa Rivera, Yash Patel, Ambuj Tewari• 2024

Related benchmarks

TaskDatasetResultRank
Conformal PredictionCIFAR-10 (test)--
21
RegressionOpenML 361247
Coverage96
12
RegressionOpenML 361235
Coverage93.6
12
RegressionOpenML 361249
Coverage95.2
12
RegressionOpenML 361243
Coverage93.6
12
RegressionOpenML 361244
Coverage94.6
12
Conformal PredictionMNIST
Coverage (alpha=0.025)97.4
11
RegressionOpenML dataset 361242 (N=21263, d=81)
Coverage95.2
6
RegressionOpenML dataset 361234
Coverage94.5
6
RegressionOpenML 361242
Coverage94.8
6
Showing 10 of 15 rows

Other info

Follow for update