arXiv:2509.15087cs.LG2025-09NeurIPS被引 12

针对异构数据的联邦微调,动态分配并选择LoRA专家以提升性能。

Adaptive LoRA Experts Allocation and Selection for Federated Fine-Tuning

  • 按客户端表征相似性自适应聚类,分配专属LoRA专家。
  • 采用自适应Top-M专家混合机制,每客户端灵活选择最优专家数。
  • 在异构数据下显著优于现有方法,通信开销低,适合隐私敏感场景。

大语言模型在多种任务中表现优异,但针对特定领域应用进行微调通常需要大量分布式领域的数据。联邦学习提供隐私保护方案,但在应用于大语言模型时面临计算资源限制。低秩适配(LoRA)是一种参数高效微调方法,但单一LoRA模块难以应对跨领域异构数据。本文针对联邦LoRA微调中的两个关键挑战:1)确定各异构客户端最优的LoRA专家数量与分配策略;2)使客户端根据自身数据特征选择性使用这些专家。提出FedLEASE框架,基于表征相似性自适应聚类客户端,分配并训练领域专用的LoRA专家,并引入自适应Top-M专家混合机制,允许每个客户端动态选择最优数量的专家。在多个基准数据集上的大量实验表明,FedLEASE在异构客户端设置下显著优于现有联邦微调方法,同时保持通信效率。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated impressive capabilities across various tasks, but fine-tuning them for domain-specific applications often requires substantial domain-specific data that may be distributed across multiple organizations. Federated Learning (FL) offers a privacy-preserving solution, but faces challenges with computational constraints when applied to LLMs. Low-Rank Adaptation (LoRA) has emerged as a parameter-efficient fine-tuning approach, though a single LoRA module often struggles with heterogeneous data across diverse domains. This paper addresses two critical challenges in federated LoRA fine-tuning: 1. determining the optimal number and allocation of LoRA experts across heterogeneous clients, and 2. enabling clients to selectively utilize these experts based on their specific data characteristics. We propose FedLEASE (Federated adaptive LoRA Expert Allocation and SElection), a novel framework that adaptively clusters clients based on representation similarity to allocate and train domain-specific LoRA experts. It also introduces an adaptive top-$M$ Mixture-of-Experts mechanism that allows each client to select the optimal number of utilized experts. Our extensive experiments on diverse benchmark datasets demonstrate that FedLEASE significantly outperforms existing federated fine-tuning approaches in heterogeneous client settings while maintaining communication efficiency.

联邦学习LoRA专家混合微调

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