arXiv:2508.18663cs.LGcs.AI2025-08被引 8

用稀疏专家模型替代LoRA,实现边缘设备高效联邦微调。

FFT-MoE: Efficient Federated Fine-Tuning for Foundation Models via Large-scale Sparse MoE under Heterogeneous Edge

  • 以稀疏MoE替代LoRA,按需激活专家适应不同设备资源。
  • 在非独立同分布数据下仍保持高泛化性能,收敛更快。
  • 适合资源差异大、数据异构的边缘智能场景应用。

随着基础模型(FMs)推动人工通用智能(AGI)发展,如何在隐私和资源约束下进行微调日益重要,尤其当高质量训练数据分布在分布式边缘设备上时。联邦学习(FL)通过联邦微调(FFT)提供解决方案,实现协同模型优化而无需共享原始数据。现有方法采用参数高效微调(PEFT)技术如低秩适配(LoRA),但其在异构联邦环境中面临两大挑战:客户端间LoRA配置结构不兼容,且对非独立同分布(non-IID)数据适应能力弱,影响收敛与泛化。为此,我们提出FFT-MoE,一种新型联邦微调框架,将LoRA替换为稀疏专家混合(MoE)适配器。每个客户端训练轻量级门控网络,选择性激活个性化专家子集,实现细粒度本地适应并保持聚合兼容性。为缓解设备与数据异构导致的专家负载不均,引入异构感知辅助损失,动态调节路由分布以确保专家多样性与均衡使用。跨IID与非IID场景的大量实验表明,FFT-MoE在泛化性能与训练效率上持续优于当前最优基线。

原文摘要 · Abstract (English)

As FMs drive progress toward Artificial General Intelligence (AGI), fine-tuning them under privacy and resource constraints has become increasingly critical particularly when highquality training data resides on distributed edge devices. Federated Learning (FL) offers a compelling solution through Federated Fine-Tuning (FFT), which enables collaborative model adaptation without sharing raw data. Recent approaches incorporate Parameter-Efficient Fine-Tuning (PEFT) techniques such as Low Rank Adaptation (LoRA) to reduce computational overhead. However, LoRA-based FFT faces two major limitations in heterogeneous FL environments: structural incompatibility across clients with varying LoRA configurations and limited adaptability to non-IID data distributions, which hinders convergence and generalization. To address these challenges, we propose FFT MoE, a novel FFT framework that replaces LoRA with sparse Mixture of Experts (MoE) adapters. Each client trains a lightweight gating network to selectively activate a personalized subset of experts, enabling fine-grained adaptation to local resource budgets while preserving aggregation compatibility. To further combat the expert load imbalance caused by device and data heterogeneity, we introduce a heterogeneity-aware auxiliary loss that dynamically regularizes the routing distribution to ensure expert diversity and balanced utilization. Extensive experiments spanning both IID and non-IID conditions demonstrate that FFT MoE consistently outperforms state of the art FFT baselines in generalization performance and training efficiency.

联邦学习MoE边缘计算微调

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