Free Lunch for LLMs Confidence: The RL Critic Already Knows

arXiv cs.CL Papers

Summary

该论文证明在基于正确性奖励的价值函数强化学习中,置信度估计与critic训练是等价问题,因此已有的RL critic可直接复用为LLM的置信度估计器,无需额外训练或数据,且在数学、代码、工具调用等多个领域中表现出最佳的置信度估计与校准性能。

arXiv:2610.31820v1 Announce Type: new Abstract: Reliable confidence estimation is essential for trustworthy deployment of large language models (LLMs), but dedicated confidence training stage requires additional data and compute. We ask a fundamental question: what prediction problem does confidence training actually solve? Inspired by prior work on external confidence estimation, we compare self- and external estimators of comparable capacity trained on the same (question, response, correctness) triples. We show that an external model trained solely on a policy's behavioral traces can estimate the policy's confidence as accurately as the policy itself, without accessing the policy's internal representations. This points to a deeper connection with value function learning in reinforcement learning (RL): under a correctness-based outcome, we prove that confidence estimation and critic training are equivalent, both predicting the likelihood that a policy's response will be correct from its trajectory. A simple consequence follows: for LLMs trained with value-based RL, the existing critic can be reused as an off-the-shelf confidence estimator with no additional training or data. The policy remains unchanged, so its task performance is preserved, and the critic provides confidence throughout generation, including intermediate reasoning steps. Across math, code, tool calling, and general reasoning spanning a dozen domains, the reused critic is the best confidence estimator on every in-distribution split and the best-calibrated on 14 of 16, without hurting the task performance.
Original Article

Similar Articles

microsoft/Mage-VL-Cowork · Hugging Face

Reddit r/LocalLLaMA

Microsoft releases Mage-VL-Cowork, a compact 4B vision-language model that combines GUI interaction and terminal/code execution in a single policy, achieving 66.0% on OSWorld-Verified — the best result at the 4B scale — via a five-stage curriculum and online RL.