Back to AI News
arxiv

Conformal Prediction Framework for Distribution-to-Distribution Regression

Researchers propose a new conformal prediction framework for distribution-to-distribution regression, a setting where both inputs and outputs are probability distributions. They show that naive Split Conformal Prediction produces prediction sets that are marginally valid but lack interpretable volume or tractable sampling. Their approach operates in an orthogonal basis space to yield interpretable prediction sets while preserving coverage guarantees, with an adaptive variant giving asymptotic conditional coverage. Empirical results on synthetic and real-world datasets validate the method, and code is publicly released.

AI Brief

It offers a concrete, code-backed method for uncertainty quantification in an emerging regression setting, useful to researchers working on reliable prediction with distribution-valued data.

arXiv:2611.06320v1 Announce Type: cross Abstract: Conformal Prediction (CP) provides a model-agnostic framework for constructing finite-sample prediction sets with coverage guarantees and has become a tool for reliable decision-making in high-risk settings. Recent advances have extended Split Conformal Prediction to increasingly complex learning problems, including multilabel, high-dimensional, and functional outputs. In this work, we study CP for Distribution-to-Distribution Regression, a setting of growing interest across many application domains. We show that a naive application of Split CP yields prediction sets that, while marginally valid, are non-interpretable, e.g., lacking a well-defined volume or a tractable sampling procedure, and of limited practical value. To overcome this, we propose a new conformal framework that operates in an orthogonal basis space, producing interpretable prediction sets while preserving coverage guarantees. We further develop an adaptive variant with asymptotic conditional coverage under mild assumptions. Empirical results on synthetic and real-world datasets validate the effectiveness of the proposed approach. The code is publicly available at https://github.com/tuanhacs/conformal_d2d.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!