让大模型能回溯错误思路,自主修正推理路径。
Toward Adaptive Reasoning in Large Language Models with Thought Rollback
- 引入思想回溯机制,允许模型回退并修正错误推理
- 在MATH数据集上,GPT-4解决率比现有最佳提升9%
- 无需人工标注,可自适应探索可靠解题路径
大型语言模型(LLMs)通常通过逐步推理解决各类任务,但其推理过程结构僵化且单向,如链式、树状或无环有向图。这种固定且仅向前的推理方式难以应对复杂任务,在模型频繁产生幻觉时易失败。本文提出一种新推理框架——思想回溯(Thought Rollback, TR),使模型在出现幻觉时能自适应构建推理结构,并持续优化解题路径。其核心机制是回溯错误思想,进行错误分析并回退至先前误判节点进行修正。通过将试错过程纳入提示引导模型,每次回溯均生成一条更可靠的推理路径。从无人工标注的简单提示开始,具备TR能力的模型可渐进式探索正确解法。在数学问题与多任务推理上的综合实验表明,该方法在解题成功率与交互成本方面均达到当前最优。例如,使用TR的GPT-4在MATH数据集上的解题率相比当前最佳高出9%。
原文摘要 · Abstract (English)
Large language models (LLMs) have been routinely used to solve various tasks using step-by-step reasoning. However, the structure of intermediate reasoning steps, or thoughts, is rigid and unidirectional, such as chains, trees, or acyclic-directed graphs. Consequently, the resulting inflexible and forward-only reasoning may not address challenging tasks and fail when the LLM frequently gives false responses, i.e., ``hallucinations''. This paper proposes a new reasoning framework, called Thought Rollback (TR), allowing LLMs to adaptively build thought structure while maintaining effective reasoning toward problem-solving under ``hallucinations''. The core mechanism of TR is rolling back thoughts, which allows LLMs to perform error analysis on thoughts, and thus roll back to any previously mistaken thought for revision. Subsequently, by including such trial-and-error in the prompt to guide the LLM, each rollback leads to one more reliable reasoning path. Therefore, starting with a simple prompt without human annotations, LLM with TR adaptively and gradually explores thoughts for a correct solution. Comprehensive experiments on mathematical problems and multi-task reasoning demonstrate the state-of-the-art performance of TR in terms of problem-solving rate and interaction cost. For instance, the solving rate of GPT-4 with TR outperforms the current best by $9\%$ on the MATH dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。