arXiv:2509.25414cs.LGcs.AI2025-09ACL被引 2

提出新型多LoRA架构,共享关键参数提升多任务与联邦微调效果。

Rethinking Parameter Sharing for LLM Fine-Tuning with Multiple LoRAs

  • 设计不对称多LoRA结构,多个A矩阵搭配共享的B矩阵。
  • 在多任务和联邦设置下实现更均衡的性能表现,平均准确率相当或更高。
  • 适合需要高效多任务微调及跨设备协作的场景,如联邦学习。

大语言模型常通过参数高效技术如低秩适应(LoRA)进行适配,形式为 $y = W_0x + BAx$,其中 $W_0$ 为预训练参数,$x$ 为输入。尽管多适配器方法常使用多个 LoRA,但先前研究认为内部 $A$ 矩阵在训练中高度相似,适合共享。我们重新审视该现象,发现其主要源于相同初始化而非知识共享,而 $B$ 矩阵在知识编码与传递中起更关键作用。基于此,我们提出 extbf{ALoRA}:一种多任务微调中的非对称多LoRA设计,包含多个 $A$ 矩阵和单一共享 $B$;以及 extbf{Fed-ALoRA},在同质与异质联邦微调设置下跨客户端共享 $B$,采用新型矩阵分解策略以支持不同客户端的异质秩。在常识推理、数学推理、多任务NLP数据集及联邦NLP数据集上的实验表明,我们的方法相较现有技术,在保持或提升平均准确率的同时,实现了各任务间更均衡的表现。代码已公开于 https://github.com/OptMN-Lab/ALoRA。

原文摘要 · Abstract (English)

Large language models are often adapted using parameter-efficient techniques such as Low-Rank Adaptation (LoRA), formulated as $y = W_0x + BAx$, where $W_0$ is the pre-trained parameters and $x$ is the input to the adapted layer. While multi-adapter extensions often employ multiple LoRAs, prior studies suggest that the inner $A$ matrices are highly similar during training and thus suitable for sharing. We revisit this phenomenon and find that this similarity is largely attributable to the identical initialization rather than shared knowledge, with $B$ playing a more critical role in knowledge encoding and transfer. Motivated by these insights, we propose \textbf{ALoRA}, an asymmetric multi-LoRA design with multiple $A$ matrices and a single shared $B$ in multi-task fine-tuning, and \textbf{Fed-ALoRA}, which shares $B$ across clients in federated fine-tuning under both homogeneous and heterogeneous settings, through a novel matrix decomposition strategy to accommodate heterogeneous ranks across clients. Experiments on commonsense reasoning, math reasoning, multi-task NLP dataset, and federated NLP dataset demonstrate that our methods achieve more balanced performance across tasks with comparable or superior average accuracy relative to existing multi-LoRA approaches. The code is available at https://github.com/OptMN-Lab/ALoRA.

LoRA多任务微调联邦学习参数共享

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。