English

Generative Conformal Prediction with Vectorized Non-Conformity Scores

Machine Learning 2025-02-12 v2 Methodology

Abstract

Conformal prediction (CP) provides model-agnostic uncertainty quantification with guaranteed coverage, but conventional methods often produce overly conservative uncertainty sets, especially in multi-dimensional settings. This limitation arises from simplistic non-conformity scores that rely solely on prediction error, failing to capture the prediction error distribution's complexity. To address this, we propose a generative conformal prediction framework with vectorized non-conformity scores, leveraging a generative model to sample multiple predictions from the fitted data distribution. By computing non-conformity scores across these samples and estimating empirical quantiles at different density levels, we construct adaptive uncertainty sets using density-ranked uncertainty balls. This approach enables more precise uncertainty allocation -- yielding larger prediction sets in high-confidence regions and smaller or excluded sets in low-confidence regions -- enhancing both flexibility and efficiency. We establish theoretical guarantees for statistical validity and demonstrate through extensive numerical experiments that our method outperforms state-of-the-art techniques on synthetic and real-world datasets.

Keywords

Cite

@article{arxiv.2410.13735,
  title  = {Generative Conformal Prediction with Vectorized Non-Conformity Scores},
  author = {Minxing Zheng and Shixiang Zhu},
  journal= {arXiv preprint arXiv:2410.13735},
  year   = {2025}
}
R2 v1 2026-06-28T19:26:09.434Z