arXiv:2505.24273cs.AI2025-05被引 16

通过控制回溯步数,提升大模型推理能力。

How Much Backtracking is Enough? Exploring the Interplay of SFT and RL in Enhancing LLM Reasoning

  • 在SFT阶段引入可调控的回溯步骤,引导模型学习非线性思考。
  • 任务越难,所需回溯深度越大,如魔方谜题需5步以上。
  • 该方法让强化学习更稳定,适合复杂推理任务研究者。

近期研究表明,强化学习(RL)能有效内化搜索策略,通过延长思维链显著提升大语言模型在复杂推理任务上的表现。尽管回溯被视为关键机制,但其动态过程及与监督微调(SFT)的协同作用仍不清晰。本文通过八项推理任务(Countdown、Sudoku、Arc 1D、Advanced Geometry、Color Cube Rotation、List Functions、Zebra Puzzles、Self Reference)研究SFT预热与RL的交互关系。发现使用自采样思维链进行常规SFT预热虽可小幅提升后续RL效果,但性能随任务难度增加而饱和。为此,提出以回溯为核心的训练方法:在SFT预热中人为调节显式回溯步数。结果表明:(i)含回溯的长思维链能稳定并增强RL;(ii)最优回溯深度随任务难度递增——Arc 1D为0步,Countdown为1步,Sudoku需5步或更多。研究证实,有原则的回溯对提升模型推理能力至关重要。

原文摘要 · Abstract (English)

Recent advancements in large language models (LLMs) suggest that reinforcement learning (RL) effectively internalizes search strategies, yielding significant improvements on challenging reasoning tasks through extended chains of thought. While backtracking is widely viewed as the core mechanism enabling this improvement, its precise dynamics and how SFT and RL interact to optimize for it, remain poorly understood. In this work, we argue that backtracking, the ability to explicitly revisit and revise earlier reasoning steps, is a key operator that enables the transition from linear generation to non-linear tree search for LLMs, unlocking strong performance on combinatorially complex tasks. We motivate this claim by first studying the interplay between supervised fine-tuning (SFT) warm-up and RL across eight reasoning tasks: Countdown, Sudoku, Arc 1D, Advanced Geometry, Color Cube Rotation, List Functions, Zebra Puzzles, and Self Reference. We find that a regular SFT warm-up using self-sampled CoT sequences provides a modest boost to subsequent RL compared to training without any SFT warm-up; however, such gains saturate as tasks become more difficult. Motivated by this failure mode, we introduce a backtracking-centric training recipe. By synthetically varying the number of explicit backtracking steps in the SFT warm-up, we show that (i) longer CoTs containing backtracks stabilize and amplify RL, and (ii) the optimal backtrack depth scales with task difficulty: zero for Arc 1D, one for Countdown, and five or more for Sudoku. Our findings establish that principled backtracking is essential for improving the model's reasoning capability.

大模型推理强化学习回溯机制思维链

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