arXiv:2601.18795cs.LGcs.AI2026-01被引 13

用旧采样结果加速大模型强化学习,让难问题训练快3倍。

Reuse your FLOPs: Scaling RL on Hard Problems by Conditioning on Very Off-Policy Prefixes

  • 用成功路径前缀做条件,结合在线策略学习,避免离线更新不稳。
  • 在难题上训练速度提升2倍,最终奖励提高3倍,且跨任务泛化好。
  • 适合想高效训练大模型推理能力的研究者和工程师。

传统大模型强化学习在难题上浪费算力,因正确轨迹稀少、策略梯度消失而停滞。本文提出PrefixRL,复用先前采样产生的离线轨迹前缀,以这些前缀为条件进行在线策略学习,避开离线学习的不稳定性。该方法通过调节前缀长度动态控制问题难度,使学习信号更强。理论证明其目标与标准强化学习一致且样本更高效。实验发现:仅在带前缀的问题上训练,能泛化到无前缀的分布外任务,学习策略常与前缀不同。我们使用基础模型通过拒绝采样生成离线轨迹,形成自进化闭环。在难题上,即使计入初始采样开销,PrefixRL仍比最强基线(离线数据SFT+RL)快2倍达到相同奖励,并将最终奖励提升3倍。性能增益可迁移至未见基准,且对来自不同模型家族的离线轨迹仍有效,验证其实际适用性。

原文摘要 · Abstract (English)

Typical reinforcement learning (RL) methods for LLM reasoning waste compute on hard problems, where correct on-policy traces are rare, policy gradients vanish, and learning stalls. To bootstrap more efficient RL, we consider reusing old sampling FLOPs (from prior inference or RL training) in the form of off-policy traces. Standard off-policy methods supervise against off-policy data, causing instabilities during RL optimization. We introduce PrefixRL, where we condition on the prefix of successful off-policy traces and run on-policy RL to complete them, side-stepping off-policy instabilities. PrefixRL boosts the learning signal on hard problems by modulating the difficulty of the problem through the off-policy prefix length. We prove that the PrefixRL objective is not only consistent with the standard RL objective but also more sample efficient. Empirically, we discover back-generalization: training only on prefixed problems generalizes to out-of-distribution unprefixed performance, with learned strategies often differing from those in the prefix. In our experiments, we source the off-policy traces by rejection sampling with the base model, creating a self-improvement loop. On hard reasoning problems, PrefixRL reaches the same training reward 2x faster than the strongest baseline (SFT on off-policy data then RL), even after accounting for the compute spent on the initial rejection sampling, and increases the final reward by 3x. The gains transfer to held-out benchmarks, and PrefixRL is still effective when off-policy traces are derived from a different model family, validating its flexibility in practical settings.

强化学习大模型推理自进化高效训练

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