发现强化学习微调大模型本质是伪RL,实为带过滤的监督学习。
RL in Name Only? Analyzing the Structural Assumptions in RL post-training for LLMs
- 将语言模型训练视为马尔可夫决策过程,假设状态为动作拼接,导致问题退化为上下文无关的老虎机问题。
- 实验表明,改进版的迭代监督微调在GSM8K和Countdown上表现与GRPO相当。
- 揭示了‘强化学习促进思考’的误解来源:模型生成更长中间序列只是结构假设的副产品。
基于强化学习的大语言模型后训练近期备受关注,尤其在DeepSeek R1采用GRPO进行微调后。尽管有大量声称其提升了推理能力,本文深入分析了相关方法背后的建模假设。我们指出,主流方法将语言模型训练建模为马尔可夫决策过程(MDP)时,存在两个关键结构性假设:(1) 状态仅为动作的拼接,状态即上下文窗口,动作即生成的词元;(2) 将轨迹奖励均匀分配给所有步骤。分析表明,这些简化假设导致实际优化目标退化为过滤后的迭代监督微调(Filtered Iterative SFT),一种在线策略的监督学习变体。在GSM8K和Countdown等基准上,对多种模型家族的实验显示,引入正负样本的Filtered Iterative SFT性能与基于GRPO的训练相当。此外,这些假设间接激励模型生成更长的中间词元序列,从而误导人们认为‘强化学习促进深度思考’,实则源于模型结构设计的副作用。
原文摘要 · Abstract (English)
Reinforcement learning based post-training of large language models (LLMs) has recently gained attention, particularly following the release of DeepSeek R1, which applied GRPO for fine-tuning. Amid the growing claims around improved reasoning abilities attributed to RL post-training, we critically examine the formulation and assumptions underlying these methods. We start by highlighting popular structural assumptions made in modeling LLM training as an MDP, and show how they lead to a degenerate MDP, that characterizes the problem as a contextual bandit, where RL updates naturally collapse into a form of on-policy variant of outcome-driven supervised learning. The two critical structural assumptions include (1) making the MDP states be just a concatenation of the actions with states becoming the context window and the actions becoming the tokens in LLMs and (2) splitting the reward of a state-action trajectory uniformly across the trajectory. Our comprehensive analysis demonstrates that, due to these simplifying assumptions, GRPO objective reduces to filtered Iterative SFT, an on-policy variant of supervised fine-tuning. Our experiments on benchmarks including GSM8K and Countdown, across a diverse set of model families show that Filtered Iterative SFT, incorporating both positive and negative samples, achieves performance comparable to GRPO-based training. We also show that these structural assumptions indirectly incentivize RL to generate longer sequences of intermediate tokens which in turn feeds into the narrative of "RL incentivizing thinking because it generates longer thinking traces."
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。