When Does Momentum Help in Reinforcement Learning with Verifiable Rewards?
可验证奖励强化学习中动量何时有效
🧭 LLM 强化学习优化器动量策略 ★ 深度解读
Peking University · Shanghai Jiao Tong University · SIMIS · Stanford University
证明 RLVR 中固定动量增益会被采样噪声淹没,需随学习率放大 β 才能稳定获益
📌 工作参考价值(3 点)
- 在 RLVR/GRPO 类 LLM post-training 里,AdamW 的 β1 别只留默认 0.9;可以尝试 β1=1-1e-6 甚至 1-1e-9,同时观察 reward/准确率是否稳定提升。
- 如果想降学习率换取更稳训练,应同步放大 β,使 η/(1-β) 维持在安全区间,避免固定 β 导致动量增益被采样噪声淹没。
- 做带采样奖励的在线策略优化(如 Agent 任务、推荐/广告的 policy gradient)时,若发现 momentum 无效,先别否定动量,检查 lr 与 β 的比值是否过小;用少数 seed 很难区分固定 β 和 no-EMA,建议多 seed 或看 SNR。