arXiv:2604.26388cs.DCcs.LG2026-04被引 1

针对大模型微调的联邦分割学习,自适应调整切分层并降低通信开销。

SplitFT: An Adaptive Federated Split Learning System For LLMs Fine-Tuning

论文配图:SplitFT: An Adaptive Federated Split Learning System For LLMs Fine-Tuning
图 1 · 摘自论文原文
  • 根据设备能力与数据差异动态设置不同客户端的模型切分层。
  • 通过降低切分层的LoRA秩,减少通信量,提升训练效率。
  • 支持真实场景下的异构数据分布,适合资源受限的边缘设备使用。

联邦分割学习被证明是缓解经典联邦学习中客户端计算资源不足问题的有效方法,同时保障数据所有者之间的数据隐私。然而,在对大型语言模型(LLMs)进行微调时,该策略面临严峻挑战:如何在不同客户端间自适应地设定切分层以应对数据和设备异构性,这会显著影响系统性能;此外,如何在微调过程中有效降低通信开销也是一大难题。现有工作尚未解决这些问题。为此,我们提出SplitFT,一种面向大模型微调的自适应联邦分割学习系统。SplitFT允许不同客户端根据其计算资源和模型表现设置不同的切分层,并提出在切分层降低LoRA秩以减少通信开销。为模拟现实应用中的异构数据分布,我们引入基于长度的狄利克雷方法将训练数据分配至各客户端。大量实验结果表明,所提方法在多个主流基准上均优于当前最优方案,在微调时效性和模型性能方面均有显著提升。

原文摘要 · Abstract (English)

Federated Split Learning has been identified as an efficient approach to address the computational resource constraints of clients in classical federated learning, while guaranteeing data privacy for distributed model training across data owners. However, it faces some critical challenges when such a training strategy meets large language models (LLMs) for fine-tuning. Such challenges include setting the cutlayer adaptively across different clients to address the data and device heterogeneity issues, which affect the system performance significantly. In addition, efficiently reducing the communication overhead during the fine-tuning procedure is also another challenge. No work tries to address these challenges. To bridge this gap, we propose SplitTF, an adaptive federated split learning system for LLMs fine-tuning. SplitFT enables different clients to set different cut layers according to their computation resources and trained model performance. SplitFT also proposes to reduce the LoRA rank in cutlayer to reduce the communication overhead. In addition to simulating the heterogeneous data in real-world applications for our proposed split federated learning system, we propose a length-based Dirichlet approach to divide the training data into different clients. Extensive experimental results show that our proposed approach outperforms the state-of-the-art approach for fine-tuning time efficiency and model performance based on various popular benchmarks.

联邦学习大模型微调分割学习通信优化

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