arXiv:2603.21276cs.LGcs.AI2026-03被引 5

解决联邦学习中专家模型因数据异构导致的路由混乱与语义模糊问题。

Aggregation Alignment for Federated Learning with Mixture-of-Experts under Data Heterogeneity

  • 通过路由一致性加权与分布正则化,对齐客户端的专家选择偏好。
  • 显式量化专家语义一致性,仅聚合语义对齐客户端的更新。
  • 适合隐私敏感场景下大规模语言模型的联邦微调,尤其在非独立同分布数据下表现优异。

大型语言模型(LLMs)越来越多采用混合专家(MoE)架构,在提升模型容量的同时降低计算开销。微调这些基于MoE的LLM通常需要访问分布式且隐私敏感的数据,使得集中式微调不切实际。因此,联邦学习(FL)为协同微调基于MoE的LLM提供了可行方案,使各客户端在不泄露数据隐私的前提下整合多样知识。然而,将基于MoE的LLM微调引入联邦学习时,由于客户端间固有的数据异构性,面临两大聚合挑战:(i)本地数据分布差异导致客户端发展出不同的门控选择偏好,直接参数聚合会产生‘一刀切’的全局门控网络;(ii)相同索引的专家在不同客户端上发展出不同的语义角色,引发专家语义模糊并削弱专家专业化能力。为此,我们提出FedAlign-MoE,一种联合实现路由一致性和专家语义对齐的联邦聚合对齐框架。具体而言,通过一致性加权对齐路由分布,并利用分布正则化优化本地门控网络,保持跨客户端稳定性而不覆盖本地差异化偏好。同时,显式量化同一索引专家在客户端间的语义一致性,选择性聚合语义对齐客户端的更新,确保全局专家具备稳定且专业化的功能角色。大量实验表明,FedAlign-MoE优于现有先进基准,在非独立同分布联邦环境中实现更快收敛与更高准确率。

原文摘要 · Abstract (English)

Large language models (LLMs) increasingly adopt Mixture-of-Experts (MoE) architectures to scale model capacity while reducing computation. Fine-tuning these MoE-based LLMs often requires access to distributed and privacy-sensitive data, making centralized fine-tuning impractical. Federated learning (FL) therefore provides a paradigm to collaboratively fine-tune MoE-based LLMs, enabling each client to integrate diverse knowledge without compromising data privacy. However, the integration of MoE-based LLM fine-tuning into FL encounters two critical aggregation challenges due to inherent data heterogeneity across clients: (i) divergent local data distributions drive clients to develop distinct gating preference for localized expert selection, causing direct parameter aggregation to produce a ``one-size-fits-none'' global gating network, and (ii) same-indexed experts develop disparate semantic roles across clients, leading to expert semantic blurring and the degradation of expert specialization. To address these challenges, we propose FedAlign-MoE, a federated aggregation alignment framework that jointly enforces routing consistency and expert semantic alignment. Specifically, FedAlign-MoE aggregates gating behaviors by aligning routing distributions through consistency weighting and optimizes local gating networks through distribution regularization, maintaining cross-client stability without overriding discriminative local preferences. Meanwhile, FedAlign-MoE explicitly quantifies semantic consistency among same-indexed experts across clients and selectively aggregates updates from semantically aligned clients, ensuring stable and specialized functional roles for global experts. Extensive experiments demonstrate that FedAlign-MoE outperforms state-of-the-art benchmarks, achieving faster convergence and superior accuracy in non-IID federated environments.

联邦学习专家模型数据异构语义对齐

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