arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.00829cs.LG
收藏

基于图的个性化联邦学习

Personalized Federated Learning With Graph

Australian Artificial Intelligence Institute(澳大利亚人工智能研究所);University of Technology Sydney(悉尼科技大学);University of Washington(华盛顿大学);University of Maryland, College Park(马里兰大学帕克分校)

Fengwen Chen, Guodong Long, Zonghan Wu, Tianyi Zhou, Jing Jiang

摘要:

知识共享和模型个性化是个性化联邦学习(PFL)概念框架中的两个关键组成部分。现有的PFL方法集中在提出新的模型个性化机制,而简单地实现知识共享的聚合模型从所有的客户端,无论他们的关系图。本文旨在通过利用客户之间基于图的结构化信息来增强PFL中的知识共享过程。我们提出了一种新的结构化联邦学习(SFL)框架,使用客户端关系图和客户端的私有数据同时学习全局和个性化模型。我们将带图的SFL转化为一种新的优化问题,可以通过统一的框架来建模客户端复杂关系和基于图的结构拓扑。此外,除了使用现有的关系图,SFL可以扩展到学习客户端之间的隐藏关系。在交通和图像基准数据集上的实验证明了该方法的有效性。所有实现代码都可以在Github上找到

英文摘要:

Knowledge sharing and model personalization are two key components in the conceptual framework of personalized federated learning (PFL). Existing PFL methods focus on proposing new model personalization mechanisms while simply implementing knowledge sharing by aggregating models from all clients, regardless of their relation graph. This paper aims to enhance the knowledge-sharing process in PFL by leveraging the graph-based structural information among clients. We propose a novel structured federated learning (SFL) framework to learn both the global and personalized models simultaneously using client-wise relation graphs and clients' private data. We cast SFL with graph into a novel optimization problem that can model the client-wise complex relations and graph-based structural topology by a unified framework. Moreover, in addition to using an existing relation graph, SFL could be expanded to learn the hidden relations among clients. Experiments on traffic and image benchmark datasets can demonstrate the effectiveness of the proposed method. All implementation codes are available on Github

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

↑