Efficient Conformal Prediction for Regression Models under Label Noise

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Cohen, Yahav, Goldberger, Jacob, Tirer, Tom
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912592768794624
author Cohen, Yahav
Goldberger, Jacob
Tirer, Tom
author_facet Cohen, Yahav
Goldberger, Jacob
Tirer, Tom
contents In high-stakes scenarios, such as medical imaging applications, it is critical to equip the predictions of a regression model with reliable confidence intervals. Recently, Conformal Prediction (CP) has emerged as a powerful statistical framework that, based on a labeled calibration set, generates intervals that include the true labels with a pre-specified probability. In this paper, we address the problem of applying CP for regression models when the calibration set contains noisy labels. We begin by establishing a mathematically grounded procedure for estimating the noise-free CP threshold. Then, we turn it into a practical algorithm that overcomes the challenges arising from the continuous nature of the regression problem. We evaluate the proposed method on two medical imaging regression datasets with Gaussian label noise. Our method significantly outperforms the existing alternative, achieving performance close to the clean-label setting.
format Preprint
id arxiv_https___arxiv_org_abs_2509_15120
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Efficient Conformal Prediction for Regression Models under Label Noise
Cohen, Yahav
Goldberger, Jacob
Tirer, Tom
Machine Learning
In high-stakes scenarios, such as medical imaging applications, it is critical to equip the predictions of a regression model with reliable confidence intervals. Recently, Conformal Prediction (CP) has emerged as a powerful statistical framework that, based on a labeled calibration set, generates intervals that include the true labels with a pre-specified probability. In this paper, we address the problem of applying CP for regression models when the calibration set contains noisy labels. We begin by establishing a mathematically grounded procedure for estimating the noise-free CP threshold. Then, we turn it into a practical algorithm that overcomes the challenges arising from the continuous nature of the regression problem. We evaluate the proposed method on two medical imaging regression datasets with Gaussian label noise. Our method significantly outperforms the existing alternative, achieving performance close to the clean-label setting.
title Efficient Conformal Prediction for Regression Models under Label Noise
topic Machine Learning
url https://arxiv.org/abs/2509.15120