When Does Momentum Help in Reinforcement Learning with Verifiable Rewards?
Not on the map. Why?
Abstract (arXiv)
Momentum is a classical tool for accelerating convex optimization and a standard component of modern large language model training. Yet recent empirical work has questioned the usefulness of first-moment momentum in reinforcement learning with verifiable rewards (RLVR). We study this question through a bandit formulation of RLVR with bias-corrected exponential moving average (EMA) momentum, where the EMA coefficient β controls how quickly the weights on past gradients decay. When policy gradients are evaluated exactly, we prove that increasing β within a range that prevents overshoot strictly accelerates optimization. When gradients are estimated from sampled responses, a positive expected gain can be obscured by sampling fluctuations. We show that as the learning rate η decreases, the expected gain with fixed β vanishes faster than these fluctuations, whereas increasing β to keep η/(1-β) within the safe regime preserves a nonvanishing gain that eventually dominates them. Bandit simulations quantitatively validate these predictions. RLVR experiments on Qwen3-4B-Base using SGD with EMA and AdamW support our principle of choosing larger coefficients in practical LLM post-training. On Qwen2.5-32B with AdamW, simply increasing β1 improves mean validation accuracy by nearly 2 percentage points over the conventional choice, demonstrating the principle's effectiveness at a larger model scale.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 35 of its 44 references that are in the map. First posted on arXiv 14 Oct 2026.
Builds on
35 of 44 references.
References · 35
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2015Trust Region Policy Optimizationcited by 2,580
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,150
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
23 more
- 2023Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-Constraintcited by 350
- 2019Provably Efficient Exploration in Policy Optimizationcited by 205
- 2020Fast Global Convergence of Natural Policy Gradient Methods with Entropy Regularizationcited by 176
- 2024DPO Meets PPO: Reinforced Token Optimization for RLHFcited by 127
- 2019Adaptive Trust Region Policy Optimization: Global Convergence and Faster Rates for Regularized MDPscited by 118
- 2014Global convergence of the Heavy-ball method for convex optimizationcited by 61
- 2024The AdEMAMix Optimizer: Better, Faster, Oldercited by 54
- 2020Momentum-Based Policy Gradient Methodscited by 30
- 2023A Theoretical Analysis of Optimistic Proximal Policy Optimization in Linear Markov Decision Processescited by 29
- 2021On the Global Optimum Convergence of Momentum-based Policy Gradientcited by 26
- 2016On the Influence of Momentum Acceleration on Online Learningcited by 24
- 2023The Marginal Value of Momentum for Small Learning Rate SGDcited by 18
- 2021Bregman Gradient Policy Optimizationcited by 17
- 2026Logarithmic-time Schedules for Scaling Language Models with Momentumcited by 9
- 2026Rethinking Muon Beyond Pretraining: Spectral Failures and High-Pass Remedies for VLA and RLVRcited by 8
- 2026Do We Need Adam? Surprisingly Strong and Sparse Reinforcement Learning with SGD in LLMscited by 7
- 2024Global Convergence of Natural Policy Gradient with Hessian-aided Momentum Variance Reductioncited by 6
- 2025AdamS: Momentum Itself Can Be A Normalizer for LLM Pretraining and Post-trainingcited by 6
- 2026Pion: A Spectrum-Preserving Optimizer via Orthogonal Equivalence Transformationcited by 6
- 2023Accelerated Policy Gradient: On the Convergence Rates of the Nesterov Momentum for Reinforcement Learningcited by 4
- 2026On the Hidden Objective Biases of Group-based Reinforcement Learningcited by 3
- 2026ISO: An RLVR-Native Optimization Stackcited by 2
- 2026Parameter Exploration for RLVR via Variational Learningcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Understanding Enrichment in Reinforcement Learning
- 2026Rate or Fate? RLV^varepsilonR: Reinforcement Learning with Verifiable Noisy Rewards
- 2025Reinforcement Learning with Verifiable yet Noisy Rewards under Imperfect Verifiers
- 2025On the optimization dynamics of RLVR: Gradient gap and step size thresholds
- 2025Variance-Aware Baselines and Adaptive Learning Rates for Reinforcement Learning with Verifiable Rewards
- 2026Verifier Errors in RLVR: Reward Hacking, Limits of Feedback, and Selective Control
- 2026What are Key Factors for Updates in RL for LLM Reasoning?
- 2026Multi-Step Likelihood-Ratio Correction for Reinforcement Learning with Verifiable Rewards