Classification with reject option: Distribution-free error guarantees via conformal prediction

Machine learning (ML) models always make a prediction, even when they are likely to be wrong. This causes problems in practical applications, as we do not know if we should trust a prediction. ML with reject option addresses this issue by abstaining from making a prediction if it is likely to be incorrect. In this work, we formalise the approach to ML with reject option in binary classification, deriving theoretical guarantees on the resulting error rate. This is achieved through conformal prediction (CP), which produce prediction sets with distribution-free validity guarantees. In binary classification, CP can output prediction sets containing exactly one, two or no labels. By accepting only the singleton predictions, we turn CP into a binary classifier with reject option. Here, CP is formally put in the framework of predicting with reject option. We state and prove the resulting error rate, and give finite sample estimates. Numerical examples provide illustrations of derived error rate through several different conformal prediction settings, ranging from full conformal prediction to offline batch inductive conformal prediction. The former has a direct link to sharp validity guarantees, whereas the latter is more fuzzy in terms of validity guarantees but can be used in practice. Error-reject curves illustrate the trade-off between error rate and reject rate, and can serve to aid a user to set an acceptable error rate or reject rate in practice. • Proven Distribution-Free Error Rates for Singleton Predictions: Delivered theoretical guarantees on error rates for accepted singleton predictions, resolving inaccuracies found in prior works. • Refinement of Reject Option in Conformal Prediction: Clarified the relationship between error rate and reject rate using error-reject curves, offering a more accurate approach than existing models. • Practical Implementation Guidelines for Reject Option Models: Presented clear practical insights for implementing conformal predictors with reject options, providing empirical guidance for managing trade-offs in real-world applications. • Corrected Error Rate Analysis for Conformal Prediction with Reject Option: Provided a rigorous and corrected analysis of the error rate for binary classification with reject option, addressing weaknesses in previous studies.

Authors

Institutions

Publication Details

Journal
Machine Learning with Applications
Published
2025-05-09
DOI
https://doi.org/10.1016/j.mlwa.2025.100664
Citations
1
Primary Topic
Machine Learning and Algorithms
Type
article
Field-Weighted Citation Impact
2.48
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Classification with reject option: Distribution-free error guarantees via conformal prediction

Ernst Ahlberg, Johan Hallberg Szabadváry, Lars Carlsson, Cecilia Sönströd et al.
1 citations
Machine Learning with Applications
Machine Learning and Algorithms
2.48
article

Classification with reject option: Distribution-free error guarantees via conformal prediction

Ernst Ahlberg, Johan Hallberg Szabadváry, Lars Carlsson, Cecilia Sönströd, Tuve Löfström, Ulf Johansson
article en
1 citations

Abstract

Machine learning (ML) models always make a prediction, even when they are likely to be wrong. This causes problems in practical applications, as we do not know if we should trust a prediction. ML with reject option addresses this issue by abstaining from making a prediction if it is likely to be incorrect. In this work, we formalise the approach to ML with reject option in binary classification, deriving theoretical guarantees on the resulting error rate. This is achieved through conformal prediction (CP), which produce prediction sets with distribution-free validity guarantees. In binary classification, CP can output prediction sets containing exactly one, two or no labels. By accepting only the singleton predictions, we turn CP into a binary classifier with reject option. Here, CP is formally put in the framework of predicting with reject option. We state and prove the resulting error rate, and give finite sample estimates. Numerical examples provide illustrations of derived error rate through several different conformal prediction settings, ranging from full conformal prediction to offline batch inductive conformal prediction. The former has a direct link to sharp validity guarantees, whereas the latter is more fuzzy in terms of validity guarantees but can be used in practice. Error-reject curves illustrate the trade-off between error rate and reject rate, and can serve to aid a user to set an acceptable error rate or reject rate in practice. • Proven Distribution-Free Error Rates for Singleton Predictions: Delivered theoretical guarantees on error rates for accepted singleton predictions, resolving inaccuracies found in prior works. • Refinement of Reject Option in Conformal Prediction: Clarified the relationship between error rate and reject rate using error-reject curves, offering a more accurate approach than existing models. • Practical Implementation Guidelines for Reject Option Models: Presented clear practical insights for implementing conformal predictors with reject options, providing empirical guidance for managing trade-offs in real-world applications. • Corrected Error Rate Analysis for Conformal Prediction with Reject Option: Provided a rigorous and corrected analysis of the error rate for binary classification with reject option, addressing weaknesses in previous studies.

Machine Learning with ApplicationsVol. 20
Stockholm University (SE), Royal Holloway University of London (GB), Mölnlycke Health Care (Sweden) (SE), Jönköping University (SE)
Openalex Percentile: Top 11%
Machine Learning and Algorithms
2.48
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.