通过对齐推理步骤提升大模型数学能力,让简单例子也能发挥大作用。
BoostStep: Boosting mathematical capability of Large Language Models via improved single-step reasoning
- 用步骤对齐的示范例增强推理准确性,避免信息错配。
- 使GPT-4o在数学基准上性能提升4.6%,比传统方法高3.4个百分点。
- 适合想用简单示例提升大模型解题能力的研究者和开发者。
大语言模型(LLMs)在多步推理解决复杂数学问题上表现优异,可通过精心设计的上下文学习(ICL)示例进一步提升。然而,现有方法常受粒度不匹配和无关信息干扰两大挑战制约:模型虽能分解数学问题,但在细粒度步骤中易出错;以题目为单位检索的示例可能遗漏关键步骤,甚至引入误导性内容。为此,我们提出BoostStep,一种通过步骤对齐的ICL机制,精准匹配参考步骤与当前推理步骤。同时引入“首次尝试”策略,确保示例与当前推理状态高度相关。BoostStep可无缝集成链式思维(CoT)与树搜索算法,优化候选选择与决策过程。实验表明,它使GPT-4o在数学基准上的CoT性能提升4.6%,显著优于传统少样本学习的1.2%;与树搜索结合时额外提升7.5%。令人惊讶的是,仅使用MATH数据集中的简单示例,便使DeepSeek-R1-671B在AIME任务上提升2.2%。
原文摘要 · Abstract (English)
Large language models (LLMs) have demonstrated impressive ability in solving complex mathematical problems with multi-step reasoning and can be further enhanced with well-designed in-context learning (ICL) examples. However, this potential is often constrained by two major challenges in ICL: granularity mismatch and irrelevant information. We observe that while LLMs excel at decomposing mathematical problems, they often struggle with reasoning errors in fine-grained steps. Moreover, ICL examples retrieved at the question level may omit critical steps or even mislead the model with irrelevant details. To address this issue, we propose BoostStep, a method that enhances reasoning accuracy through step-aligned ICL, a novel mechanism that carefully aligns retrieved reference steps with the corresponding reasoning steps. Additionally, BoostStep incorporates an effective "first-try" strategy to deliver exemplars highly relevant to the current state of reasoning. BoostStep is a flexible and powerful method that integrates seamlessly with chain-of-thought (CoT) and tree search algorithms, refining both candidate selection and decision-making. Empirical results show that BoostStep improves GPT-4o's CoT performance by 4.6% across mathematical benchmarks, significantly surpassing traditional few-shot learning's 1.2%. Moreover, it can achieve an additional 7.5\% gain combined with tree search. Surprisingly, it enhances state-of-the-art LLMs to solve challenging math problems using simpler examples. It improves DeepSeek-R1-671B's performance on AIME by 2.2%, leveraging simple examples only from the MATH dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。