arXiv:2503.01307cs.CLcs.LG2025-03被引 429

发现语言模型自我提升的关键是四种思维习惯,而非答案正确性。

Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRs

  • 分析四种推理行为:验证、回溯、设定子目标、逆向推导。
  • 用含这些行为的示例微调后,Llama性能追平甚至超过Qwen。
  • 推理过程比答案对错更重要,适合想提升模型思考能力的研究者。

测试时推理已成为让语言模型更深入思考复杂问题的强大范式,类似人类专家。尽管强化学习(RL)可推动语言模型在可验证任务上的自我改进,但部分模型表现显著优于另一些模型且很快达到瓶颈。例如,在相同的RL训练下,Qwen-2.5-3B在计数游戏(Countdown)中远超Llama-3.2-3B。这一差异引发关键问题:哪些内在特性促成了有效自我改进?我们提出框架,分析专家人类与成功语言模型共有的四种认知行为——验证、回溯、子目标设定和逆向推导。研究发现,Qwen天然具备这些行为,而Llama初始缺乏。在控制行为数据集的系统实验中,用含此类行为的例子对Llama进行提示,其在RL中实现显著提升,达到或超越Qwen表现。重要的是,推理行为的存在比答案正确性更为关键——即使提示使用错误但具合理推理路径的解法,模型表现也与基于正确解法训练相当。最后,通过使用增强推理行为的OpenWebMath数据继续预训练,使Llama达到与Qwen相当的自我改进轨迹。结果揭示了初始推理行为与提升潜力之间的根本关联,解释了为何部分模型能有效利用额外计算而另一些则停滞不前。

原文摘要 · Abstract (English)

Test-time inference has emerged as a powerful paradigm for enabling language models to ``think'' longer and more carefully about complex challenges, much like skilled human experts. While reinforcement learning (RL) can drive self-improvement in language models on verifiable tasks, some models exhibit substantial gains while others quickly plateau. For instance, we find that Qwen-2.5-3B far exceeds Llama-3.2-3B under identical RL training for the game of Countdown. This discrepancy raises a critical question: what intrinsic properties enable effective self-improvement? We introduce a framework to investigate this question by analyzing four key cognitive behaviors -- verification, backtracking, subgoal setting, and backward chaining -- that both expert human problem solvers and successful language models employ. Our study reveals that Qwen naturally exhibits these reasoning behaviors, whereas Llama initially lacks them. In systematic experimentation with controlled behavioral datasets, we find that priming Llama with examples containing these reasoning behaviors enables substantial improvements during RL, matching or exceeding Qwen's performance. Importantly, the presence of reasoning behaviors, rather than correctness of answers, proves to be the critical factor -- models primed with incorrect solutions containing proper reasoning patterns achieve comparable performance to those trained on correct solutions. Finally, leveraging continued pretraining with OpenWebMath data, filtered to amplify reasoning behaviors, enables the Llama model to match Qwen's self-improvement trajectory. Our findings establish a fundamental relationship between initial reasoning behaviors and the capacity for improvement, explaining why some language models effectively utilize additional computation while others plateau.

自提升推理行为强化学习模型对比

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