Optimal Strategies for Reject Option Classifiers
Vojtech FrancDaniel PrusaVáclav Vorácek
Unifies cost-based, bounded-improvement, and bounded-abstention selective classification models by proving they share the same optimal strategy, while developing two Fisher consistent algorithms to learn optimal rejection functions for arbitrary black-box classifiers across diverse prediction tasks.
In safety-critical applications, automated decision models can cause severe operational or financial harm when they make incorrect predictions. To mitigate this risk, systems can use selective classifiers that are permitted to abstain from making a prediction when uncertainty is high. Historically, practitioners relied on three distinct formulations: a cost-based model requiring an explicit financial penalty for abstaining, a bounded-improvement model seeking maximum coverage under a capped error risk, and a bounded-abstention model minimizing risk for a guaranteed coverage level. Defining explicit financial rejection costs is often impractical in real-world scenarios, making the relationship between these formulations and the design of optimal uncertainty scoring mechanisms critical.
The article demonstrates the mathematical equivalence of these three rejection formulations and develops theoretically grounded algorithms to learn uncertainty scores directly from data for any pre-trained, black-box classification model.
To establish these results, the authors mathematically derived the necessary and sufficient optimality conditions for each framework. They introduced two general-purpose learning algorithms to estimate an uncertainty score: one based on loss regression and another based on optimizing a novel smooth surrogate called the Selective Classifier Learning loss. To validate their approach, they performed extensive benchmark experiments using standard machine learning models across 11 standard classification tasks, 11 ordinal regression tasks, and a complex facial landmark detection task.
The analysis produced several key findings. First, all three rejection formulations share an identical underlying optimal decision structure: a base Bayes classifier paired with a randomized threshold rule on the conditional expected risk. Second, minimizing the Selective Classifier Learning loss provably delivers a proper uncertainty score that preserves the optimal ranking of error risks. Third, in classification benchmarks, the Selective Classifier Learning approach achieved the top average rank, significantly outperforming conventional margin-based scores and loss regression. Fourth, in ordinal regression and facial landmark detection, the proposed approach reduced the area under the risk-coverage curve by up to 40 to 50 percent relative to baseline heuristic scores, performing on par with specialized, model-dependent state-of-the-art methods.
These findings provide immediate practical value for managing operational risk and compliance in high-stakes deployments. Decision-makers are no longer forced to assign arbitrary monetary costs to abstentions; they can instead specify straightforward performance targets, such as maximum allowable error or minimum operational coverage. Furthermore, because the proposed scoring algorithms operate on top of existing models without altering their underlying parameters, organizations can enhance legacy predictive systems with reliable abstention capabilities at minimal engineering cost.
For practical implementation, engineering teams deploying black-box models should adopt the Selective Classifier Learning loss algorithm to construct uncertainty scores and tune rejection thresholds using empirical validation sets. When probabilistic models are already in place, simple posterior probability rules remain a viable, low-effort alternative. Future initiatives should focus on developing simultaneous training pipelines that optimize base predictors and uncertainty scores concurrently, as well as scaling non-linear scoring architectures.
Confidence in these theoretical principles and empirical rankings is high across tested benchmark distributions. However, decision-makers should note that the current experimental evaluations rely on linear scoring functions and separate two-stage training datasets. Performance should be verified with domain-specific pilots before broad deployment in production environments.
- Paper: The foundations of cost-sensitive learning, Charles Elkan (2001). It provides the foundational decision-theoretic principles and cost-matrix formulations for cost-sensitive classification that the source paper's cost-based rejection framework directly relies upon.
- Paper: Strictly Proper Scoring Rules, Prediction, and Estimation, Tilmann Gneiting et al. (2007). It establishes the mathematical theory of proper scoring rules and uncertainty quantification necessary for defining and estimating proper uncertainty scores in reject-option classifiers.
- Paper: Predicting good probabilities with supervised learning, Alexandru Niculescu-Mizil et al. (2005). It analyzes methods for calibrating posterior class probability estimates from black-box classifiers, which underpins the source's approach to learning uncertainty scores.
- Paper: MetaCost: a general method for making classifiers cost-sensitive, Pedro M. Domingos (1999). It introduces a framework for converting arbitrary black-box classifiers into cost-sensitive predictors using probability estimations, motivating the source's black-box rejection algorithms.
- Paper: Transforming classifier scores into accurate multiclass probability estimates, Bianca Zadrozny et al. (2002). It examines multiclass probability calibration techniques that are critical for deriving proper uncertainty scores across multiclass and structured prediction problems.
- Paper: Aleatoric and epistemic uncertainty in machine learning: an introduction to concepts and methods, Eyke Hüllermeier et al. (2019). It offers an extensive conceptual foundation for modeling and quantifying prediction uncertainty in supervised machine learning.
- Paper: Two-Stage Learning to Defer with Multiple Experts, Anqi Mao et al. (2023). It generalizes selective prediction and rejection models to two-stage deferral systems involving multiple downstream human or algorithmic experts.
