mlsanityData to 7 Oct 2026

Beyond Perplexity: A Geometric and Spectral Study of Low-Rank Pre-Training

arXiv 2605.13652 ↗First posted 13 May 2026cs.LG cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Pre-training large language models is dominated by the memory cost of storing full-rank weights, gradients, and optimizer states. Low-rank pre-training has emerged to address this, and the space of methods has grown rapidly. A central question remains open: do low-rank methods produce models that generalize comparably to full-rank training, or does the rank constraint fundamentally alter the solutions reached? Existing comparisons rely almost entirely on validation perplexity from single-seed runs, often carried forward from prior literature. Yet perplexity is a poor proxy for solution quality; two methods can match on perplexity while converging to different loss landscape regions and internal representations. We close this gap by characterizing the solutions found by five low-rank pre-training methods, GaLore and Fira (memory-efficient optimizers), CoLA and SLTrain (architecture reparameterizations), and ReLoRA (adapter-style updates with periodic resets), against full-rank training at three model scales (60M, 130M, 350M). We evaluate each along 16 metrics across four dimensions: 1-D loss landscape along random/top-K PCA directions, 1-D interpolation between checkpoints, spectral structure of the weights and learned updates, and activation similarity to full-rank training. We show that low-rank methods are not equivalent to full-rank training, nor to one another, even when validation perplexity is close. Full-rank training settles into a sharper basin than low-rank methods along random directions, while the reverse holds for the top-1 PCA direction. Each method converges to a geometrically distinct basin. Low-rank activations diverge from full-rank in later layers as training progresses, with GaLore tracking full-rank most closely. Further, validation perplexity does not translate to downstream performance at every scale. Adding geometric and spectral metrics improves the prediction.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 24 of its 29 references that are in the map. First posted on arXiv 13 May 2026.

Builds on

24 of 29 references.

References · 24

  1. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  2. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  3. 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
  4. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
  5. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  6. 2023ReLoRA: High-Rank Training Through Low-Rank Updatescited by 200
  7. 2024Flora: Low-Rank Adapters Are Secretly Gradient Compressorscited by 103
  8. 2023A Modern Look at the Relationship between Sharpness and Generalizationcited by 91
  9. 2022Same Pre-training Loss, Better Downstream: Implicit Bias Matters for Language Modelscited by 84
  10. 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
  11. 2022On the Maximum Hessian Eigenvalue and Generalizationcited by 57
  12. 2024SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretrainingcited by 44
12 more
  1. 2024Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?cited by 39
  2. 2024Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradientscited by 36
  3. 2025CoLA: Compute-Efficient Pre-Training of LLMs via Low-Rank Activationcited by 28
  4. 2024Flat-LoRA: Low-Rank Adaptation over a Flat Loss Landscapecited by 19
  5. 2024VeLoRA: Memory Efficient Training using Rank-1 Sub-Token Projectionscited by 18
  6. 2024Grass: Compute Efficient Low-Memory LLM Training with Structured Sparse Gradientscited by 17
  7. 2025GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projectioncited by 13
  8. 2025Unveiling the Basin-Like Loss Landscape in Large Language Modelscited by 13
  9. 2025LOST: Low-rank and Sparse Pre-training for Large Language Modelscited by 10
  10. 2024COAP: Memory-Efficient Training with Correlation-Aware Gradient Projectioncited by 8
  11. 2026Demystifying Mergeability: Interpretable Properties to Predict Model Merging Successcited by 6
  12. 2024SwitchLoRA: Switched Low-Rank Adaptation Can Learn Full-Rank Informationcited by 2

Built on by

Not cited yet.