Pré-Publication, Document De Travail Année : 2025

Optimal Transport-based Conformal Prediction

Résumé

Conformal Prediction (CP) is a principled framework for quantifying uncertainty in blackbox learning models, by constructing prediction sets with finite-sample coverage guarantees. Traditional approaches rely on scalar nonconformity scores, which fail to fully exploit the geometric structure of multivariate outputs, such as in multi-output regression or multiclass classification. Recent methods addressing this limitation impose predefined convex shapes for the prediction sets, potentially misaligning with the intrinsic data geometry. We introduce a novel CP procedure handling multivariate score functions through the lens of optimal transport. Specifically, we leverage Monge-Kantorovich vector ranks and quantiles to construct prediction region with flexible, potentially non-convex shapes, better suited to the complex uncertainty patterns encountered in multivariate learning tasks. We prove that our approach ensures finite-sample, distribution-free coverage properties, similar to typical CP methods. We then adapt our method for multi-output regression and multiclass classification, and also propose simple adjustments to generate adaptive prediction regions with asymptotic conditional coverage guarantees. Finally, we evaluate our method on practical regression and classification problems, illustrating its advantages in terms of (conditional) coverage and efficiency.
Fichier principal
Vignette du fichier
preprint.pdf (885) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04922368 , version 1 (30-01-2025)

Licence

Identifiants

  • HAL Id : hal-04922368 , version 1

Citer

Gauthier Thurin, Kimia Nadjahi, Claire Boyer. Optimal Transport-based Conformal Prediction. 2025. ⟨hal-04922368⟩
0 Consultations
0 Téléchargements

Partager

More