arXiv:2502.14634cs.LG2025-02ACL被引 22

让大模型学会评估自己推理的可信度,提升数学和开放域生成的准确率。

CER: Confidence Enhanced Reasoning in LLMs

  • 在关键步骤量化模型对中间答案的置信度,构建可信度评估框架。
  • 数学任务准确率提升7.4%,开放域生成提升5.8%,优于传统多数投票方法。
  • 适合需要高可靠性推理的场景,如数学求解、知识密集型问答。

确保大语言模型在复杂推理任务中的可靠性仍是重大挑战,尤其在需要精确数学计算和知识密集型开放域生成的场景中。本文提出一种不确定性感知框架,通过在关键决策点系统性引入模型置信度,提升响应准确性。该方法促使大模型进行多步推理,并对数学推理中的数值结果、开放域生成中的专有名词等中间答案的置信度进行量化。随后基于这些关键中间步骤的置信度评估整个推理链的总体置信度。最终,以反映各生成路径可靠性的加权方式聚合答案(而非自一致性方法中各路径同等贡献)。我们在五个数据集上开展实验,涵盖三个数学数据集和两个开放域数据集,使用四种大模型。结果一致验证了所提置信度聚合方法的有效性,在数学和开放域生成任务中分别相较基线提升7.4%和5.8%。代码已公开于 https://github.com/Aquasar11/CER。

原文摘要 · Abstract (English)

Ensuring the reliability of Large Language Models (LLMs) in complex reasoning tasks remains a formidable challenge, particularly in scenarios that demand precise mathematical calculations and knowledge-intensive open-domain generation. In this work, we introduce an uncertainty-aware framework designed to enhance the accuracy of LLM responses by systematically incorporating model confidence at critical decision points. We propose an approach that encourages multi-step reasoning in LLMs and quantify the confidence of intermediate answers such as numerical results in mathematical reasoning and proper nouns in open-domain generation. Then, the overall confidence of each reasoning chain is evaluated based on confidence of these critical intermediate steps. Finally, we aggregate the answer of generated response paths in a way that reflects the reliability of each generated content (as opposed to self-consistency in which each generated chain contributes equally to majority voting). We conducted extensive experiments in five datasets, three mathematical datasets and two open-domain datasets, using four LLMs. The results consistently validate the effectiveness of our novel confidence aggregation method, leading to an accuracy improvement of up to 7.4% and 5.8% over baseline approaches in math and open-domain generation tasks, respectively. Code is publicly available at https://github.com/ Aquasar11/CER.

大模型推理置信度评估数学推理

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