Conformal and kNN Predictive Uncertainty Quantification Algorithms in Metric Spaces

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lugosi, Gábor, Matabuena, Marcos
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912494913585152
author Lugosi, Gábor
Matabuena, Marcos
author_facet Lugosi, Gábor
Matabuena, Marcos
contents This paper introduces a framework for uncertainty quantification in regression models defined in metric spaces. Leveraging a newly defined notion of homoscedasticity, we develop a conformal prediction algorithm that offers finite-sample coverage guarantees and fast convergence rates of the oracle estimator. In heteroscedastic settings, we forgo these non-asymptotic guarantees to gain statistical efficiency, proposing a local $k$--nearest--neighbor method without conformal calibration that is adaptive to the geometry of each particular nonlinear space. Both procedures work with any regression algorithm and are scalable to large data sets, allowing practitioners to plug in their preferred models and incorporate domain expertise. We prove consistency for the proposed estimators under minimal conditions. Finally, we demonstrate the practical utility of our approach in personalized--medicine applications involving random response objects such as probability distributions and graph Laplacians.
format Preprint
id arxiv_https___arxiv_org_abs_2507_15741
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Conformal and kNN Predictive Uncertainty Quantification Algorithms in Metric Spaces
Lugosi, Gábor
Matabuena, Marcos
Machine Learning
Statistics Theory
Methodology
This paper introduces a framework for uncertainty quantification in regression models defined in metric spaces. Leveraging a newly defined notion of homoscedasticity, we develop a conformal prediction algorithm that offers finite-sample coverage guarantees and fast convergence rates of the oracle estimator. In heteroscedastic settings, we forgo these non-asymptotic guarantees to gain statistical efficiency, proposing a local $k$--nearest--neighbor method without conformal calibration that is adaptive to the geometry of each particular nonlinear space. Both procedures work with any regression algorithm and are scalable to large data sets, allowing practitioners to plug in their preferred models and incorporate domain expertise. We prove consistency for the proposed estimators under minimal conditions. Finally, we demonstrate the practical utility of our approach in personalized--medicine applications involving random response objects such as probability distributions and graph Laplacians.
title Conformal and kNN Predictive Uncertainty Quantification Algorithms in Metric Spaces
topic Machine Learning
Statistics Theory
Methodology
url https://arxiv.org/abs/2507.15741