Conda: Column-Normalized Adam for Training Large Language Models Faster
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) have demonstrated impressive generalization and emergent capabilities, yet their pre-training remains computationally expensive and sensitive to optimization dynamics. While Adam-based optimizers offer fast convergence by adapting learning rates coordinate-wise, recent studies reveal that their updates often suffer from poor spectral conditioning and low-rank structures, hindering efficiency. Muon addresses this issue via global spectral normalization but lacks the per-coordinate adaptivity of Adam. In this work, we propose Column-Normalized Adam (Conda), a novel optimizer that bridges the strengths of both approaches. Conda projects updates into an orthogonal subspace and applies column-wise second moment normalization based on the projected gradients, thereby achieving both improved spectral conditioning and maintaining coordinate-wise adaptivity. This design alleviates the spectral pathologies of Adam while preserving its fast convergence behavior. Extensive experiments on the LLaMA and GPT-2 series show that Conda consistently outperforms AdamW, Muon, and other baselines in pre-training. Remarkably, on the LLaMA series, Conda achieves 2-2.5 the convergence speed of AdamW, measured in both training steps and training time. Further ablations demonstrate its robustness under diverse training setups. These results collectively highlight Conda as an effective and broadly applicable optimizer for large-scale LLM training. The code is released on https://github.com/jie040109/Conda
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 56 of its 78 references that are in the map. First posted on arXiv 29 Sep 2025.
Builds on
56 of 78 references.
References · 56
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2023GPT-4 Technical Reportcited by 18,396
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2014Going Deeper with Convolutionscited by 4,241
38 more
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2022A ConvNet for the 2020scited by 3,579
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2017Crowdsourcing Multiple Choice Science Questionscited by 765
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 758
- 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 674
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 662
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 463
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 2023LLM-Adapters: An Adapter Family for Parameter-Efficient Fine-Tuning of Large Language Modelscited by 324
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 265
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 174
- 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 137
- 2020Towards Theoretically Understanding Why SGD Generalizes Better Than ADAM in Deep Learningcited by 126
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
- 2023A Spectral Condition for Feature Learningcited by 109
- 2022Same Pre-training Loss, Better Downstream: Implicit Bias Matters for Language Modelscited by 84
- 2023Compressing LLMs: The Truth is Rarely Pure and Never Simplecited by 62
- 2023Training Transformers with 4-bit Integerscited by 60
- 2025Overtrained Language Models Are Harder to Fine-Tunecited by 58
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 57
- 2024APOLLO: SGD-like Memory, AdamW-level Performancecited by 54
- 2024Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?cited by 43
- 2025Structured Preconditioners in Adaptive Optimization: A Unified Analysiscited by 33
- 2025COSMOS: A Hybrid Adaptive Optimizer for Memory-Efficient Training of LLMscited by 30
- 2021ZerO Initialization: Initializing Neural Networks with only Zeros and Onescited by 28
- 2025SPAM: Spike-Aware Adam with Momentum Reset for Stable LLM Trainingcited by 28
- 2024Zeroth-Order Fine-Tuning of LLMs in Random Subspacescited by 26
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfercited by 12
- 2026Enhancing LLM Training via Spectral Clippingcited by 10
- 2026Canzona: A Unified, Asynchronous, and Load-Balanced Framework for Distributed Matrix-based Optimizerscited by 4
- 2026Convergence of Spectral Descent for Non-smooth Optimizationcited by 2
- 2026OmniOpt: Taxonomy, Geometry, and Benchmarking of Modern Optimizerscited by 0
- 2026OmniOpt: Taxonomy, Geometry, and Benchmarking of Modern Optimizers4 Jul 2026
- 2026Convergence of Spectral Descent for Non-smooth Optimization26 May 2026
- 2026Enhancing LLM Training via Spectral Clipping15 Mar 2026
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfer10 Mar 2026
- 2026Canzona: A Unified, Asynchronous, and Load-Balanced Framework for Distributed Matrix-based Optimizers4 Feb 2026