arXiv:2607.04728cs.CLcs.AI2026-07

提出SIS方法,让离策略令牌变同策略,提升大模型对齐训练稳定性。

Turning Off-Policy Tokens On-Policy: A Plug-in Approach for Improving LLM Alignment

论文配图:Turning Off-Policy Tokens On-Policy: A Plug-in Approach for Improving LLM Alignment
图 1 · 摘自论文原文
  • 用拒绝采样思想筛选离策略令牌,只保留同策略样本
  • 在数学与智能体基准上均提升所有指标,抗离策略数据能力更强
  • 无需修改算法框架,可无缝接入各类强化学习训练流程

大语言模型强化学习后训练通常采用“滚动然后更新”范式,导致生成离策略数据。虽有重要性采样(IS)缓解,但词元级重要性比率在长序列中累积引发方差爆炸。本文提出选择性重要性采样(SIS),受拒绝采样启发:将离策略模型视为提议分布,对每个词元进行接受测试——被接受的视为同策略,赋予单位重要性权重;被拒绝的仍使用标准IS修正。理论证明SIS能缩小词元级与序列级离策略梯度估计差距。SIS作为即插即用模块,仅修改策略损失中的重要性比率,计算开销极小,兼容多种强化学习算法。在密集模型与MoE模型、数学与智能体基准上的实验表明,SIS在所有任务中持续提升性能,并显著增强对离策略数据的鲁棒性。

原文摘要 · Abstract (English)

Reinforcement learning (RL) post-training for large language models (LLMs) follows a efficient paradigm of "rollout then update", which inevitably results in off-policy training data. To resolve this, Importance sampling (IS) is proposed, while the token-level ratios compound over long sequences, causing severe variance exploded. A natural idea is "transferring" these off-policy token into on-policy token, so that the importance scores for correction are unnecessary. Following this idea, we propose Selective Importance Sampling (SIS), which is inspired by rejection sampling. Concretely, SIS implements by viewing off-policy model as proposal distribution, and implement a token-level rejection test: accepted tokens are viewed as on-policy, so that receive unit importance score, while rejected tokens retain the standard IS correction. Our proposed SIS is theoretically proved reducing the gap between token-level and sequence-level off-policy gradient estimators. The SIS acts as a plug-in that only modifies the importance ratio in the policy loss, adding negligible wall-clock overhead, and can be combine with a vast vary of RL post-training algorithms. Experiments on dense and MoE LLMs across math and agent benchmarks show that SIS consistently improves all objectives, while providing substantially stronger robustness under off-policy data.

强化学习大模型对齐重要性采样离策略训练

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