Navigating LLM Valley: From AdamW to Memory-Efficient and Matrix-Based Optimizers
Not on the map. Why?
Abstract (arXiv)
Training large language models requires optimization algorithms that are not only statistically effective, but also computationally and memory efficient at extreme scale. Although Adam remains the dominant optimizer for large-scale language-model pretraining and fine-tuning, recent work has revisited nearly every component of the optimization stack: adaptive moment estimation, decoupled weight decay, memory footprint, curvature approximation, sign-based updates, large-batch stability, low-rank gradient structure, and matrix-wise orthogonalized updates. This survey reviews optimizer design for large language models through a systems-and-optimization lens. We organize the literature into classical first-order optimizers, adaptive optimizers, memory-efficient variants, second-order and curvature-aware methods, sign-based and discovered optimizers, low-rank and projection-based methods, and matrix-based optimizers such as Muon. We also discuss benchmarking methodology, including hyperparameter fairness, scale dependence, wall-clock efficiency, token efficiency, memory overhead, and downstream evaluation. We argue that optimizer research for LLMs is entering a new phase: moving from single-algorithm speedup claims toward rigorous, scale-aware comparisons that jointly evaluate convergence, stability, memory, and implementation complexity.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 28 of its 54 references that are in the map. First posted on arXiv 9 May 2026.
Builds on
28 of 54 references.
References · 28
- 2017Attention Is All You Needcited by 60,361
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2023Mistral 7Bcited by 3,164
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2016An overview of gradient descent optimization algorithmscited by 1,033
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 632
16 more
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2018Horovod: fast and easy distributed deep learning in TensorFlowcited by 279
- 2019A Survey of Optimization Methods from a Machine Learning Perspectivecited by 91
- 2020Practical Quasi-Newton Methods for Training Deep Neural Networkscited by 68
- 2025Benchmarking Optimizers for Large Language Model Pretrainingcited by 68
- 2023A Theory on Adam Instability in Large-Scale Machine Learningcited by 56
- 2024APOLLO: SGD-like Memory, AdamW-level Performancecited by 54
- 2023A Survey on Efficient Training of Transformerscited by 39
- 2026Convergence of Muon with Newton-Schulzcited by 36
- 2020Scalable Second Order Optimization for Deep Learningcited by 21
- 2019Adaptive Loss Scaling for Mixed Precision Trainingcited by 7
- 2025Gradient Heterogeneity Complements Hessian Heterogeneity in Transformer Optimizationcited by 7
- 2023mL-BFGS: A Momentum-based L-BFGS for Distributed Large-Scale Neural Network Optimizationcited by 4
- 2023Stabilizing Backpropagation in 16-bit Neural Training with Modified Adam Optimizercited by 2
- 2025Symmetric Rank-One Quasi-Newton Methods for Deep Learning Using Cubic Regularizationcited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.