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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Conformal Prediction | CIFAR-10 (test) | -- | 21 | |
| Regression | OpenML 361247 | Coverage96 | 12 | |
| Regression | OpenML 361235 | Coverage93.6 | 12 | |
| Regression | OpenML 361249 | Coverage95.2 | 12 | |
| Regression | OpenML 361243 | Coverage93.6 | 12 | |
| Regression | OpenML 361244 | Coverage94.6 | 12 | |
| Conformal Prediction | MNIST | Coverage (alpha=0.025)97.4 | 11 | |
| Regression | OpenML dataset 361242 (N=21263, d=81) | Coverage95.2 | 6 | |
| Regression | OpenML dataset 361234 | Coverage94.5 | 6 | |
| Regression | OpenML 361242 | Coverage94.8 | 6 |