mlsanityData to 7 Oct 2026

Stabilizing Native Low-Rank LLM Pretraining

arXiv 2602.12429 ↗First posted 12 Feb 2026cs.LGcited by 6

Not on the map. Why?

Abstract (arXiv)

Foundation models have achieved remarkable success, yet their growing parameter counts pose significant computational and memory challenges. Low-rank factorization offers a promising route to reduce training and inference costs, but the community lacks a stable recipe for training models from scratch using exclusively low-rank weights while matching the performance of the dense model. We demonstrate that Large Language Models (LLMs) can be trained from scratch using exclusively low-rank factorized weights for all non-embedding matrices without auxiliary "full-rank" guidance required by prior methods. While native low-rank training often suffers from instability and loss spikes, we identify uncontrolled growth in the spectral norm (largest singular value) of the weight matrix update as the dominant factor. To address this, we introduce Spectron: Spectral renormalization with orthogonalization, which dynamically bounds the resultant weight updates based on the current spectral norms of the factors. Our method enables stable, end-to-end factorized training with negligible overhead. Finally, we establish compute-optimal scaling laws for natively low-rank transformers, demonstrating predictable power-law behavior and improved inference efficiency relative to dense models.

Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 48 of its 63 references that are in the map. First posted on arXiv 12 Feb 2026.

Builds on

48 of 63 references.

References · 48

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  4. 2020Scaling Laws for Neural Language Modelscited by 6,411
  5. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
  6. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  7. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  8. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  9. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  10. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
  11. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  12. 2022Emergent Abilities of Large Language Modelscited by 2,483
36 more
  1. 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,320
  2. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  3. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
  4. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
  5. 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
  6. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  7. 2025Muon is Scalable for LLM Trainingcited by 381
  8. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 272
  9. 2019PowerSGD: Practical Low-Rank Gradient Compression for Distributed Optimizationcited by 234
  10. 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 211
  11. 2023ReLoRA: High-Rank Training Through Low-Rank Updatescited by 200
  12. 2025Training Deep Learning Models with Norm-Constrained LMOscited by 177
  13. 2024Old Optimizer, New Norm: An Anthologycited by 173
  14. 2023Beyond Chinchilla-Optimal: Accounting for Inference in Language Model Scaling Lawscited by 155
  15. 2023The Truth is in There: Improving Reasoning in Language Models with Layer-Selective Rank Reductioncited by 120
  16. 2023A Spectral Condition for Feature Learningcited by 101
  17. 2025Understanding Gradient Orthogonalization for Deep Learning via Non-Euclidean Trust-Region Optimizationcited by 81
  18. 2025A Note on the Convergence of Muoncited by 78
  19. 2025Muon Optimizes Under Spectral Norm Constraintscited by 77
  20. 2021Pufferfish: Communication-efficient Models At No Extra Costcited by 61
  21. 2025AdaMuon: Adaptive Muon Optimizercited by 59
  22. 2021Initialization and Regularization of Factorized Neural Layerscited by 58
  23. 2020On the distance between two neural networks and the stability of learningcited by 46
  24. 2024Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?cited by 39
  25. 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 33
  26. 2025CoLA: Compute-Efficient Pre-Training of LLMs via Low-Rank Activationcited by 28
  27. 2024Compute Better Spent: Replacing Dense Layers with Structured Matricescited by 24
  28. 2024SWAN: SGD with Normalization and Whitening Enables Stateless LLM Trainingcited by 20
  29. 2024Training Neural Networks from Scratch with Parallel Low-Rank Adapterscited by 17
  30. 2025How Muon's Spectral Design Benefits Generalization: A Study on Imbalanced Datacited by 16
  31. 2025Subspace Networks: Scaling Decentralized Training with Communication-Efficient Model Parallelismcited by 10
  32. 2024PETRA: Parallel End-to-end Training with Reversible Architecturescited by 6
  33. 2024ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Trainingcited by 6
  34. 2024Investigating Low-Rank Training in Transformer Language Models: Efficiency and Scaling Analysiscited by 6
  35. 2024Sparse Spectral Training and Inference on Euclidean and Hyperbolic Neural Networkscited by 4
  36. 2026Singular Bayesian Neural Networkscited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 6.

6 papers.