Learning to Reject with a Fixed Predictor: Application to Decontextualization
Not on the map. Why?
Abstract (arXiv)
We study the problem of classification with a reject option for a fixed predictor, applicable in natural language processing. We introduce a new problem formulation for this scenario, and an algorithm minimizing a new surrogate loss function. We provide a complete theoretical analysis of the surrogate loss function with a strong H-consistency guarantee. For evaluation, we choose the decontextualization task, and provide a manually-labelled dataset of 2mathord,000 examples. Our algorithm significantly outperforms the baselines considered, with a ~!!25% improvement in coverage when halving the error rate, which is only ~!! 3 % away from the theoretical limit.
Key facts. Cited by 32 papers in the map (data through 7 Oct 2026). Builds on 29 of its 61 references that are in the map. First posted on arXiv 22 Jan 2023.
Builds on
29 of 61 references.
References · 29
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2020mT5: A massively multilingual pre-trained text-to-text transformercited by 1,813
- 2019How Can We Know What Language Models Know?cited by 953
- 2020On Faithfulness and Factuality in Abstractive Summarizationcited by 895
- 2017Selective Classification for Deep Neural Networkscited by 799
- 2020How Can We Know When Language Models Know? On the Calibration of Language Models for Question Answeringcited by 439
17 more
- 2018Deep Learning for Classical Japanese Literaturecited by 415
- 2020Selective Question Answering under Domain Shiftcited by 228
- 2022Scaling Up Models and Data with textttt5x and textttseqiocited by 187
- 2018Understanding Measures of Uncertainty for Adversarial Example Detectioncited by 177
- 2021Machine Learning with a Reject Option: A surveycited by 107
- 2021Decontextualization: Making Sentences Stand-Alonecited by 101
- 2020Controlled Hallucinations: Learning to Generate Faithfully from Noisy Datacited by 83
- 2019Calibration of Encoder Decoder Models for Neural Machine Translationcited by 67
- 2020Classification with Rejection Based on Cost-sensitive Classificationcited by 55
- 2021Calibration and Consistency of Adversarial Surrogate Lossescited by 47
- 2021Knowing More About Questions Can Help: Improving Calibration in Question Answeringcited by 47
- 2019On the Calibration of Multiclass Classification with Rejectioncited by 41
- 2018Confidence Modeling for Neural Semantic Parsingcited by 33
- 2021A Finer Calibration Analysis for Adversarial Robustnesscited by 26
- 2020Towards Improving Selective Prediction Ability of NLP Systemscited by 19
- 2021Will this Question be Answered? Question Filtering via Answer Model Distillation for Efficient Question Answeringcited by 19
- 2017The Relationship Between Agnostic Selective Classification Active Learning and the Disagreement Coefficientcited by 6
Built on by
Citing papers per year: 2023: 4; 2024: 12; 2025: 8; 2026: 8.
32 papers.
Sort:
- 2024Language Models with Conformal Factuality Guaranteescited by 128
- 2024Learning to Route LLMs with Confidence Tokenscited by 66
- 2024Learning to Decode Collaboratively with Multiple Language Modelscited by 61
- 2023Predictor-Rejector Multi-Class Abstention: Theoretical Analysis and Algorithmscited by 36
- 2023Theoretically Grounded Loss Functions and Algorithms for Score-Based Multi-Class Abstentioncited by 32
- 2024Regression with Multi-Expert Deferralcited by 29
- 2024Realizable H-Consistent and Bayes-Consistent Loss Functions for Learning to Defercited by 26
- 2024A Universal Growth Rate for Learning with Smooth Surrogate Lossescited by 24
- 2024Cardinality-Aware Set Prediction and Top-k Classificationcited by 23
- 2024Multi-Label Learning with Stronger Consistency Guaranteescited by 22
- 2024H-Consistency Guarantees for Regressioncited by 21
- 2025Conformal Linguistic Calibration: Trading-off between Factuality and Specificitycited by 20
20 more
- 2024Enhanced H-Consistency Boundscited by 18
- 2025Mastering Multiple-Expert Routing: Realizable H-Consistency and Strong Guarantees for Learning to Defercited by 16
- 2023Selective Generation for Controllable Language Modelscited by 15
- 2024Optimal Query Allocation in Extractive QA with LLMs: A Learning-to-Defer Framework with Theoretical Guaranteescited by 15
- 2025Balancing the Scales: A Theoretical and Algorithmic Framework for Learning from Imbalanced Datacited by 15
- 2023Learning to Make Adherence-Aware Advicecited by 13
- 2025Budgeted Multiple-Expert Deferralcited by 11
- 2025Improved Balanced Classification with Theoretically Grounded Loss Functionscited by 9
- 2025Why Ask One When You Can Ask k? Learning-to-Defer to the Top-k Expertscited by 6
- 2025Principled Algorithms for Optimizing Generalized Metrics in Binary Classificationcited by 6
- 2024Top-k Classification and Cardinality-Aware Predictioncited by 5
- 2025Adversarial Robustness in One-Stage Learning-to-Defercited by 5
- 2026A Theoretical Framework for Modular Learning of Robust Generative Modelscited by 4
- 2026Optimized Deferral for Imbalanced Settingscited by 2
- 2026Linear-Core Surrogates: Smooth Loss Functions with Linear Rates for Classification and Structured Predictioncited by 2
- 2026Selective Prior Synchronization via SYNC Losscited by 0
- 2026Flexible Routing via Uncertainty Decompositioncited by 0
- 2026Online Learning-to-Defer with Varying Expertscited by 0
- 2026Principled Algorithms for Optimizing Generalized Metrics in Multi-Label Learningcited by 0
- 2026Human-AI Teaming Through the Lens of Calibrationcited by 0
- 2026Human-AI Teaming Through the Lens of Calibration9 Jun 2026
- 2026Principled Algorithms for Optimizing Generalized Metrics in Multi-Label Learning27 May 2026
- 2026Online Learning-to-Defer with Varying Experts12 May 2026
- 2026Flexible Routing via Uncertainty Decomposition8 May 2026
- 2026Linear-Core Surrogates: Smooth Loss Functions with Linear Rates for Classification and Structured Prediction30 Apr 2026
- 2026Optimized Deferral for Imbalanced Settings30 Apr 2026
- 2026A Theoretical Framework for Modular Learning of Robust Generative Models19 Feb 2026
- 2026Selective Prior Synchronization via SYNC Loss11 Feb 2026
- 2025Improved Balanced Classification with Theoretically Grounded Loss Functions30 Dec 2025
- 2025Principled Algorithms for Optimizing Generalized Metrics in Binary Classification29 Dec 2025
- 2025Budgeted Multiple-Expert Deferral30 Oct 2025
- 2025Adversarial Robustness in One-Stage Learning-to-Defer13 Oct 2025
20 more
- 2025Mastering Multiple-Expert Routing: Realizable H-Consistency and Strong Guarantees for Learning to Defer25 Jun 2025
- 2025Why Ask One When You Can Ask k? Learning-to-Defer to the Top-k Experts17 Apr 2025
- 2025Conformal Linguistic Calibration: Trading-off between Factuality and Specificity26 Feb 2025
- 2025Balancing the Scales: A Theoretical and Algorithmic Framework for Learning from Imbalanced Data14 Feb 2025
- 2024Optimal Query Allocation in Extractive QA with LLMs: A Learning-to-Defer Framework with Theoretical Guarantees21 Oct 2024
- 2024Learning to Route LLMs with Confidence Tokens17 Oct 2024
- 2024Multi-Label Learning with Stronger Consistency Guarantees18 Jul 2024
- 2024Realizable H-Consistent and Bayes-Consistent Loss Functions for Learning to Defer18 Jul 2024
- 2024Enhanced H-Consistency Bounds18 Jul 2024
- 2024Cardinality-Aware Set Prediction and Top-k Classification9 Jul 2024
- 2024A Universal Growth Rate for Learning with Smooth Surrogate Losses9 May 2024
- 2024Top-k Classification and Cardinality-Aware Prediction28 Mar 2024
- 2024Regression with Multi-Expert Deferral28 Mar 2024
- 2024H-Consistency Guarantees for Regression28 Mar 2024
- 2024Learning to Decode Collaboratively with Multiple Language Models6 Mar 2024
- 2024Language Models with Conformal Factuality Guarantees15 Feb 2024
- 2023Predictor-Rejector Multi-Class Abstention: Theoretical Analysis and Algorithms23 Oct 2023
- 2023Theoretically Grounded Loss Functions and Algorithms for Score-Based Multi-Class Abstention23 Oct 2023
- 2023Learning to Make Adherence-Aware Advice1 Oct 2023
- 2023Selective Generation for Controllable Language Models18 Jul 2023
- 2024Language Models with Conformal Factuality Guaranteescited by 128
- 2024Learning to Route LLMs with Confidence Tokenscited by 66
- 2024Learning to Decode Collaboratively with Multiple Language Modelscited by 61
- 2023Predictor-Rejector Multi-Class Abstention: Theoretical Analysis and Algorithmscited by 36
- 2023Theoretically Grounded Loss Functions and Algorithms for Score-Based Multi-Class Abstentioncited by 32
- 2024Regression with Multi-Expert Deferralcited by 29
- 2024Realizable H-Consistent and Bayes-Consistent Loss Functions for Learning to Defercited by 26
- 2024A Universal Growth Rate for Learning with Smooth Surrogate Lossescited by 24
- 2024Cardinality-Aware Set Prediction and Top-k Classificationcited by 23
- 2024Multi-Label Learning with Stronger Consistency Guaranteescited by 22
- 2024H-Consistency Guarantees for Regressioncited by 21
- 2025Conformal Linguistic Calibration: Trading-off between Factuality and Specificitycited by 20
12 most cited of 32 All 32 (JSON)
- 2026Human-AI Teaming Through the Lens of Calibration9 Jun 2026
- 2026Principled Algorithms for Optimizing Generalized Metrics in Multi-Label Learning27 May 2026
- 2026Online Learning-to-Defer with Varying Experts12 May 2026
- 2026Flexible Routing via Uncertainty Decomposition8 May 2026
- 2026Linear-Core Surrogates: Smooth Loss Functions with Linear Rates for Classification and Structured Prediction30 Apr 2026
- 2026Optimized Deferral for Imbalanced Settings30 Apr 2026
- 2026A Theoretical Framework for Modular Learning of Robust Generative Models19 Feb 2026
- 2026Selective Prior Synchronization via SYNC Loss11 Feb 2026
- 2025Improved Balanced Classification with Theoretically Grounded Loss Functions30 Dec 2025
- 2025Principled Algorithms for Optimizing Generalized Metrics in Binary Classification29 Dec 2025
- 2025Budgeted Multiple-Expert Deferral30 Oct 2025
- 2025Adversarial Robustness in One-Stage Learning-to-Defer13 Oct 2025
12 newest of 32 All 32 (JSON)