arXiv:2608.01418cs.AIcs.LG2026-08被引 1

通过归一化前缀概率,提升大模型强化学习中滚动数据的重用效率。

Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning

  • 用几何平均替代累积权重,压缩重要性采样动态范围。
  • 四轮更新下,平均准确率达50.24,比GSPO高3.00个百分点。
  • 适合长期训练中需频繁重用滚动数据的研究者。

自回归式滚动生成是大语言模型强化学习中的主要计算开销。重用每批滚动数据进行多次学习器更新可分摊该成本,但随着学习器偏离行为策略,后续更新越来越偏离策略。在某一标记位置,精确的离策略修正需同时考虑当前动作及其前缀的可达概率。累积重要性比率提供了这种修正,但其乘积形式会产生过大的动态范围。我们研究了前缀归一化策略优化(PNPO),用每个因果前缀上似然比率的几何平均替代累积比率,在保持各位置因果依赖性的同时压缩对数权重尺度。在控制的长上下文数学推理实验中,通过每批滚动数据使用1或4个策略更新周期诱导两种离策略状态。在单周期下,PNPO不始终优于GSPO;但在四周期下,其在各基准测试中均达到最高观测的Avg@32;三个独立基准峰值的未加权平均为50.24,比GSPO高出3.00个百分点。在匹配的2,400次更新预算下,四周期PNPO在150批次后达到最终宏观Avg@32为49.66,与单周期在600批次后达到的49.56相当。这些结果为PNPO在训练远离策略时具有优势提供了初步证据。

原文摘要 · Abstract (English)

Autoregressive rollout generation is a major computational cost in reinforcement learning for large language models. Reusing each rollout batch for additional learner updates amortizes this cost, but later updates become increasingly off-policy as the learner departs from the behavior policy. At a token position, exact off-policy correction must account for both the current action and the probability of reaching its prefix. The cumulative importance ratio provides this correction, but its product form can produce an unwieldy dynamic range. We study Prefix-Normalized Policy Optimization (PNPO), which replaces the cumulative ratio with the geometric mean of likelihood ratios along each causal prefix, preserving causal-prefix dependence at each position while compressing the log-weight scale. In controlled long-context mathematical reasoning experiments, we induce two off-policy regimes by using one or four policy-update epochs per rollout batch. PNPO does not consistently outperform GSPO with one epoch. With four epochs, it attains the highest observed Avg@32 on each benchmark; the unweighted mean of the three independently selected benchmark peaks is 50.24, 3.00 percentage points above GSPO. Under a matched 2,400-update budget, four-epoch PNPO reaches a final macro Avg@32 of 49.66 after 150 rollout batches, comparable to the 49.56 reached after 600 batches with one epoch. These results provide preliminary evidence that PNPO can be advantageous as training moves further off-policy.

强化学习大模型策略优化离策略

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