Score: 1

Minimum Volume Conformal Sets for Multivariate Regression

Published: March 24, 2025 | arXiv ID: 2503.19068v1

By: Sacha Braun , Liviu Aolaritei , Michael I. Jordan and more

BigTech Affiliations: University of California, Berkeley

Potential Business Impact:

Makes computer guesses more honest and useful.

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.

Country of Origin
🇺🇸 United States

Page Count
34 pages

Category
Statistics:
Machine Learning (Stat)