arXiv:2501.04436cs.LGcs.AI2025-01被引 15

比较三种联邦微调大模型框架,找隐私保护下的高效方案

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

  • 用知识蒸馏和分割学习降低通信与算力开销
  • 三框架在准确率、通信量、客户端负载上各有优劣
  • 适合关注隐私计算与大模型落地的研究者参考

联邦学习(FL)为使用分布式私有数据集微调预训练大语言模型(LLMs)提供隐私保护方案,可在不泄露数据的前提下实现任务适配。然而,在资源受限的联邦场景中,微调大规模参数的LLM面临显著的通信和计算成本挑战。本文对比分析三种融合知识蒸馏(KD)与分割学习(SL)的先进联邦大模型(FedLLM)框架:1)FedLLMs,客户端上传参数或梯度,实现简单有效的微调;2)KD-FedLLMs,通过输出逻辑值(logits)实现高效知识共享;3)Split-FedLLMs,将模型分两部分,客户端运行一部分,服务端运行另一部分,平衡计算负荷。各框架基于模型准确率、通信开销、客户端计算负载等关键指标评估,揭示其在不同场景下的适用性。研究识别出框架优化机会,并探讨更广泛的科研方向,提出真实应用中的开放问题。通过用例展示三种框架在不同配置下的性能表现。

原文摘要 · Abstract (English)

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.

联邦学习大模型微调隐私计算知识蒸馏

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