Preconditioned Norms: A Unified Framework for Steepest Descent, Quasi-Newton and Adaptive Methods
Not on the map. Why?
Abstract (arXiv)
Optimization lies at the core of modern deep learning, yet existing methods often face a fundamental trade-off between adapting to problem geometry and leveraging curvature utilization. Steepest descent algorithms adapt to different geometries through norm choices but remain strictly first-order, whereas quasi-Newton and adaptive optimizers incorporate curvature information but are restricted to Frobenius geometry, limiting their applicability across diverse architectures. In this work, we propose a unified framework generalizing steepest descent, quasi-Newton methods, and adaptive methods through the novel notion of preconditioned matrix norms. This abstraction reveals that widely used optimizers such as SGD and Adam, as well as more advanced approaches like Muon and KL-Shampoo, and recent hybrids including SOAP and SPlus, all emerge as special cases of the same principle. Within this framework, we provide the first systematic treatment of affine and scale invariance in the matrix-parameterized setting, establishing necessary and sufficient conditions under generalized norms. Building on this foundation, we introduce two new methods, textttMuAdam and textttMuAdam-SANIA, which combine the spectral geometry of Muon with Adam-style preconditioning. Our experiments demonstrate that these optimizers are competitive with, and in some cases outperform, existing state-of-the-art methods. Our code is available at https://github.com/brain-lab-research/LIB/tree/quasidescent
Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Builds on 34 of its 70 references that are in the map. First posted on arXiv 12 Oct 2025.
Builds on
34 of 70 references.
References · 34
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 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
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 2014The Loss Surfaces of Multilayer Networkscited by 280
- 2024Old Optimizer, New Norm: An Anthologycited by 193
22 more
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
- 2025The Polar Express: Optimal Matrix Sign Methods and Their Application to the Muon Algorithmcited by 110
- 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
- 2024Modular Duality in Deep Learningcited by 77
- 2015Beyond Convexity: Stochastic Quasi-Convex Optimizationcited by 76
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
- 2020Practical Quasi-Newton Methods for Training Deep Neural Networkscited by 68
- 2025Benchmarking Optimizers for Large Language Model Pretrainingcited by 68
- 2025Structured Preconditioners in Adaptive Optimization: A Unified Analysiscited by 33
- 2024LoRA Done RITE: Robust Invariant Transformation Equilibration for LoRA Optimizationcited by 32
- 2021Adaptivity without Compromise: A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic Optimizationcited by 28
- 2025An Exploration of Non-Euclidean Gradient Descent: Muon and its Many Variantscited by 28
- 2025A Stable Whitening Optimizer for Efficient Neural Network Trainingcited by 18
- 2025Understanding and Improving Shampoo and SOAP via Kullback-Leibler Minimizationcited by 16
- 2025Unveiling the Basin-Like Loss Landscape in Large Language Modelscited by 13
- 2025On The Concurrence of Layer-wise Preconditioning Methods and Provable Feature Learningcited by 12
- 2024Gradient Methods with Online Scalingcited by 11
- 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGradcited by 8
- 2023SANIA: Polyak-type Optimization Framework Leads to Scale Invariant Stochastic Algorithmscited by 5
- 2025A Survey on Memory-Efficient Transformer-Based Model Training in AI for Sciencecited by 3
- 2024Quasi-Newton method of Optimization is proved to be a steepest descent method under the ellipsoid normcited by 1
Built on by
Citing papers per year: 2026: 11.
11 papers.
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfercited by 12
- 2026Symmetry-Compatible Principle for Optimizer Design: Embeddings, LM Heads, SwiGLU MLPs, and MoE Routerscited by 11
- 2026Decoupling Variance and Scale-Invariant Updates in Adaptive Gradient Descent for Unified Vector and Matrix Optimizationcited by 4
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Trainingcited by 2
- 2026Convergence Rate Analysis of the AdamW-style Shampoo: Unifying One-Sided and Two-Sided Preconditioningcited by 1
- 2026Beyond the Matrix Sign: Quadratic Spectral Descentcited by 1
- 2026From SGD to Muon: Adaptive Optimization via Schatten-p Normscited by 0
- 2026Optimizers for Diffusion Models: A Controlled Benchmarkcited by 0
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Trainingcited by 0
- 2026Optimizing Large Language Models with Chained LMOscited by 0
- 2026Which Geometry on Which Layer? Measured Sign-Spectral Assignment in Mixed-Optimizer Trainingcited by 0
- 2026Which Geometry on Which Layer? Measured Sign-Spectral Assignment in Mixed-Optimizer Training12 Oct 2026
- 2026Optimizing Large Language Models with Chained LMOs7 Oct 2026
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Training28 Sep 2026
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2026Beyond the Matrix Sign: Quadratic Spectral Descent7 Sep 2026
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Training19 May 2026
- 2026From SGD to Muon: Adaptive Optimization via Schatten-p Norms19 May 2026
- 2026Symmetry-Compatible Principle for Optimizer Design: Embeddings, LM Heads, SwiGLU MLPs, and MoE Routers18 May 2026
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfer10 Mar 2026
- 2026Decoupling Variance and Scale-Invariant Updates in Adaptive Gradient Descent for Unified Vector and Matrix Optimization6 Feb 2026
- 2026Convergence Rate Analysis of the AdamW-style Shampoo: Unifying One-Sided and Two-Sided Preconditioning12 Jan 2026