Performance of Conformal Prediction in Capturing Aleatoric Uncertainty
By: Misgina Tsighe Hagos, Claes Lundström
Potential Business Impact:
Shows how sure a computer is about its guesses.
Conformal prediction is a model-agnostic approach to generating prediction sets that cover the true class with a high probability. Although its prediction set size is expected to capture aleatoric uncertainty, there is a lack of evidence regarding its effectiveness. The literature presents that prediction set size can upper-bound aleatoric uncertainty or that prediction sets are larger for difficult instances and smaller for easy ones, but a validation of this attribute of conformal predictors is missing. This work investigates how effectively conformal predictors quantify aleatoric uncertainty, specifically the inherent ambiguity in datasets caused by overlapping classes. We perform this by measuring the correlation between prediction set sizes and the number of distinct labels assigned by human annotators per instance. We further assess the similarity between prediction sets and human-provided annotations. We use three conformal prediction approaches to generate prediction sets for eight deep learning models trained on four datasets. The datasets contain annotations from multiple human annotators (ranging from five to fifty participants) per instance, enabling the identification of class overlap. We show that the vast majority of the conformal prediction outputs show a very weak to weak correlation with human annotations, with only a few showing moderate correlation. These findings underscore the necessity of critically reassessing the prediction sets generated using conformal predictors. While they can provide a higher coverage of the true classes, their capability in capturing aleatoric uncertainty remains limited.
Similar Papers
Conformal Prediction and Human Decision Making
Machine Learning (CS)
Helps AI make better guesses for people.
Validation of Conformal Prediction in Cervical Atypia Classification
Image and Video Processing
Helps doctors trust AI for cancer screening.
Aerial Image Classification in Scarce and Unconstrained Environments via Conformal Prediction
Machine Learning (CS)
Shows how sure a computer is about its guesses.