针对移动端资源有限问题,用专家选择优化联邦微调大模型。
HFedMoE: Resource-aware Heterogeneous Federated Learning with Mixture-of-Experts
- 按专家对本地微调的贡献度动态选子集,适配不同设备算力。
- 在多个数据集上实现更高准确率和更快收敛速度。
- 适合算力差异大的设备协同训练大模型,如手机端部署。
联邦学习(FL)可在不泄露数据隐私的前提下微调大型语言模型(LLMs),但模型体量过大导致资源受限设备(如手机)难以进行本地训练。为此,稀疏激活的混合专家(MoE)模型成为高效计算方案,仅激活部分专家以降低负担而不牺牲性能。然而将MoE引入联邦微调仍面临三大挑战:一是缺乏可靠指标评估各专家对本地微调的影响,难以合理选择;二是客户端算力异构,动态激活机制可能超出资源受限设备承受能力;三是客户端专属专家子集与路由偏好导致全局聚合困难,更新错位与门控网络不一致引发破坏性干扰。为此,我们提出HFedMoE框架,为每个客户端定制专家子集,实现资源感知的异构联邦微调。具体地,基于专家对微调性能的贡献识别其重要性,并从信息瓶颈角度自适应选择符合客户端计算预算的专家子集。同时设计一种稀疏感知的聚合策略,按重要性加权聚合活跃专家和门控参数。大量实验表明,HFedMoE在训练精度与收敛速度上均优于现有先进基准。
原文摘要 · Abstract (English)
While federated learning (FL) enables fine-tuning of large language models (LLMs) without compromising data privacy, the substantial size of an LLM renders on-device training impractical for resource-constrained clients, such as mobile devices. Thus, Mixture-of-Experts (MoE) models have emerged as a computation-efficient solution, which activates only a sparse subset of experts during model training to reduce computing burden without sacrificing performance. Though integrating MoE into FL fine-tuning holds significant potential, it still encounters three key challenges: i) selecting appropriate experts for clients remains challenging due to the lack of a reliable metric to measure each expert's impact on local fine-tuning performance, ii) the heterogeneous computing resources across clients severely hinder MoE-based LLM fine-tuning, as dynamic expert activations across diverse input samples can overwhelm resource-constrained devices, and iii) client-specific expert subsets and routing preference undermine global aggregation, where misaligned expert updates and inconsistent gating networks in troduce destructive interference. To address these challenges, we propose HFedMoE, a heterogeneous MoE-based FL fine-tuning framework that customizes a subset of experts to each client for computation-efficient LLM fine-tuning. Specifically, HFedMoE identifies the expert importance based on its contributions to fine-tuning performance, and then adaptively selects a subset of experts from an information bottleneck perspective to align with each client' s computing budget. A sparsity-aware model aggregation strategy is also designed to aggregate the actively fine-tuned experts and gating parameters with importance weighted contributions. Extensive experiments demonstrate that HFedMoE outperforms state-of-the-art benchmarks in training accuracy and convergence speed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。