arXiv:2503.14749cs.CLcs.LG2025-03被引 14

教大模型准确说出答案可信度,让信心表达更真实。

Uncertainty Distillation: Teaching Language Models to Express Semantic Confidence

  • 用校准数据训练模型,让其输出与真实错误率匹配的信心值。
  • 信心表达与实际错误率高度一致,且推理速度比基线快20倍以上。
  • 适用于无法修改的黑盒模型,效果更好且效率更高。

随着大语言模型在事实问答中的广泛应用,模型准确表达答案正确性的置信度变得愈发重要。当前模型在表达信心时,其错误率与宣称的信心水平不一致,亟需不确定性量化方法。以往方法多关注词汇层面的不确定性,而本文提出语义不确定性校准:利用保留数据将初始不确定度映射为有意义的概率,生成带口头化概率标注的训练样本,通过监督微调使模型输出更真实的信心表达。实验表明,该方法生成的信心值与实际错误率高度相关,优于多个强基线,部分基线推理速度慢超过20倍。此外,该方法可应用于仅支持API微调的黑盒模型,获得更高效、更有效的不确定性估计。

原文摘要 · Abstract (English)

As large language models (LLMs) are increasingly used for factual question-answering, it becomes more important for LLMs to have the capability to communicate the likelihood that their answer is correct. For these verbalized expressions of uncertainty to be meaningful, they should reflect the error rates at the expressed level of confidence. However, when prompted to express confidence, the error rates of current LLMs are inconsistent with their communicated confidences, highlighting the need for uncertainty quantification methods. Many prior methods calculate lexical uncertainty, estimating a model's confidence in the specific string it generated. In some cases, however, it may be more useful to estimate semantic uncertainty, or the model's confidence in the answer regardless of how it is verbalized. We propose a simple procedure, uncertainty distillation, to teach an LLM to verbalize calibrated semantic confidences. Using held-out data to map initial uncertainty estimates to meaningful probabilities, we create examples annotated with verbalized probabilities for supervised fine-tuning. We find that our method yields verbalized confidences that correlate well with observed error rates, even when compared to strong baselines, some of which are more than twenty times slower at inference time. Additionally, we demonstrate that our method can be applied to black-box models that allow API-based fine-tuning, resulting in estimates of uncertainty that are both more effective and more efficient than any of our baselines.

不确定性

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