arXiv:2605.15588cs.CLcs.LG2026-05

让大模型在语义层面更准地判断自己对错,提升可信度。

Calibrating LLMs with Semantic-level Reward

论文配图:Calibrating LLMs with Semantic-level Reward
图 1 · 摘自论文原文
  • 在语义空间中设计奖励机制,不依赖口头信心表达。
  • 在多个数据集上降低40%错误率,提升31%判断准确率。
  • 适合医疗、法律等高风险场景下需要可靠判断的模型使用。

随着大语言模型被应用于医疗问答、法律推理等关键场景,准确估计其输出正确性的能力至关重要,需具备良好的不确定性校准。标准强化学习(RLVR)采用二值正确性奖励,对置信度无惩罚,导致高自信错误预测得不到纠正,损害校准效果。近期工作通过训练模型生成口头信心分数并奖励与正确性一致来改进,但该方法仅在词元层面校准,相同语义的不同表述间存在不一致。本文提出「语义级奖励校准(CSR)」框架,在不依赖口头信心接口的前提下,直接在语义空间中校准模型。CSR结合正确性奖励与新型语义校准奖励:对正确推演促进语义一致性,对错误推演抑制虚假一致性。在HotpotQA(分布内)及TriviaQA、MSMARCO、NQ-Open(分布外)四个数据集上的实验表明,CSR在几乎所有设置下均显著优于基于口头信心的基线,错误校准误差(ECE)降低最高达40%,曲线下面积(AUROC)提升最高达31%,且校准行为在四类测试中均表现稳健。

原文摘要 · Abstract (English)

As large language models (LLMs) are deployed in consequential settings such as medical question answering and legal reasoning, the ability to estimate when their outputs are likely to be correct is essential for safe and reliable use, requiring well-calibrated uncertainty. Standard reinforcement learning with verifiable rewards (RLVR) trains models with a binary correctness reward that is indifferent to confidence, providing no penalty for confident but wrong predictions and thereby degrading calibration. Recent work addresses this by training models to produce verbalized confidence scores alongside answers and rewarding agreement with correctness. However, verbalized confidence is calibrated at the token level and thus exhibits inconsistency across textual variations with same semantic meaning. We propose \textbf{Calibration with Semantic Reward (CSR)}, a framework that calibrates language models directly in semantic space without a verbalized confidence interface. CSR combines the correctness reward with a novel semantic calibration reward that encourages exploitation among correct rollouts by promoting semantic agreement, and exploration among incorrect ones by discouraging spurious consistency. Experiments across three model families on HotpotQA (in-distribution) and TriviaQA, MSMARCO, and NQ-Open (out-of-distribution) show that CSR consistently achieves lower ECE and higher AUROC than verbalized-confidence baselines across nearly all settings, reducing ECE by up to $40\%$ and improving AUROC by up to $31\%$ over verbalized-confidence baselines, with calibration behavior generalizing robustly across all four evaluation settings.

大模型校准语义奖励可靠性评估

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。