arXiv:2411.19128cs.LG2024-11被引 13

让大模型在不同机构间协作训练时,自动适配各自数据特点的个性化架构。

Personalized Federated Fine-Tuning for LLMs via Data-Driven Heterogeneous Model Architectures

  • 按客户端数据特征动态分配不同低秩适配模块,实现异构模型协同。
  • 在7个场景中平均提升性能5.97%,且开销可控。
  • 适合医疗、金融等数据敏感且分布差异大的行业使用。

大型语言模型正广泛应用于网络应用,其效果依赖于大规模指令数据的微调。然而,这些数据常包含重要或敏感信息,限制了企业间公开共享。联邦学习(FL)可在不访问原始数据的前提下实现模型协作微调。现有联邦大模型微调方法通常采用统一模型架构,难以适应不同领域和任务的客户端数据异质性,例如医疗机构与金融机构因领域差异可能需要不同的模型结构。为此,我们提出FedAMoLE,一种轻量级个性化联邦学习框架,支持数据驱动的异构模型架构。该框架包含异构低秩适配(LoRA)专家混合模块,用于聚合架构各异的模型,并采用基于反向选择的专家分配策略,根据数据分布为每个客户端定制模型架构。在七个场景下的实验表明,相比现有方法,FedAMoLE在客户端性能上平均提升5.97%,同时保持较低的内存、通信与计算开销。

原文摘要 · Abstract (English)

Large language models (LLMs) are increasingly powering web-based applications, whose effectiveness relies on fine-tuning with large-scale instruction data. However, such data often contains valuable or sensitive information that limits its public sharing among business organizations. Federated learning (FL) enables collaborative fine-tuning of LLMs without accessing raw data. Existing approaches to federated LLM fine-tuning usually adopt a uniform model architecture, making it challenging to fit highly heterogeneous client-side data in varying domains and tasks, e.g., hospitals and financial institutions conducting federated fine-tuning may require different LLM architectures due to the distinct nature of their domains and tasks. To address this, we propose FedAMoLE, a lightweight personalized FL framework that enables data-driven heterogeneous model architectures. It features a heterogeneous mixture of low-rank adaptation (LoRA) experts module to aggregate architecturally heterogeneous models and a reverse selection-based expert assignment strategy to tailor model architectures for each client based on data distributions. Experiments across seven scenarios demonstrate that FedAMoLE improves client-side performance by an average of 5.97% over existing approaches while maintaining practical memory, communication, and computation overhead.

联邦学习大模型微调个性化建模

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