Minimum Volume Conformal Sets for Multivariate Regression
About
Conformal prediction provides a principled framework for constructing predictive sets with finite-sample validity. While much of the focus has been on univariate response variables, existing multivariate methods either impose rigid geometric assumptions or rely on flexible but computationally expensive approaches that do not explicitly optimize prediction set volume. We propose an optimization-driven framework based on a novel loss function that directly learns minimum-volume covering sets while ensuring valid coverage. This formulation naturally induces a new nonconformity score for conformal prediction, which adapts to the residual distribution and covariates. Our approach optimizes over prediction sets defined by arbitrary norm balls, including single and multi-norm formulations. Additionally, by jointly optimizing both the predictive model and predictive uncertainty, we obtain prediction sets that are tight, informative, and computationally efficient, as demonstrated in our experiments on real-world datasets.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multivariate Regression | Energy | Coverage99.4 | 8 | |
| Multivariate Regression | taxi | Coverage98.9 | 8 | |
| Multivariate Regression | House | Normalized Volume1.33 | 4 | |
| Multivariate Regression | Exp. Transformed (test) | Coverage90 | 4 | |
| Multivariate Regression | Gau. Fixed | Normalized Volume10.89 | 4 | |
| Multivariate Regression | SCM1d | Coverage (%)91 | 4 | |
| Multivariate Regression Uncertainty Quantification | Bias correction | Coverage99.4 | 4 | |
| Multivariate Regression Uncertainty Quantification | RF2 | Coverage99.2 | 4 | |
| Multivariate Uncertainty Quantification | Energy | Normalized Volume2.85 | 4 | |
| Multivariate Uncertainty Quantification | SCM20d | Normalized Volume4.25 | 4 |