用词元概率聚合提升数学题回答置信度校准
From token probabilities to calibrated confidence: An empirical study of mathematical question answering

- 用全序列词元概率平均作为置信度信号
- 多轮验证方法可实现良好校准,误差显著降低
- 校准方法对数据量和模型有依赖性,需谨慎迁移
大型语言模型的置信度估计旨在预测生成答案正确的概率,而校准则要求这些估计与实际准确率一致。以往研究指出词元概率常过度自信,本文探讨其在数学问答任务中是否仍可用于校准置信度。我们比较了单次生成的估计算法(复用原始生成中的词元概率)与多轮估计算法(通过验证或随机前向传播获取额外信号)。尽管单个词元概率高度饱和,但全序列概率聚合能捕捉正确与错误生成间的微小差异,提供更有效的置信度判断。多轮方法如重提示自验证(含低成本原位变体)和蒙特卡洛丢弃法(基于多次随机前向传播的方差),均可实现校准。此外,后处理校准方法(Platt缩放与等倾回归)显著降低域内校准误差,但其数据效率随数据集难度变化,且校准映射常在不同数据集与模型间不对称迁移。
原文摘要 · Abstract (English)
Confidence estimation for large language models (LLMs) aims to estimate the probability that a generated answer is correct, while calibration aligns these estimates with empirical accuracy. Prior work has shown that token probabilities are often overconfident, we investigate whether these readily available signals can nevertheless provide well-calibrated confidence estimation for mathematical question answering. We compare single-pass estimators, which reuse token probabilities from the original generation, with multi-pass estimators, which obtain additional confidence signals through verification or stochastic forward passes. While individual token probabilities can be highly saturated, we find that aggregating token probabilities over the full sequence captures small but consistent differences between correct and incorrect generations, yielding more informative confidence estimates. Multi-pass methods can yield calibrated confidence estimates. We study two such approaches: self-verification through re-prompting, including a lower-cost in-situ variant, and Monte Carlo Dropout, which derives confidence from variation across stochastic forward passes. We further evaluate two post-hoc calibration methods, Platt scaling and isotonic regression, both of which substantially reduce in-domain calibration error. However, their data efficiency varies with dataset difficulty, and the calibration mappings often transfer asymmetrically across datasets and models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。