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

大语言模型的联邦微调:框架比较与研究方向

Federated Fine-Tuning of LLMs: Framework Comparison and Research Directions

King’s College London(伦敦国王学院);Friedrich-Alexander-Universität Erlangen-Nürnberg(埃尔朗根-纽伦堡弗里德里希-亚历山大大学)

Na Yan, Yang Su, Yansha Deng, Robert Schober

AI总结:

本文比较三类结合参数上传、知识蒸馏与分割学习的 FedLLM 微调框架,从准确率、通信和客户端计算负载评估其优劣,并指出优化方向。

摘要:

联邦学习(FL)为使用分布式私有数据集微调预训练大语言模型(LLM)提供了一种隐私保护方案,能够在保持数据隐私的同时实现面向任务的适配。然而,由于巨大的通信与计算成本,在资源受限的联邦场景中微调 LLM 的大量参数尤其具有挑战性。为更深入地理解如何应对这些挑战,本文对三种先进的联邦 LLM(FedLLM)框架进行了比较分析,这些框架结合知识蒸馏(KD)和分割学习(SL)来缓解上述问题:1)FedLLMs,客户端上传模型参数或梯度,以实现直接而有效的微调;2)KD-FedLLMs,利用 KD 通过 logits 进行高效知识共享;3)Split-FedLLMs,将 LLM 拆分为两部分,一部分在客户端执行,另一部分在服务器执行,以平衡计算负载。每个框架都依据模型准确率、通信开销和客户端计算负载等关键性能指标进行评估,从而为其在不同联邦微调场景中的有效性提供洞见。通过该分析,我们识别了各框架特有的优化机会,以提升 FedLLM 的效率,并讨论更广泛的研究方向,强调使 FedLLM 更好地适配现实应用的开放机会。文中还给出了一个用例,以展示这三种框架在不同配置和设置下的性能比较。

英文摘要:

Federated learning (FL) provides a privacy-preserving solution for fine-tuning pre-trained large language models (LLMs) using distributed private datasets, enabling task-specific adaptation while preserving data privacy. However, fine-tuning the extensive parameters in LLMs is particularly challenging in resource-constrained federated scenarios due to the significant communication and computational costs. To gain a deeper understanding of how these challenges can be addressed, this article conducts a comparative analysis three advanced federated LLM (FedLLM) frameworks that integrate knowledge distillation (KD) and split learning (SL) to mitigate these issues: 1) FedLLMs, where clients upload model parameters or gradients to enable straightforward and effective fine-tuning; 2) KD-FedLLMs, which leverage KD for efficient knowledge sharing via logits; and 3) Split-FedLLMs, which split the LLMs into two parts, with one part executed on the client and the other one on the server, to balance the computational load. Each framework is evaluated based on key performance metrics, including model accuracy, communication overhead, and client-side computational load, offering insights into their effectiveness for various federated fine-tuning scenarios. Through this analysis, we identify framework-specific optimization opportunities to enhance the efficiency of FedLLMs and discuss broader research directions, highlighting open opportunities to better adapt FedLLMs for real-world applications. A use case is presented to demonstrate the performance comparison of these three frameworks under varying configurations and settings.

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

↑