arXiv:2410.13284cs.CLcs.AI2024-10ICML被引 65

让大模型自己判断答案可信度,提高决策可靠性。

Learning to Route LLMs with Confidence Tokens

  • 引入置信度标记,让模型自动生成可信度信号。
  • 在路由和拒绝学习任务中,准确率显著提升。
  • 适合高风险场景下需要可靠输出的系统使用。

大型语言模型(LLMs)在多个任务中表现出色,正被广泛部署于实际应用。但在高风险场景中,识别模型输出是否不可靠至关重要。根据答案可信度,系统可将其转给专家或采用安全默认行为。本文研究了LLM表达置信度的能力及其对下游任务的增益。提出轻量级训练策略Self-REF,通过引入置信度标记,使模型能以可靠方式表达答案正确性信心。与传统方法如口语化置信度或分析词元概率相比,实验表明置信度标记在下游路由和拒绝学习任务中表现更优。

原文摘要 · Abstract (English)

Large language models (LLMs) have demonstrated impressive performance on several tasks and are increasingly deployed in real-world applications. However, especially in high-stakes settings, it becomes vital to know when the output of an LLM may be unreliable. Depending on whether an answer is trustworthy, a system can then choose to route the question to another expert, or otherwise fall back on a safe default behavior. In this work, we study the extent to which LLMs can reliably indicate confidence in their answers, and how this notion of confidence can translate into downstream accuracy gains. We propose Self-Reflection with Error-based Feedback (Self-REF), a lightweight training strategy to teach LLMs to express confidence in whether their answers are correct in a reliable manner. Self-REF introduces confidence tokens into the LLM, from which a confidence score can be extracted. Compared to conventional approaches such as verbalizing confidence and examining token probabilities, we demonstrate empirically that confidence tokens show significant improvements in downstream routing and rejection learning tasks.

大模型置信度路由自我反思

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