通过分块对角矩阵提升LoRA表达能力,用少量参数实现更高秩。
BoRA: Towards More Expressive Low-Rank Adaptation with Block Diversity
- 将LoRA分解为分块矩阵乘法,引入每块独立的对角缩放。
- 在仅增加$ b^2r $参数下,使权重秩提升$ b $倍。
- 适合追求高表达力且参数受限的模型微调场景。
低秩适配(LoRA)是一种广泛用于大语言模型的参数高效微调方法,通过两个低秩矩阵 $BA$ 近似预训练权重矩阵 $W\in\mathbb{R}^{m\times n}$,其中 $A \in\mathbb{R}^{r\times n}$,$B\in\mathbb{R}^{m\times r}$($r\ll\min\{m,n\}$)。增大 $r$ 可提高权重秩并改善性能,但会显著增加可训练参数。本文提出块多样性低秩适配(BoRA),在极少新增参数下提升LoRA表达能力。BoRA将 $BA$ 视作分块矩阵乘法,将 $A$ 和 $B$ 分别按列和行划分为 $b$ 块(即 $A=[A_1,\dots,A_b]$,$B=[B_1,\dots,B_b]^\top$),使得 $BA$ 成为所有块乘积 $B_iA_j$ 的拼接。为增强不同块乘积的多样性,BoRA为每个 $B_iA_j$ 引入唯一对角矩阵 $Σ_{i,j} \in \mathbb{R}^{r\times r}$,形成 $B_i Σ_{i,j} A_j$。该设计使权重秩提升 $b$ 倍,仅需额外 $b^2r$ 参数。在多个数据集与模型上的实验验证了其优越性,消融研究进一步证明其可扩展性。
原文摘要 · Abstract (English)
Low-rank adaptation (LoRA) is a parameter-efficient fine-tuning (PEFT) method widely used in large language models (LLMs). It approximates the update of a pretrained weight matrix $W\in\mathbb{R}^{m\times n}$ by the product of two low-rank matrices, $BA$, where $A \in\mathbb{R}^{r\times n}$ and $B\in\mathbb{R}^{m\times r} (r\ll\min\{m,n\})$. Increasing the dimension $r$ can raise the rank of LoRA weights (i.e., $BA$), which typically improves fine-tuning performance but also significantly increases the number of trainable parameters. In this paper, we propose Block Diversified Low-Rank Adaptation (BoRA), which improves the rank of LoRA weights with a small number of additional parameters. Specifically, BoRA treats the product $BA$ as a block matrix multiplication, where $A$ and $B$ are partitioned into $b$ blocks along the columns and rows, respectively (i.e., $A=[A_1,\dots,A_b]$ and $B=[B_1,\dots,B_b]^\top$). Consequently, the product $BA$ becomes the concatenation of the block products $B_iA_j$ for $i,j\in[b]$. To enhance the diversity of different block products, BoRA introduces a unique diagonal matrix $Σ_{i,j} \in \mathbb{R}^{r\times r}$ for each block multiplication, resulting in $B_i Σ_{i,j} A_j$. By leveraging these block-wise diagonal matrices, BoRA increases the rank of LoRA weights by a factor of $b$ while only requiring $b^2r$ additional parameters. Extensive experiments across multiple datasets and models demonstrate the superiority of BoRA, and ablation studies further validate its scalability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。