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