通过多步梯度信息提升LoRA微调性能,效果超越现有方法。
LoRA-GA$^2$: Low Rank Adaptation with Multi-step Gradient Adaptive Alignment

- 利用预训练权重的多步梯度设计轻量探测器,不增加显存
- 基于梯度谱特性的动态秩分配与最优初始化,提升更新效率
- 在GLUE、GSM8K等任务上显著优于主流LoRA变体,适合高效微调场景
低秩适配(LoRA)是大型模型微调的重要方法,以较低内存开销实现良好性能。然而,其表现仍落后于全参数微调。现有方法通过单步梯度近似来对齐LoRA更新方向,但难以捕捉梯度的完整动态特性。本文提出LoRA-GA²,一种充分利用多步梯度信息的微调算法。我们设计了一个轻量级探测器,用于获取预训练权重的多步梯度,无额外显存开销且时间成本极低。进一步地,基于梯度谱特性实施重要性驱动的秩分配和最优初始化策略。大量实验表明,LoRA-GA²持续优于现有LoRA变体,同时保持原始LoRA的高效性:在GLUE基准平均领先0.66分,在GSM8K上领先1.03分,在HumanEval上领先0.87分。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) is a prominent fine-tuning method for large models, achieving competitive performance with reduced memory overhead. However, a persistent performance gap remains between LoRA and full fine-tuning. Recent studies have sought to narrow this gap by employing one-step gradient approximations of pretrained weights to align LoRA updates with the principal directions or intrinsic dimensionalities of full fine-tuning updates. Nevertheless, these approaches fail to capture the full dynamics of the gradients. In this paper, we propose LoRA-GA$^2$, an effective fine-tuning algorithm that fully leverages multi-step gradient information. Specifically, we introduce a lightweight probe for multi-step gradients of pretrained weights that incurs no additional GPU memory cost and only marginal time overhead. We further employ a spectrum-aware, importance-based rank allocation and optimal initialization derived from multi-step gradients. Extensive experimental results demonstrate that LoRA-GA$^2$ consistently outperforms existing LoRA variants while preserving the efficiency advantages of vanilla LoRA. For instance, LoRA-GA$^2$ surpasses the leading baseline by an average of 0.66 points on the GLUE benchmark, and outperforms the strongest baseline by 1.03 points on GSM8K and 0.87 points on HumanEval, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。