mlsanityData to 7 Oct 2026

When Does Muon Help Agentic Reinforcement Learning?

arXiv 2607.16169 ↗First posted 17 Jul 2026cs.LG cs.AIcited by 0

On the map: LLM agents and tool use → Reinforcement learning for LLM agents

Abstract (arXiv)

Muon is competitive with AdamW in large-scale pre-training, but its operating regime in reinforcement-learning post-training remains unclear. We map this regime on ALFWorld, a sparse-reward agentic benchmark, using three group-based objectives and Qwen2.5 models from 0.5B to 3B. Under a shared KL and clipping recipe, matched optimizer comparisons and AdamW rate controls trace the usable step-size range. AdamW responds non-monotonically to rate, whereas fan-in Muon remains stable at a more aggressive effective step: at 3 × 10-5 it improves late success over an AdamW 10-6 baseline after correction across rate-metric tests. Its normalized-AUC effect is directionally positive but less uniform; the heuristic-matched lower-rate effect is less consistent, and tuned AdamW nearly matches high-rate Muon at 3B GraphGPO. High-rate Muon applies 3.53 × AdamW's hidden-matrix update RMS; a full-budget RMS-matched control removes the late-success gain. Together, these results identify a recipe-level operating regime in which fan-in Muon supports a more aggressive stable effective step under shared KL and clipping: the margin is largest when optimization headroom remains and contracts near saturation, after AdamW tuning, or under magnitude matching. The scale-matched control ties this spectral effect to Muon's scale convention rather than establishing a universal optimizer ranking. Code is available at https://github.com/x66ccff/verl-muon.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 19 of its 30 references that are in the map. First posted on arXiv 17 Jul 2026.

Builds on

19 of 30 references.

Closest ancestors · 4

Same direction, newest first

  1. 2026CRAFT: Counterfactual Credit Assignment from Free Sibling Rollouts for Self-Distilled Agentic Reinforcement Learning28 Jun
  2. 2026Semantic Consistency Policy Optimization for Reinforcement Learning of LLM Agents24 Jun
  3. 2026Beyond Trajectory-Level Attribution: Graph-Based Credit Assignment for Agentic Reinforcement Learning26 May
  4. 2025Group-in-Group Policy Optimization for LLM Agent Training16 May

Other references · 15

  1. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  2. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  3. 2024Qwen2.5 Technical Reportcited by 4,713
  4. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  5. 2022WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agentscited by 1,197
  6. 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,143
  7. 2025Muon is Scalable for LLM Trainingcited by 381
  8. 2025Muon Optimizes Under Spectral Norm Constraintscited by 77
  9. 2026Preconditioning Benefits of Spectral Orthogonalization in Muoncited by 28
  10. 2026Adam Improves Muon: Adaptive Moment Estimation with Orthogonalized Momentumcited by 13
  11. 2026HTMuon: Improving Muon via Heavy-Tailed Spectral Correctioncited by 11
  12. 2026Can Muon Fine-tune Adam-Pretrained Models?cited by 6
3 more
  1. 2026Rethinking Muon Beyond Pretraining: Spectral Failures and High-Pass Remedies for VLA and RLVRcited by 5
  2. 2026Optimizer-Model Consistency: Full Finetuning with the Same Optimizer as Pretraining Forgets Lesscited by 3
  3. 2026MiMuon: Mixed Muon Optimizer with Improved Generalization for Large Modelscited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Muon Meets Mamba: Spectral Optimization for State Space Models
  2. 2025Muon: Training and Trade-offs with Latent Attention and MoE
  3. 2025Kimi K2: Open Agentic Intelligence
  4. 2026The Dark Room in the Reward Channel: Dense Prediction Rewards Collapse GRPO-Trained LLM Agents -- and The Channel, Not the Content, Decides What Works
  5. 2026Neglected Free Lunch from Post-training: Progress Advantage for LLM Agents
  6. 2025Xmodel-2.5: 1.3B Data-Efficient Reasoning SLM
  7. 2026Move on Muon : A Hamiltonian probability gradient flow perspective of Muon optimizer
  8. 2026Muon2: Boosting Muon via Adaptive Second-Moment Preconditioning