LoRA-TSD: Tangent-Space Spectral Descent for LoRA via Muon-Style Updates
Not on the map. Why?
Abstract (arXiv)
Low-rank adaptation (LoRA) is the standard way to fine-tune large models, yet when its two factors are trained independently, the update ignores the geometry of the low-rank weight change it induces. We introduce LoRA-TSD, an optimizer that treats every LoRA step as a tangent vector of the fixed-rank matrix manifold and takes the spectral-norm steepest-descent step of Muon inside that tangent space, mapping the result back to the factors through a retraction native to the LoRA parametrization. The step avoids expensive operations on full weight matrices, and its retraction is up to 2.8× cheaper than the truncated-SVD retraction used by prior manifold methods. We prove that the Frobenius-norm version of our surrogate recovers LoRA-Pro, and we identify the tangent-projected gradient, the Riemannian gradient of the manifold, as the stationarity measure natural to LoRA training and computable from the factor gradients alone. Under this measure we give the first global convergence guarantees for both LoRA-Pro and LoRA-TSD, with rates that drive the factor-gradient norms to zero. Across six commonsense and natural-language-inference benchmarks with Llama-3.2-1B, Llama-3.1-8B and Qwen3-32B, LoRA-TSD outperforms every competing LoRA optimizer and stays robust to the adapter rank. Code is available at https://github.com/brain-lab-research/LoRA-TSD.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 37 references that are in the map. First posted on arXiv 2 Sep 2026.
Builds on
23 of 37 references.
References · 23
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 758
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
- 2024LoRA+: Efficient Low Rank Adaptation of Large Modelscited by 377
- 2023AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuningcited by 357
11 more
- 2024PiSSA: Principal Singular Values and Singular Vectors Adaptation of Large Language Modelscited by 269
- 2024LoRA-GA: Low-Rank Adaptation with Gradient Approximationcited by 141
- 2024LoRA-Pro: Are Low-Rank Adapters Properly Optimized?cited by 77
- 2024Riemannian Preconditioned LoRA for Fine-Tuning Foundation Modelscited by 52
- 2024LoRA Done RITE: Robust Invariant Transformation Equilibration for LoRA Optimizationcited by 32
- 2025LoFT: Low-Rank Adaptation That Behaves Like Full Fine-Tuningcited by 14
- 2025LoRA meets Riemannion: Muon Optimizer for Parametrization-independent Low-Rank Adapterscited by 13
- 2025On the Convergence Rate of LoRA Gradient Descentcited by 10
- 2025WeightLoRA: Keep Only Necessary Adapterscited by 4
- 2026LoRA-Muon: Spectral Steepest Descent on the Low-Rank Manifoldcited by 4
- 2026On the Convergence of Stochastic Low-Rank Adaptationcited by 1
Built on by
Not cited yet.