mlsanityData to 14 Oct 2026

No Subspace to Track: Non-Identifiability and Optimizer State in Low-Rank Training

arXiv 2607.05872 ↗First posted 7 Jul 2026cs.LG math.OC stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Memory-efficient optimizers such as GaLore train large language models by projecting gradients onto a rank-r subspace recomputed every T steps, assuming this subspace is a slowly drifting object that can be tracked. We show that beyond a small reproducible core, there is no such object. Two estimates of the top-r subspace computed at the same step from disjoint minibatches disagree as much as estimates computed T steps apart (0.73 vs 0.74 of the maximal chordal distance sqrt(2r), at Pythia-160M with r=128): the apparent rotation at each refresh is dominated by estimator noise. This holds across four model families in three architecture classes from 70M to 6.9B parameters, strengthening with scale, and more weakly in a vision transformer. Only ~39 of 128 directions are reproducible across minibatches, and averaging cannot recover the rest: under N-fold averaging the gradient's spectral tail shrinks as N^(-1/4) rather than the N^(-1/2) of pure noise, so no averaging budget makes the subspace well defined. What helps instead follows from treating each refresh as a change of coordinates for Adam's state. Carrying the second moment blindly is provably about (r-k*)/2 worse than the best rotation-blind estimator, while the first moment transports exactly through the rotation, the optimal linear map under isotropic gradients and the rule LDAdam uses. At 1B over 40k steps (3 seeds), full LDAdam reaches 18.7 perplexity at beta2=0.999, beating untransported GaLore after its best beta2 fix (19.3); shortening the second-moment memory to beta2=0.99 helps the refreshing optimizers, though for canonical GaLore the effect is small and a full-rank control reverses it. One measurable fact, subspace non-identifiability, clarifies why GaLore works, which patches work, and what to check before trusting a low-rank assumption: the reproducible rank k*.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 32 references that are in the map. First posted on arXiv 7 Jul 2026.

Builds on

19 of 32 references.

References · 19

  1. 2016Pointer Sentinel Mixture Modelscited by 3,096
  2. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  3. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
  4. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 463
  5. 2018Gradient Descent Happens in a Tiny Subspacecited by 217
  6. 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 137
  7. 2024APOLLO: SGD-like Memory, AdamW-level Performancecited by 54
  8. 2024Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradientscited by 38
  9. 2024LDAdam: Adaptive Optimization from Low-Dimensional Gradient Statisticscited by 38
  10. 2024Subspace Optimization for Large Language Models with Convergence Guaranteescited by 34
  11. 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Trainingcited by 19
  12. 2025SubTrack++ : Gradient Subspace Tracking for Scalable LLM Trainingcited by 16
7 more
  1. 2025GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projectioncited by 14
  2. 2024FRUGAL: Memory-Efficient Optimization by Reducing State Overhead for Scalable Trainingcited by 10
  3. 2025Towards Efficient Optimizer Design for LLM via Structured Fisher Approximation with a Low-Rank Extensioncited by 9
  4. 2025Unbiased Gradient Low-Rank Projectioncited by 7
  5. 2026Spectral Lens: Activation and Gradient Spectra as Diagnostics of LLM Optimizationcited by 5
  6. 2026Spectra: Rethinking Optimizers for LLMs Under Spectral Anisotropycited by 4
  7. 2025Geometrically Principled Randomized Optimization for Efficient LLM Trainingcited by 1

Built on by

Not cited yet.