LLM2D
PM-MOE: 私有模型参数上的混合专家 untuk 个性化联邦学习
PM-MOE: Mixture of Experts on Private Model Parameters for Personalized Federated Learning
作者: Yu Feng, Yangli-ao Geng, Yifan Zhu, Zongfu Han, Xie Yu, Kaiwen Xue, Haoran Luo, Mengyang Sun, Guangwei Zhang, Meina Song
发布日期: 2/4/2025
arXiv ID: oai:arXiv.org:2502.00354v1

摘要

arXiv:2502.00354v1 宣布类型: cross 摘要: 联邦学习(FL)因其隐私保护和协作学习的能力而引起了广泛关注。由于显著的统计异质性,传统的FL难以在多样化数据域中泛化共享模型。个性化联邦学习通过将模型分为一个全球共享部分和一个本地私有部分来解决这一问题,本地模型纠正由全局模型引入的表示偏差。然而,局部收敛的参数更能捕捉域特定的知识,而当前方法忽视了这些参数的潜在好处。为了应对这些局限性,我们提出了一种PM-MoE架构。该架构结合了个性化模块的混合以及基于能量的个性化模块去噪,使每个客户端能够从其他客户端选择有益的个性化参数。我们将PM-MoE架构应用于九种最近的基于模型划分的个性化联邦学习算法,实现性能改进,且额外训练量极小。在六个广泛采用的数据集和两个异质性设置下的广泛实验验证了我们方法的有效性。源代码可在https://github.com/dannis97500/PM-MOE获得。