用精心设计的初始化让低秩微调媲美全量微调,参数少27到90倍。
Initialization using Update Approximation is a Silver Bullet for Extremely Efficient Low-Rank Fine-Tuning
- 通过改进LoRA-XS结构实现梯度最优近似初始化
- 在数学推理等任务上超越基线,参数量仅1/90
- 无需调节缩放因子,适合资源有限的高效微调场景
低秩适配器已成为高效微调大语言模型的标准方法,但性能常不及全量微调。本文提出一种名为LoRA Silver Bullet(LoRA-SB)的方法,通过精心设计的初始化策略,在低秩子空间内逼近全量微调。理论上证明,将可学习的 r×r 矩阵插入到 LoRA-XS 的 A 与 B 之间,能提供精确的近似条件。利用其受限的更新空间,实现高秩梯度更新的最优缩放,同时无需缩放因子调参。我们证明该初始化能提供初始梯度的最优低秩近似,并保持训练中更新方向一致。在数学推理、常识推理和语言理解任务上的大量实验表明,该方法性能优于LoRA及基线,参数量仅为后者的27至90倍,全面超越LoRA-XS。结果表明,可在低秩子空间中模拟全量微调,实现显著参数效率提升且不牺牲性能。代码已公开:https://github.com/CERT-Lab/lora-sb。
原文摘要 · Abstract (English)
Low-rank adapters have become standard for efficiently fine-tuning large language models, but they often fall short of achieving the performance of full fine-tuning. We propose a method, LoRA Silver Bullet or LoRA-SB, that approximates full fine-tuning within low-rank subspaces using a carefully designed initialization strategy. We theoretically demonstrate that the architecture of LoRA-XS, which inserts a learnable r x r matrix between B and A while keeping other matrices fixed, provides the precise conditions needed for this approximation. We leverage its constrained update space to achieve optimal scaling for high-rank gradient updates while removing the need for scaling factor tuning. We prove that our initialization offers an optimal low-rank approximation of the initial gradient and preserves update directions throughout training. Extensive experiments across mathematical reasoning, commonsense reasoning, and language understanding tasks demonstrate that our approach exceeds the performance of LoRA (and baselines) while using 27-90 times fewer learnable parameters, and comprehensively outperforms LoRA-XS. Our findings establish that it is possible to simulate full fine-tuning in low-rank subspaces, and achieve significant parameter efficiency gains without sacrificing performance. Our code is publicly available at: https://github.com/CERT-Lab/lora-sb.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。