AltLoRA改进低秩微调的梯度近似,提升性能且不增加内存开销。
AltLoRA: Towards Better Gradient Approximation in Low-Rank Adaptation with Alternating Projections
- 通过交替投影避免联合更新带来的梯度近似误差
- 在多个任务上逼近全量微调效果,内存效率仍优于传统方法
- 支持动量机制,适合需要高效微调的大模型场景
低秩微调(LoRA)能有效降低大语言模型微调的内存开销,但常因更新受限于低秩空间而导致性能不如全量微调。尽管如LoRA-Pro等变体试图通过调整低秩矩阵梯度来近似全梯度,但其解不唯一,且在引入动量或自适应优化时需计算等效梯度,导致内存开销接近全量微调。本文提出AltLoRA,一种基于交替投影的方法,在不增加内存复杂度的前提下,避免了联合更新带来的梯度近似问题,并自然融入动量机制。理论分析表明其具备收敛性,可实现稳定特征学习和对变换不变性的鲁棒性。在多个任务上的实验显示,AltLoRA优于LoRA及其变体,显著缩小与全量微调的差距,同时保持优异的内存效率。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) has emerged as an effective technique for reducing memory overhead in fine-tuning large language models. However, it often suffers from sub-optimal performance compared with full fine-tuning since the update is constrained in the low-rank space. Recent variants such as LoRA-Pro attempt to mitigate this by adjusting the gradients of the low-rank matrices to approximate the full gradient. However, LoRA-Pro's solution is not unique, and different solutions can lead to significantly varying performance in ablation studies. Besides, to incorporate momentum or adaptive optimization design, approaches like LoRA-Pro must first compute the equivalent gradient, causing a higher memory cost close to full fine-tuning. A key challenge remains in integrating momentum properly into the low-rank space with lower memory cost. In this work, we propose AltLoRA, an alternating projection method that avoids the difficulties in gradient approximation brought by the joint update design, meanwhile integrating momentum without higher memory complexity. Our theoretical analysis provides convergence guarantees and further shows that AltLoRA enables stable feature learning and robustness to transformation invariance. Extensive experiments across multiple tasks demonstrate that AltLoRA outperforms LoRA and its variants, narrowing the gap toward full fine-tuning while preserving superior memory efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。