Zero-Shot Confidence Estimation for Small LLMs: When Supervised Baselines Aren't Worth Training
Not on the map. Why?
Abstract (arXiv)
How reliably can a small language model estimate its own correctness? The answer determines whether local-to-cloud routing-escalating queries a cheap local model cannot handle-can work without supervised training data. As inference costs dominate large language model (LLM) deployment budgets, routing most queries to a cheap local model while reserving expensive cloud calls for hard cases is an increasingly common cost-control strategy. We compare zero-shot confidence signals against RouteLLM-style supervised baselines across three 7-8B model families and two datasets (1,000 and 500 queries per model, respectively). Average token log-probability, which requires no training data, matches or exceeds supervised baselines in-distribution (Area Under the Receiver Operating Characteristic curve (AUROC) 0.650-0.714 vs. 0.644-0.676) and substantially outperforms them out-of-distribution (0.717-0.833 vs. 0.512-0.564), because it measures a property of the model's generation rather than the query distribution. This paper further proposes retrieval-conditional self-assessment, a pre-generation signal that selectively injects retrieved knowledge when similarity is high, improving over bare self-assessment by up to +0.069 AUROC at 3-10x lower latency than log-probability. A supervised baseline trained on 1,000 labeled examples never exceeds the zero-shot signal. We release all code, data, and experiment logs.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 33 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
22 of 33 references.
References · 22
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2022Language Models (Mostly) Know What They Knowcited by 1,783
- 2023Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMscited by 946
- 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 795
- 2023Just Ask for Calibration: Strategies for Eliciting Calibrated Confidence Scores from Language Models Fine-Tuned with Human Feedbackcited by 788
- 2024RouteLLM: Learning to Route LLMs with Preference Datacited by 633
- 2024Semantic Entropy Probes: Robust and Cheap Hallucination Detection in LLMscited by 246
- 2024A Survey on Uncertainty Quantification of Large Language Models: Taxonomy, Open Research Challenges, and Future Directionscited by 115
- 2023Self-Evaluation Improves Selective Generation in Large Language Modelscited by 74
- 2025Confident or Seek Stronger: Exploring Uncertainty-Based On-device LLM Routing From Benchmarking to Generalizationcited by 29
- 2024Log Probabilities Are a Reliable Estimate of Semantic Plausibility in Base and Instruction-Tuned Language Modelscited by 24
10 more
- 2025Leveraging Uncertainty Estimation for Efficient LLM Routingcited by 21
- 2025Adaptive LLM Routing under Budget Constraintscited by 21
- 2025Uncertainty Quantification for Language Models: A Suite of Black-Box, White-Box, LLM Judge, and Ensemble Scorerscited by 16
- 2025Learning to Route LLMs from Bandit Feedback: One Policy, Many Trade-offscited by 13
- 2025UQLM: A Python Package for Uncertainty Quantification in Large Language Modelscited by 9
- 2026LLMs Encode Their Failures: Predicting Success from Pre-Generation Activationscited by 9
- 2025LLM Routing with Dueling Feedbackcited by 6
- 2026ParetoBandit: Budget-Paced Adaptive Routing for Non-Stationary LLM Servingcited by 3
- 2025Confidence-Aware Routing for Large Language Model Reliability Enhancement: A Multi-Signal Approach to Pre-Generation Hallucination Mitigationcited by 1
- 2026When Can We Trust LLM Graders? Calibrating Confidence for Automated Assessmentcited by 1
Built on by
Not cited yet.