arXiv:2602.03019cs.LGcs.AI2026-02中稿 · INFOCOM 2026被引 2

提出高效联邦微调方法,降低大模型训练的通信与内存开销。

FedKRSO: Communication and Memory Efficient Federated Fine-Tuning of Large Language Models

  • 客户端在服务器生成的低维随机子空间内更新模型,节省内存。
  • 仅传输子空间上的模型更新累加器,减少通信量。
  • 性能接近全参数微调,适合边缘设备部署。

微调对将通用大语言模型(LLM)适配到特定任务至关重要。作为保护隐私、利用分布式数据进行协同训练的框架,联邦学习(FL)在LLM微调中日益流行,但受限于高通信成本和客户端资源有限。尽管参数高效微调(PEFT)可降低通信与内存开销,常导致性能下降。本文提出FedKRSO(联邦K-种子随机子空间优化),一种新型联邦微调方法,可在资源受限环境下高效实现大模型微调。客户端在服务器生成的共享随机低维子空间内更新模型,降低内存使用;不传输完整模型参数,仅发送子空间上的模型更新累加器,实现高效的全局聚合与分发。该策略显著降低通信与内存开销,同时克服了传统PEFT的性能瓶颈,逼近联邦全参数微调性能。在多样化联邦场景下的GLUE基准测试中,FedKRSO展现出优异性能与极低开销,为边缘端联邦大模型微调提供了可行路径。

原文摘要 · Abstract (English)

Fine-tuning is essential to adapt general-purpose large language models (LLMs) to domain-specific tasks. As a privacy-preserving framework to leverage decentralized data for collaborative model training, Federated Learning (FL) is gaining popularity in LLM fine-tuning, but remains challenging due to the high cost of transmitting full model parameters and computing full gradients on resource-constrained clients. While Parameter-Efficient Fine-Tuning (PEFT) methods are widely used in FL to reduce communication and memory costs, they often sacrifice model performance compared to FFT. This paper proposes FedKRSO (Federated $K$-Seed Random Subspace Optimization), a novel method that enables communication and memory efficient FFT of LLMs in federated settings. In FedKRSO, clients update the model within a shared set of random low-dimension subspaces generated by the server to save memory usage. Furthermore, instead of transmitting full model parameters in each FL round, clients send only the model update accumulators along the subspaces to the server, enabling efficient global model aggregation and dissemination. By using these strategies, FedKRSO can substantially reduce communication and memory overhead while overcoming the performance limitations of PEFT, closely approximating the performance of federated FFT. The convergence properties of FedKRSO are analyzed rigorously under general FL settings. Extensive experiments on the GLUE benchmark across diverse FL scenarios demonstrate that FedKRSO achieves both superior performance and low communication and memory overhead, paving the way towards on federated LLM fine-tuning at the resource-constrained edge.

联邦学习大模型微调内存效率通信优化

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