提出FedSRD框架,大幅降低联邦微调大模型的通信开销。
FedSRD: Sparsify-Reconstruct-Decompose for Communication-Efficient Federated Large Language Models Fine-Tuning
- 通过重要性感知稀疏化减少上传参数量,保留LoRA更新结构完整性。
- 在异构客户端数据上实验,通信成本降低最高达90%,性能更优。
- 适合资源受限环境下高效训练隐私敏感的大模型,如医疗、金融场景。
当前在公开网络数据上训练大语言模型的范式正变得不可持续,因高质量专业领域数据日益枯竭。联邦学习(FL)成为下一代去中心化Web上隐私保护协作微调的可行方案,可在分散的私有数据上进行联合训练。尽管低秩适配(LoRA)是高效微调的标准方法,但其在联邦环境下的应用面临关键瓶颈:在异构网络条件下通信开销过高。LoRA参数中的结构冗余导致通信成本上升并引发聚合冲突。为此,我们提出FedSRD——一种用于高效通信的联邦大模型微调框架,包含稀疏化-重构-分解三阶段。引入重要性感知稀疏化以减少上传参数量,同时保持LoRA更新的结构完整性;服务器在全秩空间聚合更新,缓解冲突后,将全局更新分解为稀疏低秩格式广播,实现对称高效循环。此外,我们还提出了高效变体FedSRD-e以降低计算开销。在10个基准测试上的实验表明,该框架可将通信成本最高降低90%,并在异构客户端数据上提升性能。
原文摘要 · Abstract (English)
The current paradigm of training large language models (LLMs) on public available Web data is becoming unsustainable as high-quality data sources in specialized domains near exhaustion. Federated Learning (FL) emerges as a practical solution for the next generation of AI on a decentralized Web, enabling privacy-preserving collaborative fine-tuning on decentralized private data. While Low-Rank Adaptation (LoRA) is standard for efficient fine-tuning, its federated application faces a critical bottleneck: communication overhead under heterogeneous network conditions. Structural redundancy in LoRA parameters increases communication costs and causes aggregation conflicts. To address this, we propose FedSRD, a Sparsify-Reconstruct-Decompose framework for communication-efficient federated LLM fine-tuning. We introduce importance-aware sparsification to reduce the upload parameter count while preserving the structural integrity of LoRA updates. The server aggregates updates in full-rank space to mitigate conflicts, then decomposes the global update into a sparse low-rank format for broadcast, ensuring a symmetrically efficient cycle. We also propose an efficient variant, FedSRD-e, to reduce computational overhead. Experiments on 10 benchmarks show our framework significantly reduces communication costs by up to 90\% while improving performance on heterogeneous client data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。