Score: 0

Volume-Sorted Prediction Set: Efficient Conformal Prediction for Multi-Target Regression

Published: March 4, 2025 | arXiv ID: 2503.02205v1

By: Rui Luo, Zhixin Zhou

Potential Business Impact:

Shows how sure a computer is about its guesses.

Business Areas:
A/B Testing Data and Analytics

We introduce Volume-Sorted Prediction Set (VSPS), a novel method for uncertainty quantification in multi-target regression that uses conditional normalizing flows with conformal calibration. This approach constructs flexible, non-convex predictive regions with guaranteed coverage probabilities, overcoming limitations of traditional methods. By learning a transformation where the conditional distribution of responses follows a known form, VSPS identifies dense regions in the original space using the Jacobian determinant. This enables the creation of prediction regions that adapt to the true underlying distribution, focusing on areas of high probability density. Experimental results demonstrate that VSPS produces smaller, more informative prediction regions while maintaining robust coverage guarantees, enhancing uncertainty modeling in complex, high-dimensional settings.

Page Count
23 pages

Category
Computer Science:
Machine Learning (CS)