Papers
arxiv:2608.05136

The Loss Does Not See the Basis, but Adam Does

Published on Aug 5
· Submitted by
Devender Singh
on Aug 11
Authors:

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

Paper author Paper submitter
edited about 16 hours ago

loss-basis-adam-fig1-zoo

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

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

Sign up or log in to comment

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

Cite arxiv.org/abs/2608.05136 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2608.05136 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2608.05136 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.