mlsanityData to 14 Oct 2026

Optimizer Memory Schedules for Outscaling the Overtraining Axis

arXiv 2609.04577 ↗First posted 4 Sep 2026cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

We investigate how optimizers scale across the overtraining axis and show that relative optimizer performance and optimal hyperparameters change substantially with training horizon. In particular, we study how matrix-preconditioned methods (Muon and SOAP) and a momentum-scheduled method (ADANA) scale relative to AdamW. We compare these four optimizers across models from 51M to 253M parameters and overtraining (OT) factors from 1x to 256x, sweeping the base learning rate at every setting. The preferred learning rate schedule can reverse across the overtraining axis, the best weight decay coefficient scales approximately as sqrt(OT), and longer horizons generally favor longer fixed memory. ADANA's scaling advantage over AdamW persists after tuning AdamW's fixed memory separately at each horizon. Log-time weight decay and momentum cooldown provide substantial gains for ADANA that compound as training increases. With this treatment, ADANA outscales AdamW with an exponent advantage close to that predicted by DANA theory on power-law random features. Muon and SOAP instead provide roughly constant token-efficiency advantages over AdamW across most of the measured range, although SOAP may gain further at the highest overtraining factors. ADANA begins behind both matrix-preconditioned optimizers but closes the gaps as training increases, surpassing Muon and becoming competitive with SOAP at our highest OT factors. These results establish training horizon as an essential axis for optimizer evaluation and design.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 43 of its 62 references that are in the map. First posted on arXiv 4 Sep 2026.

Builds on

43 of 62 references.

References · 43

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  3. 2020Scaling Laws for Neural Language Modelscited by 6,586
  4. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  5. 2016Training Deep Nets with Sublinear Memory Costcited by 848
  6. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  7. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
  8. 2025Muon is Scalable for LLM Trainingcited by 428
  9. 2023Scaling Data-Constrained Language Modelscited by 403
  10. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
  11. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 170
  12. 2024The Road Less Scheduledcited by 163
31 more
  1. 2023Beyond Chinchilla-Optimal: Accounting for Inference in Language Model Scaling Lawscited by 159
  2. 2024Language models scale reliably with over-training and on downstream taskscited by 113
  3. 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
  4. 2025Practical Efficiency of Muon for Pretrainingcited by 80
  5. 2025Don't be lazy: CompleteP enables compute-efficient deep transformerscited by 78
  6. 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 77
  7. 2025Benchmarking Optimizers for Large Language Model Pretrainingcited by 68
  8. 2024A New Perspective on Shampoo's Preconditionercited by 59
  9. 2024The AdEMAMix Optimizer: Better, Faster, Oldercited by 54
  10. 2024How to set AdamW's weight decay as you scale model and dataset sizecited by 43
  11. 2025Small Batch Size Training for Language Models: When Vanilla SGD Works, and Why Gradient Accumulation Is Wastefulcited by 41
  12. 2025Straight to Zero: Why Linearly Decaying the Learning Rate to Zero Works Best for LLMscited by 35
  13. 2025Communication-Efficient Language Model Training Scales Reliably and Robustly: Scaling Laws for DiLoCocited by 27
  14. 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 27
  15. 2020Scalable Second Order Optimization for Deep Learningcited by 21
  16. 2025Hyperparameter Transfer Enables Consistent Gains of Matrix-Preconditioned Optimizers Across Scalescited by 21
  17. 2020LaProp: Separating Momentum and Adaptivity in Adamcited by 20
  18. 2025Scaling Inference-Efficient Language Modelscited by 16
  19. 2026Fantastic Pretraining Optimizers and Where to Find Them II: Hyperball Optimizationcited by 16
  20. 2025Dimension-adapted Momentum Outscales SGDcited by 12
  21. 2025Apertus: Democratizing Open and Compliant LLMs for Global Language Environmentscited by 12
  22. 2025Connections between Schedule-Free Optimizers, AdEMAMix, and Accelerated SGD Variantscited by 10
  23. 2026Deriving Hyperparameter Scaling Laws via Modern Optimization Theorycited by 10
  24. 2025Scaling with Collapse: Efficient and Predictable Training of LLM Familiescited by 9
  25. 2026Logarithmic-time Schedules for Scaling Language Models with Momentumcited by 9
  26. 2026ScheduleFree+: Scaling Learning-Rate-Free & Schedule-Free Learning to Large Language Modelscited by 7
  27. 2026Towards Robust Scaling Laws for Optimizerscited by 6
  28. 2026SOAP, Muon, and Beyond: Pushing LLM Pretraining Scalescited by 6
  29. 2026Phases of Muon: When Muon Eclipses SignSGDcited by 3
  30. 2026Skaling: Chinchilla's Exponents Meet Kaplan's Couplingcited by 3
  31. 2025Predicting Training Re-evaluation Curves Enables Effective Data Curriculums for LLMscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.