arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.03019cs.LGcs.AI
收藏

FedKRSO: 在联邦学习中实现高效通信和内存的大型语言模型微调

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

University of Texas at San Antonio(德克萨斯大学圣安东尼奥分校);Penn State University(宾夕法尼亚州立大学)

Guohao Yang, Tongle Wu, Yuanxiong Guo, Ying Sun, Yanmin Gong

AI总结:

FedKRSO通过在联邦学习中利用随机子空间优化,实现了高效通信和内存的LLM微调,兼顾性能与资源节约。

摘要:

微调对于将通用大型语言模型(LLMs)适应到特定领域任务至关重要。作为一种保护隐私的框架,利用去中心化数据进行协作模型训练的联邦学习(FL)在LLM微调中变得流行,但由于在资源受限的客户端上计算完整梯度的成本高,仍然面临挑战。尽管参数高效微调(PEFT)方法在FL中被广泛使用,以减少通信和内存成本,但它们通常在性能上牺牲了与FFT相比的性能。本文提出了FedKRSO(联邦K-种子随机子空间优化),一种新的方法,能够在联邦设置中实现高效通信和内存的LLM微调。在FedKRSO中,客户端在服务器生成的共享随机低维子空间内更新模型以节省内存使用。此外,而不是在每次FL轮次中传输完整的模型参数,客户端只将沿子空间的模型更新累积器发送到服务器,从而实现高效的全局模型聚合和传播。通过使用这些策略,FedKRSO可以显著减少通信和内存开销,同时克服PEFT的性能限制,接近联邦FFT的性能。FedKRSO的收敛性质在一般FL设置下进行了严格分析。在GLUE基准测试上广泛实验,跨多样化的FL场景证明,FedKRSO在性能和通信及内存开销方面都表现出色,为在资源受限边缘实现联邦LLM微调铺平了道路。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑