Abstract
Optimizer behavior in factored matrix models depends on gauge equivariance, with coordinate-wise methods breaking low-rank bias and causing divergent solutions in transformers and sensing tasks.
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 W_Q^top W_K 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.
Community
Gradient descent on a factored model W = UV^T is implicitly biased
toward low rank; Adam, from the same small initialization, is not. The cause is a
symmetry: the loss is invariant under (U,V) -> (UQ, VQ) for orthogonal Q, and
gradient flow's low-rank mechanism transfers to an optimizer only if that optimizer
respects the same symmetry. GD, momentum, "shared-scalar" Adam, Muon and Shampoo do.
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 one-parameter dial from coordinate-wise
to shared-scalar preconditioning restores the bias monotonically - so anisotropy,
not adaptivity, is what breaks it.
In transformers Adam separates two gauge-equivalent
initializations at the very first step and ends with the per-head invariants
W_Q^T W_K 56% apart in relative Frobenius distance, a gap no per-head rotation can
close; on two hyperspectral datasets at matched training loss, GD cuts held-out
error by 43-44% at the lowest sampling density, and at lower effective rank. A
"spectral schedule" also reconciles the two opposing reports about Muon.
Happy to discuss -- especially the equivariance-vs-recovery gap (necessary, not
sufficient) and where the dial sits for real architectures.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Reassessing Muon for Matrix Factorization (2026)
- No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Training (2026)
- PoLoRA: A Preconditioned Orthogonalized LoRA Optimizer (2026)
- Dead-Direction Signatures: A Cheap Spectral Reading of Singular Complexity (2026)
- Fantastic Pretraining Optimizers and Where to Find Them II: Hyperball Optimization (2026)
- ManifoldFlow: SPD-Relaxed Stiefel Layers with Learnable Singular Spectrum (2026)
- Improving Neural Network Training by Decoupling the Magnitude and Direction of Weight Vectors (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2608.05136 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper
