提出SFed-LoRA,解决联邦学习中高秩适配器的梯度崩溃问题。
Stabilized Fine-Tuning with LoRA in Federated Learning: Mitigating the Side Effect of Client Size and Rank via the Scaling Factor
- 设计理论化缩放因子,缓解客户端聚合引发的统计方差
- 在多客户端场景下保持高秩适配器性能,收敛速度更快
- 无需修改模型结构,适合隐私敏感的分布式LLM微调
大语言模型在自然语言处理中至关重要。全量微调不切实际,促使参数高效微调(PEFT)方法如低秩适配(LoRA)的发展,其通过优化低秩矩阵A和B实现。但在需保护隐私的联邦学习(FL)场景中,LoRA集成常不稳定。我们发现,多客户端更新聚合引入的统计方差随客户端数量增加而增长,导致使用高秩适配器时出现梯度坍塌。现有缩放因子(如Rank-Stabilized LoRA所用)忽略聚合过程中的交互效应。为此,本文提出稳定联邦LoRA(SFed-LoRA),理论上刻画了适配器秩与联邦聚合之间的交互关系。推导出最优缩放因子,有效缓解跨N个客户端累积的聚合误差。通过纠正先前方法中的缩放不匹配,SFed-LoRA恢复高秩适配的有效性,且不改变原模型架构或增加推理延迟。在多种任务、模型架构及异构数据分布下进行大量实验,验证其效果:成功防止高秩坍塌,相比最先进基线显著提升稳定性与收敛速度。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are pivotal in natural language processing. The impracticality of full fine-tuning has prompted Parameter-Efficient Fine-Tuning (PEFT) methods like Low-Rank Adaptation (LoRA), optimizing low-rank matrices A and B. In distributed scenarios where privacy constraints necessitate Federated Learning (FL), however, the integration of LoRA is often unstable. Specifically, we identify that aggregating updates from multiple clients introduces statistical variance that scales with the client count, causing gradient collapse when using high-rank adapters. Existing scaling factor candidates, such as the one used by Rank-Stabilized LoRA, ignore the interaction caused by the aggregation process. To bridge this gap, this paper introduces Stabilized Federated LoRA (SFed-LoRA), a framework that theoretically characterizes the interaction between adapter rank and federated aggregation. We derive an optimal scaling factor designed to effectively mitigate the aggregation error accumulating across N clients. By correcting the scaling mismatch inherent in previous approaches, SFed-LoRA restores the efficacy of high-rank adaptation without altering the original model architecture or increasing inference latency. Extensive experiments in diverse tasks, model architectures, and heterogeneous data distributions are conducted to validate our results. We demonstrate that SFed-LoRA prevents high-rank collapse, and achieves significantly improved stability and faster convergence compared with state-of-the-art baselines for high-rank adaptation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。