arXiv:2502.15436cs.LGcs.AI2025-02被引 13

Fed-SB让大模型联邦微调通信量降低230倍,同时保持顶尖性能。

Fed-SB: A Silver Bullet for Extreme Communication Efficiency and Performance in (Private) Federated LoRA Fine-Tuning

  • 通过学习适配器间的小型矩阵R,实现精确聚合,无需高通信开销。
  • 在多个任务上达到当前最优性能,通信成本仅与客户端数无关。
  • 适合注重隐私和效率的分布式大模型微调场景。

低秩适配(LoRA)已成为高效微调基础模型的通用方法。然而,使用LoRA进行联邦微调面临挑战:传统联邦平均法导致适配器更新次优。现有方案要么通信成本随客户端数量线性增长,难以承受;要么因表达能力受限而性能下降。我们提出联邦银弹(Fed-SB),一种基于新近提出的低秩适配方法LoRA-SB的联邦微调框架。LoRA-SB通过在适配器B与A之间学习一个小型方阵R,使优化轨迹最优逼近全参数微调的投影方向,其余组件保持固定。直接对R进行平均可保证精确更新,显著降低通信成本,且其开销与客户端数量无关,具备良好可扩展性。Fed-SB在常识推理、算术推理和语言推断任务中均达到当前最优性能,通信成本最高降低230倍。在私有设置下,Fed-SB进一步提升表现:(1) 减少可训练参数,降低差分隐私所需噪声;(2) 避免其他方法引入的噪声放大。总体而言,Fed-SB为私有与非私有联邦微调提供了一种兼具先进性、高效性和可扩展性的解决方案。代码已公开于:https://github.com/CERT-Lab/fed-sb。

原文摘要 · Abstract (English)

Low-Rank Adaptation (LoRA) has become ubiquitous for efficiently fine-tuning foundation models. However, federated fine-tuning using LoRA is challenging due to suboptimal updates arising from traditional federated averaging of individual adapters. Existing solutions either incur prohibitively high communication cost that scales linearly with the number of clients or suffer from performance degradation due to limited expressivity. We introduce Federated Silver Bullet (Fed-SB), a novel approach for federated fine-tuning of LLMs using LoRA-SB, a recently proposed low-rank adaptation method. LoRA-SB optimally aligns the optimization trajectory with the ideal low-rank full fine-tuning projection by learning a small square matrix (R) between adapters B and A, keeping other components fixed. Direct averaging of R guarantees exact updates, substantially reducing communication cost, which remains independent of the number of clients, and enables scalability. Fed-SB achieves state-of-the-art performance across commonsense reasoning, arithmetic reasoning, and language inference tasks while reducing communication costs by up to 230x. In private settings, Fed-SB further improves performance by (1) reducing trainable parameters, thereby lowering the noise required for differential privacy and (2) avoiding noise amplification introduced by other methods. Overall, Fed-SB offers a state-of-the-art, efficient, and scalable solution for both private and non-private federated fine-tuning. Our code is publicly available at: https://github.com/CERT-Lab/fed-sb.

联邦学习低秩适配通信效率隐私保护

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