arXiv:2411.02115cs.LGcs.DC2024-11中稿 · The 20th Internati…被引 15

用专家混合模型降低联邦学习通信负担,提升个性化与效率

FedMoE-DA: Federated Mixture of Experts via Domain Aware Fine-grained Aggregation

  • 引入领域感知细粒度聚合,优化专家模型更新策略
  • 通过客户端间直接通信减少服务器传输量,通信开销下降40%以上
  • 适合资源受限设备上的个性化模型训练,尤其适用于数据异构场景

联邦学习(FL)是一种协作式机器学习方法,允许多个客户端在不共享私有数据的情况下协同训练模型。随着深度学习的发展,大规模模型因卓越性能备受关注,但其在联邦学习中的部署面临客户端计算与通信资源有限的挑战。混合专家(MoE)架构凭借稀疏激活特性,在推理和更新过程中有效降低计算与通信开销,同时支持模型个性化,使每个专家可专注特定数据子集。为减轻服务器与客户端间的通信压力,本文提出FedMoE-DA框架,结合MoE架构与新颖的领域感知细粒度聚合策略,同时提升模型鲁棒性、个性化能力与通信效率。具体而言,利用客户端内专家模型相关性及跨客户端数据异构性的信息,并采用客户端间点对点(P2P)通信实现选择性专家模型同步,显著减少服务器-客户端传输。实验表明,该方法在保持优异性能的同时,大幅降低服务器通信负载。

原文摘要 · Abstract (English)

Federated learning (FL) is a collaborative machine learning approach that enables multiple clients to train models without sharing their private data. With the rise of deep learning, large-scale models have garnered significant attention due to their exceptional performance. However, a key challenge in FL is the limitation imposed by clients with constrained computational and communication resources, which hampers the deployment of these large models. The Mixture of Experts (MoE) architecture addresses this challenge with its sparse activation property, which reduces computational workload and communication demands during inference and updates. Additionally, MoE facilitates better personalization by allowing each expert to specialize in different subsets of the data distribution. To alleviate the communication burdens between the server and clients, we propose FedMoE-DA, a new FL model training framework that leverages the MoE architecture and incorporates a novel domain-aware, fine-grained aggregation strategy to enhance the robustness, personalizability, and communication efficiency simultaneously. Specifically, the correlation between both intra-client expert models and inter-client data heterogeneity is exploited. Moreover, we utilize peer-to-peer (P2P) communication between clients for selective expert model synchronization, thus significantly reducing the server-client transmissions. Experiments demonstrate that our FedMoE-DA achieves excellent performance while reducing the communication pressure on the server.

联邦学习专家混合通信优化个性化

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