arXiv:2609.02734cs.LG2026-09

LoRA-TSD优化器让低秩微调更高效,提升模型性能并保证收敛性。

LoRA-TSD: Tangent-Space Spectral Descent for LoRA via Muon-Style Updates

  • 将低秩更新视为流形切向量,用谱范数梯度下降优化
  • 在多个大模型上优于现有方法,且对适配器秩不敏感
  • 首次给出LoRA-Pro和LoRA-TSD的全局收敛证明

低秩适配(LoRA)是微调大模型的标准方法,但当其两个因子独立训练时,更新忽略了其所诱导的低秩权重变化的几何结构。我们提出LoRA-TSD,一种将每一步LoRA更新视为固定秩矩阵流形切向量的优化器,并在该切空间内采用类似于Muon的谱范数最速下降步,通过原生的LoRA参数化收缩映射回因子。该方法避免了对完整权重矩阵的操作,其收缩计算比先前流形方法使用的截断SVD快达2.8倍。我们证明了该代理的Frobenius范数版本可恢复LoRA-Pro,并将切向投影梯度识别为自然的洛拉训练平稳性度量,仅需因子梯度即可计算。基于此度量,我们首次为LoRA-Pro和LoRA-TSD提供了全局收敛保证,收敛速率使因子梯度范数趋于零。在六项常识与自然语言推理基准测试中,使用Llama-3.2-1B、Llama-3.1-8B和Qwen3-32B模型,LoRA-TSD均超越所有对比优化器,且对适配器秩保持鲁棒性。代码已开源。

原文摘要 · Abstract (English)

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\times$ 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.

低秩微调优化器收敛分析流形学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。