Label Noise Robustness of Conformal Prediction

Bat-Sheva Einbinder,Shai Feldman,Stephen Bates,Anastasios N. Angelopoulos,Asaf Gendler,Yaniv Romano
2024-11-26
Abstract:We study the robustness of conformal prediction, a powerful tool for uncertainty quantification, to label noise. Our analysis tackles both regression and classification problems, characterizing when and how it is possible to construct uncertainty sets that correctly cover the unobserved noiseless ground truth labels. We further extend our theory and formulate the requirements for correctly controlling a general loss function, such as the false negative proportion, with noisy labels. Our theory and experiments suggest that conformal prediction and risk-controlling techniques with noisy labels attain conservative risk over the clean ground truth labels whenever the noise is dispersive and increases variability. In other adversarial cases, we can also correct for noise of bounded size in the conformal prediction algorithm in order to ensure achieving the correct risk of the ground truth labels without score or data regularity.
Machine Learning,Artificial Intelligence,Statistics Theory,Methodology
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to ensure the effectiveness and robustness of conformal prediction in the presence of label noise. Specifically, the paper focuses on how to construct uncertainty sets that can cover unobserved noiseless true labels and control the risk of general loss functions when the labels in the training data are affected by noise in regression and classification tasks. Through theoretical analysis and experimental verification, the paper explores the performance of conformal prediction methods under different types of label noise and their impact on risk - control techniques. In addition, the paper also proposes some adjustment strategies to deal with noise of a specific magnitude, ensuring that the prediction sets reach the correct risk level when facing clean true labels.