mlsanityData to 14 Oct 2026

Muon Sublates the Edge of Stability in LLM Pretraining

arXiv 2609.34915 ↗First posted 28 Sep 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Muon is increasingly used for language-model pretraining, yet its large-step dynamics are not captured by the classical edge-of-stability (EoS) picture of gradient descent (GD). In GD, loss neutrality, equal-magnitude update reversal, and marginal stability meet at a single learning-rate-dependent edge. We show that Muon breaks this coupling. For stochastic no-momentum Muon, we derive a coherence-corrected conditional loss-neutral boundary 2ρb/η, while temporal alignment follows a separate geometry. Controlled experiments show that loss balance and temporal alignment respond differently to learning rate and batch size. Across our language model experiments, the 130M Llama-like LLM runs exhibit loss-boundary tracking with weak negative alignment, whereas the studied 1B LLM configuration shows stronger partial cancellation; in both settings, directions remain far from coherent reversal while training continues to improve. These results support a split EoS picture for Muon: a stochastic loss-neutral edge survives, but it is not accompanied by a universal temporal-direction signature. The source code for reproducing the experiments can be found in https://github.com/cyzebra/Muon-Sublates-the-Edge-of-Stability-in-LLM-Pretraining

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 27 references that are in the map. First posted on arXiv 28 Sep 2026.

Builds on

23 of 27 references.

References · 23

  1. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  2. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
  3. 2025Muon is Scalable for LLM Trainingcited by 428
  4. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
  5. 2020The large learning rate phase of deep learning: the catapult mechanismcited by 216
  6. 2024Old Optimizer, New Norm: An Anthologycited by 193
  7. 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 155
  8. 2020The Break-Even Point on Optimization Trajectories of Deep Neural Networkscited by 150
  9. 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 101
  10. 2018On the Relation Between the Sharpest Directions of DNN Loss and the SGD Step Lengthcited by 88
  11. 2022Understanding the unstable convergence of gradient descentcited by 86
  12. 2025Muon Optimizes Under Spectral Norm Constraintscited by 86
11 more
  1. 2022Analyzing Sharpness along GD Trajectory: Progressive Sharpening and Edge of Stabilitycited by 63
  2. 2025Muon Outperforms Adam in Tail-End Associative Memory Learningcited by 46
  3. 2022Beyond the Edge of Stability via Two-step Gradient Updatescited by 38
  4. 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 35
  5. 2024Edge of Stochastic Stability: Revisiting the Edge of Stability for SGDcited by 30
  6. 2023Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theorycited by 26
  7. 2024Directional Smoothness and Gradient Methods: Convergence and Adaptivitycited by 17
  8. 2026A Defense of the Quadratic Modelcited by 14
  9. 2026Non-Euclidean Gradient Descent Operates at the Edge of Stabilitycited by 9
  10. 2026A Scalable Measure of Loss Landscape Curvature for Analyzing the Training Dynamics of LLMscited by 8
  11. 2026The Origin of Edge of Stabilitycited by 6

Built on by

Not cited yet.