基于渐近分析的LoRA初始化方法,提升微调效果与稳定性。
LoRA-DA: Data-Aware Initialization for Low-Rank Adaptation via Asymptotic Analysis
- 通过理论推导构建数据感知初始化框架,融合费雪信息与梯度特性。
- 在多个基准上显著提升最终精度,收敛更快更稳定。
- 适合追求高效微调性能的NLP研究者与工程师使用。
LoRA已成为主流的参数高效微调方法,其初始化策略日益受到关注。现有方法存在明显局限:多数未引入目标域数据,而基于梯度的方法仅浅层利用数据,依赖单步梯度分解。本文建立数据感知的LoRA初始化理论框架,从最小化微调后模型与目标模型间参数偏差的期望出发,推导出包含偏置项(用费雪-梯度形式近似以保持各向异性)和方差项(通过费雪信息捕捉采样随机性引入的不确定性)的优化问题。求解该问题得到最优初始化策略,并据此提出高效算法LoRA-DA。多基准实验表明,LoRA-DA在各项指标上均优于现有初始化方法。额外分析显示其收敛更快更稳定,对不同秩值鲁棒,且初始化开销极小。代码已公开于https://github.com/zqy0126/LoRA-DA。
原文摘要 · Abstract (English)
LoRA has become a widely adopted method for PEFT, and its initialization methods have attracted increasing attention. However, existing methods have notable limitations: many methods do not incorporate target-domain data, while gradient-based methods exploit data only at a shallow level by relying on one-step gradient decomposition. In this paper, we establish a theoretical framework for data-aware LoRA initialization. Starting from minimizing the expectation of the parameter discrepancy between the fine-tuned and target models, we derive an optimization problem with two components: a bias term, which is related to the parameter distance between the fine-tuned and target models, and is approximated using a Fisher-gradient formulation to preserve anisotropy; and a variance term, which accounts for the uncertainty introduced by sampling stochasticity through the Fisher information. Solving this problem yields an optimal initialization strategy for LoRA, based on which we develop an efficient algorithm, LoRA-DA. Empirical results across multiple benchmarks demonstrate that LoRA-DA consistently improves final accuracy over existing initialization methods. Additional studies show faster, more stable convergence, robustness across ranks, and only a small initialization overhead for LoRA-DA. The source code is available at https://github.com/zqy0126/LoRA-DA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。