2501.12749

Total: 1

#1 Conformal Prediction of Classifiers with Many Classes based on Noisy Labels [PDF1] [Copy] [Kimi] [REL]

Authors: Coby Penso, Jacob Goldberger, Ethan Fetaya

Conformal Prediction (CP) controls the prediction uncertainty of classification systems by producing a small prediction set, ensuring a predetermined probability that the true class lies within this set. This is commonly done by defining a score, based on the model predictions, and setting a threshold on this score using a validation set. In this study, we address the problem of CP calibration when we only have access to a calibration set with noisy labels. We show how we can estimate the noise-free conformal threshold based on the noisy labeled data. We derive a finite sample coverage guarantee for uniform noise that remains effective even in tasks with a large number of classes. We dub our approach Noise-Aware Conformal Prediction (NACP). We illustrate the performance of the proposed results on several standard image classification datasets with a large number of classes.

Subjects: Machine Learning , Artificial Intelligence , Machine Learning

Publish: 2025-01-22 09:35:58 UTC