通过分层注入可学习扰动,缓解大语言模型强化学习中的策略僵化问题。
Adaptive Layerwise Perturbation: Unifying Off-Policy Corrections for LLM RL
- 在每一层隐藏状态中注入小的可学习扰动,作为重要性采样分子
- 实验显示能有效抑制重要性比率尾部爆炸和KL值突增
- 适合需要稳定训练与深度探索的大模型强化学习场景
大语言模型强化学习中,策略僵化和训练-推理不一致成为训练稳定性和探索能力的主要瓶颈。由于提升推理效率的技术,推断策略与更新后策略间的分布差距扩大,导致重要性比率呈现重尾现象。当策略局部尖锐时,重尾比率会放大梯度,使更新超出信任区域。为此,我们提出自适应分层扰动(ALP),在更新过程中向每层输入隐藏状态注入小的可学习扰动,并以该扰动后的策略作为重要性比率的分子,与未改变的推断策略对比。通过在中间表示中引入可控噪声,ALP防止更新策略偏离推断策略过远,扩展策略族以覆盖推断时的偏差噪声,从而自然缩小更新与推断策略间的差距,降低重要性比率尾部,维持训练稳定。实验表明,在单轮数学和多轮工具集成推理任务上,ALP不仅提升最终性能,还避免了重要性比率尾部爆炸和迭代训练中的KL突增,同时增强探索能力。消融实验证明,全层表示级扰动最有效,显著优于部分层或仅在逻辑输出层扰动的变体。
原文摘要 · Abstract (English)
Off-policy problems such as policy staleness and training--inference mismatch have become a major bottleneck for training stability and further exploration in LLM RL. The distribution gap between the inference and updated policies grows because of the techniques to enhance inference efficiency, leading to heavy-tailed importance ratios. Heavy-tailed ratios arise when the policy is locally sharp, which further inflates gradients and can push updates outside the trust region. To address this, we propose Adaptive Layerwise Perturbation (ALP), which injects small learnable perturbations into the input hidden states of each layer during updates and uses the resulting perturbed policy as the numerator of the importance ratio against the unchanged inference policy in the objective. Intuitively, by adding controlled noise to intermediate representations, ALP prevents the updated policy from deviating too sharply from the inference policy and enlarges the policy family to cover inference-time mismatch noise. Hence, the flattened distribution can naturally tighten the gap between the updated and inference policies and reduce the tail of importance ratios, thus maintaining training stability. This is further validated empirically. Experiments on single-turn math and multi-turn tool-integrated reasoning tasks show that ALP not only improves final performance, but also avoids blow-up in the importance-ratio tail and KL spikes during iterative training, along with boosted exploration. Ablations show that representation-level perturbations across all layers are most effective, substantially outperforming partial-layer and logits-only variants.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。