通过多听众机制提升大模型推理的可信度与性能平衡。
Balancing Faithfulness and Performance in Reasoning via Multi-Listener Soft Execution
- 让多个模型作为听众验证推理链,激励生成更易理解的中间步骤。
- 在多个基准上同时提升推理可信度和准确率,尤其改善了错误注入检测能力。
- 适合需要高可信推理解释的场景,如医疗、金融等严谨领域应用。
链式思维(CoT)推理常无法真实反映大语言模型(LLM)的实际计算过程,限制其解释能力。同时,优化推理的可信度与可解释性往往导致任务性能下降。为解决这一权衡问题,我们提出多听众推理执行(REMUL),一种基于多方强化学习的方法。该方法假设:能被其他模型理解的推理链更可信。由说话者模型生成推理链,截断后传递给一组听者模型,听者“执行”该链并完成推理。说话者根据听者能否顺利跟进获得奖励,并通过掩码监督微调增强正确性以缓解可信度与性能间的冲突。在多个推理基准(BIG-Bench Extra Hard、MuSR、ZebraLogicBench、FOLIO)上,REMUL持续显著提升了三项可信度指标——提示归因、早期回答面积曲线下面积(AOC)、错误注入AOC,同时提高了准确率。分析表明,这些提升在不同训练领域均稳健,且带来推理链可读性的增强,表现为更短、更直接的推理路径。
原文摘要 · Abstract (English)
Chain-of-thought (CoT) reasoning sometimes fails to faithfully reflect the true computation of a large language model (LLM), hampering its utility in explaining how LLMs arrive at their answers. Moreover, optimizing for faithfulness and interpretability in reasoning often degrades task performance. To address this tradeoff and improve CoT faithfulness, we propose Reasoning Execution by Multiple Listeners (REMUL), a multi-party reinforcement learning approach. REMUL builds on the hypothesis that reasoning traces which other parties can follow will be more faithful. A speaker model generates a reasoning trace, which is truncated and passed to a pool of listener models who "execute" the trace, continuing the trace to an answer. Speakers are rewarded for producing reasoning that is clear to listeners, with additional correctness regularization via masked supervised finetuning to counter the tradeoff between faithfulness and performance. On multiple reasoning benchmarks (BIG-Bench Extra Hard, MuSR, ZebraLogicBench, and FOLIO), REMUL consistently and substantially improves three measures of faithfulness -- hint attribution, early answering area over the curve (AOC), and mistake injection AOC -- while also improving accuracy. Our analysis finds that these gains are robust across training domains, translate to legibility gains, and are associated with shorter and more direct CoTs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。