mlsanityData to 14 Oct 2026

The Loss Does Not See the Basis, but Adam Does

arXiv 2608.05136 ↗First posted 5 Aug 2026cs.LG math.OC stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Gradient descent on a factored model W = UV^top is implicitly biased toward low-rank solutions, while Adam, starting from the same small initialization, is not. We trace the difference to the gauge symmetry of the loss, its invariance under (U, V) mapsto (UQ, VQ). Gradient flow's low-rank mechanism is available to an optimizer only if that optimizer is gauge-equivariant, a condition necessary for the transfer but not sufficient for low-rank recovery. Gradient descent, momentum, "shared-scalar" Adam, Muon, and Shampoo satisfy it. Adam, RMSProp, and the other coordinate-wise methods do not. A structure theorem characterizes the memoryless equivariant rules as exactly the Gram-determined left preconditioners, and a transfer theorem carries gradient flow's pathwise properties to common-scalar flows. We then sort nine update rules on underdetermined matrix sensing by recovery error against the planted ground truth. A one-parameter family from coordinate-wise to shared-scalar preconditioning restores the bias monotonically, isolating anisotropy as the cause. A "spectral schedule" reconciles two opposing reports about Muon: equal-rate updates recover exactly low-rank targets but lose their edge as the spectral tail grows. In transformers, Adam separates two gauge-equivalent initializations at the first step, where the equivariant optimizers stay at float precision, and ends with the per-head invariants WQ^top WK 56% apart in relative Frobenius distance, a gap no per-head rotation can close. On two hyperspectral datasets at matched training loss, gradient descent cuts held-out error by 43-44% at the lowest sampling density, and at lower effective rank. Basis choice is therefore not a tuning detail but a decision about which interpolant the optimizer selects.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 48 of its 61 references that are in the map. First posted on arXiv 5 Aug 2026.

Builds on

48 of 61 references.

References · 48

  1. 2017Attention Is All You Needcited by 60,361
  2. 2019Root Mean Square Layer Normalizationcited by 1,480
  3. 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 950
  4. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
  5. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
  6. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  7. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
  8. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
  9. 2023Symbolic Discovery of Optimization Algorithmscited by 434
  10. 2025Muon is Scalable for LLM Trainingcited by 428
  11. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
  12. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
36 more
  1. 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
  2. 2017Implicit Regularization in Matrix Factorizationcited by 322
  3. 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 284
  4. 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
  5. 2024Old Optimizer, New Norm: An Anthologycited by 193
  6. 2018Algorithmic Regularization in Learning Deep Homogeneous Models: Layers are Automatically Balancedcited by 177
  7. 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 150
  8. 2017Improving Generalization Performance by Switching from Adam to SGDcited by 125
  9. 2020Implicit Regularization in Deep Learning May Not Be Explainable by Normscited by 119
  10. 2020Descending through a Crowded Valley - Benchmarking Deep Learning Optimizerscited by 119
  11. 2020Neural Mechanics: Symmetry and Broken Conservation Laws in Deep Learning Dynamicscited by 96
  12. 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 80
  13. 2020Accelerating Ill-Conditioned Low-Rank Matrix Estimation via Scaled Gradient Descentcited by 71
  14. 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
  15. 2023Benchmarking Neural Network Training Algorithmscited by 52
  16. 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 41
  17. 2024The Implicit Bias of Adam on Separable Datacited by 36
  18. 2024LoRA Done RITE: Robust Invariant Transformation Equilibration for LoRA Optimizationcited by 32
  19. 2023On the Implicit Bias of Adamcited by 27
  20. 2024Adam Exploits ℓ_∞-geometry of Loss Landscape via Coordinate-wise Adaptivitycited by 23
  21. 2022VectorAdam for Rotation Equivariant Geometry Optimizationcited by 18
  22. 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networkscited by 11
  23. 2026Symmetry-Compatible Principle for Optimizer Design: Embeddings, LM Heads, SwiGLU MLPs, and MoE Routerscited by 11
  24. 2026To Use or not to Use Muon: How Simplicity Bias in Optimizers Matterscited by 10
  25. 2026Uniform Spectral Growth and Convergence of Muon in LoRA-Style Matrix Factorizationcited by 9
  26. 2025Adam Reduces a Unique Form of Sharpness: Theoretical Insights Near the Minimizer Manifoldcited by 5
  27. 2026Symmetry Breaking in Transformers for Efficient and Interpretable Trainingcited by 5
  28. 2025How do simple rotations affect the implicit bias of Adam?cited by 3
  29. 2026Understanding and inverse design of implicit bias in stochastic learning: a geometric perspectivecited by 3
  30. 2026The Spectral Dynamics and Noise Geometry of Muoncited by 3
  31. 2025Implicit Bias of Per-sample Adam on Separable Data: Departure from the Full-batch Regimecited by 2
  32. 2026PolarAdamW: Disentangling Spectral Control and Schur Gauge-Equivariance in Matrix Optimisationcited by 2
  33. 2026How the Optimizer Shapes Learned Solutions in Equivariant Neural Networkscited by 2
  34. 2026Towards Understanding the Power and Limits of the Muon Optimizer: A River-Valley Perspectivecited by 2
  35. 2026Dead-Direction Conditioners: Gauge-Equivariant Preconditioning for Deep Networkscited by 2
  36. 2026FlowAdam: Implicit Regularization via Geometry-Aware Soft Momentum Injectioncited by 1

Built on by

Not cited yet.