mlsanityData to 14 Oct 2026

Preconditioned Norms: A Unified Framework for Steepest Descent, Quasi-Newton and Adaptive Methods

arXiv 2510.10777 ↗First posted 12 Oct 2025cs.LG math.OCcited by 11

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  4. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  5. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  6. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  7. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
  8. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
  9. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
  10. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
  11. 2014The Loss Surfaces of Multilayer Networkscited by 280
  12. 2024Old Optimizer, New Norm: An Anthologycited by 193
22 more
  1. 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
  2. 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
  3. 2025The Polar Express: Optimal Matrix Sign Methods and Their Application to the Muon Algorithmcited by 110
  4. 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
  5. 2024Modular Duality in Deep Learningcited by 77
  6. 2015Beyond Convexity: Stochastic Quasi-Convex Optimizationcited by 76
  7. 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
  8. 2020Practical Quasi-Newton Methods for Training Deep Neural Networkscited by 68
  9. 2025Benchmarking Optimizers for Large Language Model Pretrainingcited by 68
  10. 2025Structured Preconditioners in Adaptive Optimization: A Unified Analysiscited by 33
  11. 2024LoRA Done RITE: Robust Invariant Transformation Equilibration for LoRA Optimizationcited by 32
  12. 2021Adaptivity without Compromise: A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic Optimizationcited by 28
  13. 2025An Exploration of Non-Euclidean Gradient Descent: Muon and its Many Variantscited by 28
  14. 2025A Stable Whitening Optimizer for Efficient Neural Network Trainingcited by 18
  15. 2025Understanding and Improving Shampoo and SOAP via Kullback-Leibler Minimizationcited by 16
  16. 2025Unveiling the Basin-Like Loss Landscape in Large Language Modelscited by 13
  17. 2025On The Concurrence of Layer-wise Preconditioning Methods and Provable Feature Learningcited by 12
  18. 2024Gradient Methods with Online Scalingcited by 11
  19. 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGradcited by 8
  20. 2023SANIA: Polyak-type Optimization Framework Leads to Scale Invariant Stochastic Algorithmscited by 5
  21. 2025A Survey on Memory-Efficient Transformer-Based Model Training in AI for Sciencecited by 3
  22. 2024Quasi-Newton method of Optimization is proved to be a steepest descent method under the ellipsoid normcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 11.

11 papers.