mlsanityData to 7 Oct 2026

Zero-Shot Confidence Estimation for Small LLMs: When Supervised Baselines Aren't Worth Training

arXiv 2605.02241 ↗First posted 4 May 2026cs.AI cs.CL cs.ETcited by 0

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

  1. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  2. 2017On Calibration of Modern Neural Networkscited by 4,860
  3. 2022Language Models (Mostly) Know What They Knowcited by 1,783
  4. 2023Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMscited by 946
  5. 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 795
  6. 2023Just Ask for Calibration: Strategies for Eliciting Calibrated Confidence Scores from Language Models Fine-Tuned with Human Feedbackcited by 788
  7. 2024RouteLLM: Learning to Route LLMs with Preference Datacited by 633
  8. 2024Semantic Entropy Probes: Robust and Cheap Hallucination Detection in LLMscited by 246
  9. 2024A Survey on Uncertainty Quantification of Large Language Models: Taxonomy, Open Research Challenges, and Future Directionscited by 115
  10. 2023Self-Evaluation Improves Selective Generation in Large Language Modelscited by 74
  11. 2025Confident or Seek Stronger: Exploring Uncertainty-Based On-device LLM Routing From Benchmarking to Generalizationcited by 29
  12. 2024Log Probabilities Are a Reliable Estimate of Semantic Plausibility in Base and Instruction-Tuned Language Modelscited by 24
10 more
  1. 2025Leveraging Uncertainty Estimation for Efficient LLM Routingcited by 21
  2. 2025Adaptive LLM Routing under Budget Constraintscited by 21
  3. 2025Uncertainty Quantification for Language Models: A Suite of Black-Box, White-Box, LLM Judge, and Ensemble Scorerscited by 16
  4. 2025Learning to Route LLMs from Bandit Feedback: One Policy, Many Trade-offscited by 13
  5. 2025UQLM: A Python Package for Uncertainty Quantification in Large Language Modelscited by 9
  6. 2026LLMs Encode Their Failures: Predicting Success from Pre-Generation Activationscited by 9
  7. 2025LLM Routing with Dueling Feedbackcited by 6
  8. 2026ParetoBandit: Budget-Paced Adaptive Routing for Non-Stationary LLM Servingcited by 3
  9. 2025Confidence-Aware Routing for Large Language Model Reliability Enhancement: A Multi-Signal Approach to Pre-Generation Hallucination Mitigationcited by 1
  10. 2026When Can We Trust LLM Graders? Calibrating Confidence for Automated Assessmentcited by 1

Built on by

Not cited yet.