Can Language Models Learn When to Abstain? Policy Learning and Calibrated Answer Selection
cs.CL
Submitted: 2026-10-09
Updated: 2026-10-09
Comments: 17 pages, 2 figures, 10 tables
Code: https://github.com/jrterven/Can-LLMs-learn-to-abstain
License: http://creativecommons.org/licenses/by/4.0/
The gist: A useful language model should know when to answer and when to say "I don't know." Can we improve that decision by training the model itself or by training a separate component to check its answers?
Abstract
A useful language model should know when to answer and when to say "I don't know." Can we improve that decision by training the model itself or by training a separate component to check its answers? We study both approaches on Qwen3-14B for factual questions, using a score that rewards correct answers, penalizes mistakes, and allows abstention. Across three training seeds and two held-out confidence thresholds, a learned answer selector improves threshold-averaged utility over the original model's calibrated confidence filter. On TriviaQA, mean coverage increases from 20.2% to 30.9%, while the pooled error rate among scored answers decreases from 23.0% to 14.5% under exact-match grading. The tested reinforcement-learning changes do not establish consistent gains. On NQ-Open, the selector improves threshold-averaged utility over the filter but still performs worse than always abstaining. A partial human-reviewed audit preserves the TriviaQA finding. Our results support learning to select answers in this setting while showing that reliable abstention across datasets remains unresolved.
Sources
- Why Language Models Hallucinate
- Language Models (Mostly) Know What They Know
- Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMs
- Mitigating LLM Hallucinations via Conformal Abstention
- Rewarding Intellectual Humility Learning When Not To Answer In Large Language Models
- I-CALM: Incentivizing Confidence-Aware Abstention for LLM Hallucination Mitigation
- Qwen3 Technical Report
- LoRA: Low-Rank Adaptation of Large Language Models
- Understanding R1-Zero-Like Training: A Critical Perspective
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering