arXiv:2602.07729cs.LGcs.AI2026-02被引 4

用简单梯度下降训练大模型强化学习,效果不输甚至超越主流优化器。

Do We Need Adam? Surprisingly Strong and Sparse Reinforcement Learning with SGD in LLMs

  • 用SGD替代AdamW,避免复杂参数自适应调整。
  • 仅更新不到0.02%参数,内存效率提升超千倍。
  • 适合资源受限场景,揭示强化学习的高效优化机制。

强化学习(RL),特别是可验证奖励强化学习(RLVR),已成为大语言模型(LLMs)训练的关键阶段,也是当前规模化努力的重点。然而,现有的强化学习优化实践仍沿用下一令牌预测阶段(如预训练和监督微调)的方法,尽管近期研究已指出二者存在根本差异。其中一项常见做法是使用内存开销高的AdamW优化器。我们的分析表明,与监督微调(SFT)相比,AdamW中的动量和自适应学习率在强化学习中影响较小,由此我们提出假设:强化学习对自适应学习率和动量的依赖较低。实验验证了该假设:更节省内存的随机梯度下降(SGD)在强化学习中表现媲美甚至优于AdamW,且无需任何稀疏性正则化,全量微调仅更新少于0.02%的模型参数,比AdamW减少1000倍以上。我们分析了这一更新稀疏性的潜在原因。这些发现为大模型强化学习的优化动态提供了新见解,表明其参数效率远高于以往认知。

原文摘要 · Abstract (English)

Reinforcement learning (RL), particularly RL from verifiable reward (RLVR), has become a crucial phase of training large language models (LLMs) and a key focus of current scaling efforts. However, optimization practices in RL largely follow those of next-token prediction stages (e.g., pretraining and supervised fine-tuning), despite fundamental differences between RL and these stages highlighted by recent work. One such practice is the use of the AdamW optimizer, which is widely adopted for training large-scale transformers despite its high memory overhead. Our analysis shows that both momentum and adaptive learning rates in AdamW are less influential in RL than in SFT, leading us to hypothesize that RL benefits less from Adam-style per-parameter adaptive learning rates and momentum. Confirming this hypothesis, our experiments demonstrate that the substantially more memory-efficient SGD, which is known to perform poorly in supervised learning of large-scale transformers, matches or even outperforms AdamW in RL for LLMs. Remarkably, full fine-tuning with SGD updates fewer than 0.02% of model parameters without any sparsity-promoting regularization, more than 1000 times fewer than AdamW. Our analysis offers potential reasons for this update sparsity. These findings provide new insights into the optimization dynamics of RL in LLMs and show that RL can be substantially more parameter-efficient than previously recognized.

强化学习优化器参数效率SGD

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