arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.15589cs.LGcs.DC
收藏

PFL-MoE:基于混合专家模型的个性化联邦学习

PFL-MoE: Personalized Federated Learning Based on Mixture of Experts

Sun Yat-sen University(中山大学);Tencent Inc.(腾讯公司)

Binbin Guo, Yuan Mei, Danyang Xiao, Weigang Wu, Ye Yin, Hongli Chang

摘要:

联邦学习(Federated learning,FL)是一种新兴的分布式机器学习范式,它避免了训练节点之间的数据共享,从而保护了数据隐私。在FL服务器的协调下,每个客户端使用自己的计算资源和私有数据集进行模型训练。可以通过汇总客户的培训结果来创建全局模型。为了应对高度非IID的数据分布,已经提出了个性化的联邦学习(PFL),以通过允许每个客户端学习个性化的模型来提高整体性能。然而,个性化模型的一个主要缺点是泛化能力的丧失。为了实现模型个性化,同时保持泛化,在本文中,我们提出了一种新的方法,名为PFL-MoE,通过MoE架构混合输出的个性化模型和全球模型。PFL-MoE是一种通用的方法,可以通过集成现有的PFL算法来实例化。特别地,我们提出了PFL-MF算法,这是PFL-MoE的一个实例,基于冻结基PFL算法。通过提高MoE选通网络的决策能力,进一步改进了PFL-MF算法,提出了一种改进的PFL-MFE算法。我们通过在具有非IID分区的Fashion-MNIST和CIFAR-10数据集上训练LeNet-5和VGG-16模型来证明PFL-MoE的有效性。

英文摘要:

Federated learning (FL) is an emerging distributed machine learning paradigm that avoids data sharing among training nodes so as to protect data privacy. Under coordination of the FL server, each client conducts model training using its own computing resource and private data set. The global model can be created by aggregating the training results of clients. To cope with highly non-IID data distributions, personalized federated learning (PFL) has been proposed to improve overall performance by allowing each client to learn a personalized model. However, one major drawback of a personalized model is the loss of generalization. To achieve model personalization while maintaining generalization, in this paper, we propose a new approach, named PFL-MoE, which mixes outputs of the personalized model and global model via the MoE architecture. PFL-MoE is a generic approach and can be instantiated by integrating existing PFL algorithms. Particularly, we propose the PFL-MF algorithm which is an instance of PFL-MoE based on the freeze-base PFL algorithm. We further improve PFL-MF by enhancing the decision-making ability of MoE gating network and propose a variant algorithm PFL-MFE. We demonstrate the effectiveness of PFL-MoE by training the LeNet-5 and VGG-16 models on the Fashion-MNIST and CIFAR-10 datasets with non-IID partitions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑