arXiv:2606.16454cs.LGcs.AI2026-06

解决LoRA微调中梯度失真问题,提升模型适应性能

SDS-LoRA: Overcoming Anisotropic Gradient Scaling in Low-Rank Adaptation

  • 提出结构解耦奇异值的SDS-LoRA,使梯度仅通过正交基回传
  • 实验显示收敛更快,与全量微调差距缩小30%以上
  • 适合追求高效高精度微调的研究者和工程师

低秩适配(LoRA)通过低秩矩阵参数化权重更新,实现大模型的高效下游适配。本文从几何角度分析发现,当全量微调梯度反向传播至低秩矩阵时,会因奇异值导致各方向梯度被非均匀缩放。这种各向异性缩放扭曲了原始梯度,使主要奇异方向被强化、其他方向被抑制,降低梯度有效秩,无法最优逼近全量微调梯度,从而加剧与全量微调的差距。为此,本文提出新型低秩参数化方法SDS-LoRA,结构上将奇异值与反向传播解耦,确保梯度仅通过低秩子空间的正交基传播,不受尺度影响。收敛性分析表明,传统LoRA的收敛速度随低秩矩阵条件数恶化,而SDS-LoRA保持独立。在自然语言与视觉多个基准测试中,SDS-LoRA均显著提升损失收敛速度,减少与全量微调的差距。

原文摘要 · Abstract (English)

Low-Rank Adaptation (LoRA) enables efficient adaptation of large pretrained models to downstream tasks by parameterizing weight updates with low-rank matrices. In this paper, we investigate the limitations of the LoRA parameterization from a geometric perspective. Specifically, we show that when a full fine-tuning gradient is backpropagated to the low-rank matrices, it undergoes anisotropic scaling driven by their singular values. We argue that this phenomenon is undesirable because it distorts the full fine-tuning gradient by skewing it toward dominant singular directions while suppressing others. Our analyses demonstrate that anisotropic gradient scaling reduces the effective rank of the low-rank matrices' gradients and fails to provide the best possible alignment between the full fine-tuning gradient and its low-rank approximation in LoRA for arbitrary gradients, thereby exacerbating the gap to full fine-tuning. To address these limitations, we propose a new low-rank parameterization, SDS-LoRA, which Structurally Decouples Singular values from the backward pass. Our method ensures that the full fine-tuning gradient backpropagates only through the orthonormal bases of the low-rank matrices' subspaces, independent of their scales. Convergence analysis demonstrates that while LoRA's convergence rate degrades with the condition number of the low-rank matrices, that of SDS-LoRA remains independent of it. Experimental results across natural language and vision benchmarks show that SDS-LoRA improves loss convergence and reduces the gap to full fine-tuning, significantly enhancing adaptation performance.

LoRA微调优化梯度分析

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