Post-hoc estimators for learning to defer to an expert
Harikrishna Narasimhan, Wittawat Jitkrittum, Aditya Krishna Menon, Ankit Singh Rawat, Sanjiv Kumar
Abstract
Many practical settings allow a classifier to defer predictions to one or more costly experts . For example, the learning to defer paradigm allows a classifier to defer to a human expert, at some monetary cost. Similarly, the adaptive inference paradigm allows a base model to defer to one or more large models, at some computational cost. The goal in these settings is to learn classification and deferral mechanisms to optimise a suitable accuracy-cost tradeo � . To achieve this, a central issue studied in prior work is the design of a coherent loss function for both mechanisms. In this work, we demonstrate that existing losses can underfit the training set when there is a non-trivial deferral cost, owing to an implicit application of a high level of label smoothing. To resolve this, we propose two post-hoc estimators that fit a deferral function on top of a base model, either by threshold correction, or by learning when the base model’s error rate exceeds the cost of deferring to the expert. Both approaches are equipped with theoretical guarantees, and empirically yield e � ective accuracy-cost tradeo � s on learning to defer and adaptive inference benchmarks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 38b7b9bf-2da7-4f35-8bf4-c2997fcc3991Cited by top-tier papers21
- Universal Model Routing for Efficient LLM InferenceWittawat Jitkrittum, Harikrishna Narasimhan, Ankit Singh Rawat, Jeevesh Juneja et al.ICLR 2026 · 99 citations
- Two-Stage Learning to Defer with Multiple ExpertsAnqi Mao, Christopher Mohri, Mehryar Mohri, Yutao ZhongNeurIPS 2023 · 98 citations
- Plan-Then-Execute: An Empirical Study of User Trust and Team Performance When Using LLM Agents As A Daily AssistantGaole He, Gianluca Demartini, Ujwal GadirajuCHI 2025 · 91 citations
- When Does Confidence-Based Cascade Deferral Suffice?Wittawat Jitkrittum, Neha Gupta, Aditya Krishna Menon, Harikrishna Narasimhan et al.NeurIPS 2023 · 76 citations
- Realizable H-Consistent and Bayes-Consistent Loss Functions for Learning to DeferAnqi Mao, Mehryar Mohri, Yutao ZhongNeurIPS 2024 · 37 citations
Builds on7
- Consistent Estimators for Learning to Defer to an ExpertHussein Mozannar, David A. SontagICML 2020 · 267 citations
- Is the Most Accurate AI the Best Teammate? Optimizing AI for TeamworkGagan Bansal, Besmira Nushi, Ece Kamar, Eric Horvitz et al.AAAI 2021 · 185 citations
- Classification with Rejection Based on Cost-sensitive ClassificationNontawat Charoenphakdee, Zhenghang Cui, Yivan Zhang, Masashi SugiyamaICML 2021 · 78 citations
- Calibrated Learning to Defer with One-vs-All ClassifiersRajeev Verma, Eric T. NalisnickICML 2022 · 76 citations
- Unbiased Risk Estimators Can Mislead: A Case Study of Learning with Complementary LabelsYu-Ting Chou, Gang Niu, Hsuan-Tien Lin, Masashi SugiyamaICML 2020 · 66 citations
Related papers
- Optimized Deferral for Imbalanced SettingsCorinna Cortes, Anqi Mao, Mehryar Mohri, Yutao ZhongICML 2026 · 7 citations
- Regression with Multi-Expert DeferralAnqi Mao, Mehryar Mohri, Yutao ZhongICML 2024 · 31 citations
- Sample Efficient Learning of Predictors that Complement HumansMohammad-Amin Charusaie, Hussein Mozannar, David A. Sontag, Samira SamadiICML 2022 · 52 citations
- In Defense of Softmax Parametrization for Calibrated and Consistent Learning to DeferYuzhou Cao, Hussein Mozannar, Lei Feng, Hongxin Wei et al.NeurIPS 2023 · 36 citations
- Is the acquisition worth the cost? Surrogate losses for Consistent Two-stage ClassifiersFlorence Regol, Joseph Cotnareanu, Theodore Glavas, Mark CoatesNeurIPS 2025 · 3 citations
