arXiv:2502.02508cs.CLcs.AI2025-02ICML被引 57

让大模型自己一步步试错,用强化学习提升推理能力。

Satori: Reinforcement Learning with Chain-of-Action-Thought Enhances LLM Reasoning via Autoregressive Search

  • 用自回归搜索+自我反思的思维链增强推理
  • 70亿参数模型在数学推理上达顶尖水平
  • 适合研究大模型自主推理与训练方法的人

大语言模型在多个领域展现出强大的推理能力。近期研究表明,增加推理时的计算量能显著提升其表现,通常依赖外部LLM验证器进行大量采样,形成双主体系统。尽管有外部引导,该系统仍证明单个模型具备处理复杂任务的潜力。因此,我们提出新问题:能否将搜索能力内化,从根本上提升单个大模型的推理能力?本文探索了一条正交方向,聚焦于后训练阶段的自回归搜索(即扩展的推理过程,包含自我反思和策略探索)。为此,我们提出动作思维链(COAT)推理机制及两阶段训练范式:1)小规模格式调优以内化COAT推理格式;2)大规模自提升阶段利用强化学习。我们的方法生成了Satori——一个基于开源模型与数据训练的70亿参数模型。大量实证评估表明,Satori在数学推理基准上达到当前最优性能,并展现出对域外任务的强大泛化能力。代码、数据与模型均已开源。

原文摘要 · Abstract (English)

Large language models (LLMs) have demonstrated remarkable reasoning capabilities across diverse domains. Recent studies have shown that increasing test-time computation enhances LLMs' reasoning capabilities. This typically involves extensive sampling at inference time guided by an external LLM verifier, resulting in a two-player system. Despite external guidance, the effectiveness of this system demonstrates the potential of a single LLM to tackle complex tasks. Thus, we pose a new research problem: Can we internalize the searching capabilities to fundamentally enhance the reasoning abilities of a single LLM? This work explores an orthogonal direction focusing on post-training LLMs for autoregressive searching (i.e., an extended reasoning process with self-reflection and self-exploration of new strategies). To achieve this, we propose the Chain-of-Action-Thought (COAT) reasoning and a two-stage training paradigm: 1) a small-scale format tuning stage to internalize the COAT reasoning format and 2) a large-scale self-improvement stage leveraging reinforcement learning. Our approach results in Satori, a 7B LLM trained on open-source models and data. Extensive empirical evaluations demonstrate that Satori achieves state-of-the-art performance on mathematical reasoning benchmarks while exhibits strong generalization to out-of-domain tasks. Code, data, and models are fully open-sourced.

大模型推理强化学习自回归搜索

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