用一个向量就能实现全模型高效微调,性能超越现有方法。
Uni-LoRA: One Vector is All You Need
- 将多种LoRA方法统一为投影框架,仅需一个可训练向量重建参数
- 在多个基准上达到顶尖参数效率,微调开销降低近一半
- 适合需要极简参数更新的场景,如资源受限部署
低秩适配(LoRA)已成为大语言模型参数高效微调的主流方法,通过将权重更新限制在低秩矩阵中实现。近期工作如Tied-LoRA、VeRA和VB-LoRA通过引入额外约束进一步压缩可训练参数空间。本文提出统一框架Uni-LoRA,将LoRA参数空间视为高维向量空间$R^D$,可通过低维子空间$R^d$($d \ll D$)的投影重建。我们发现不同LoRA方法的核心差异在于投影矩阵$P \in R^{D \times d}$的选择。多数现有方法依赖层内或结构特定的投影,限制跨层参数共享,影响效率。为此,我们设计了一种保距的高效投影矩阵,支持全局参数共享并降低计算开销。在统一视角下,该设计仅需一个可训练向量即可重构整个LLM的LoRA参数,实现‘一向量即足’。在GLUE、数学推理和指令微调等多个基准上的实验证明,Uni-LoRA在保持优异预测性能的同时,达到最先进的参数效率。代码已开源。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) has become the de facto parameter-efficient fine-tuning (PEFT) method for large language models (LLMs) by constraining weight updates to low-rank matrices. Recent works such as Tied-LoRA, VeRA, and VB-LoRA push efficiency further by introducing additional constraints to reduce the trainable parameter space. In this paper, we show that the parameter space reduction strategies employed by these LoRA variants can be formulated within a unified framework, Uni-LoRA, where the LoRA parameter space, flattened as a high-dimensional vector space $R^D$, can be reconstructed through a projection from a subspace R^d, with $d \ll D$. We demonstrate that the fundamental difference among various LoRA methods lies in the choice of the projection matrix, $P \in R^{D \times d}$.Most existing LoRA variants rely on layer-wise or structure-specific projections that limit cross-layer parameter sharing, thereby compromising parameter efficiency. In light of this, we introduce an efficient and theoretically grounded projection matrix that is isometric, enabling global parameter sharing and reducing computation overhead. Furthermore, under the unified view of Uni-LoRA, this design requires only a single trainable vector to reconstruct LoRA parameters for the entire LLM - making Uni-LoRA both a unified framework and a "one-vector-only" solution. Extensive experiments on GLUE, mathematical reasoning, and instruction tuning benchmarks demonstrate that Uni-LoRA achieves state-of-the-art parameter efficiency while outperforming or matching prior approaches in predictive performance. Our code is available at https://github.com/KaiyangLi1992/Uni-LoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。