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

Hybrid FedGraph:一种使用图卷积神经网络的高效混合联邦学习算法

Hybrid FedGraph: An efficient hybrid federated learning algorithm using graph convolutional neural network

The Catholic University of Korea(韩国天主教大学);Northwestern University(西北大学)

Jaeyeon Jang, Diego Klabjan, Veena Mendiratta, Fanfei Meng

AI总结:

针对现有联邦学习对更常见的混合数据分布场景研究不足的问题,提出通用算法FedGraph,引入图卷积神经网络捕捉特征共享信息,并设计隐私保护的聚类算法聚合客户端深度特征。

摘要:

联邦学习是一种新兴的范式,可在分布式客户端上进行机器学习模型的去中心化训练,且无需向中央服务器泄露数据。现有大多数研究聚焦于水平或垂直数据分布场景,前者中各客户端拥有共享特征的不同样本,后者中各客户端仅完全共享样本索引。然而,混合方案的研究要少得多,尽管它在现实世界中更为常见。因此,本文提出了一种通用算法FedGraph,该算法引入图卷积神经网络,在从部分客户端学习特征的同时捕捉特征共享信息。我们还开发了一种简单但有效的聚类算法,可在保护数据隐私的前提下聚合各客户端深度神经网络生成的特征。

英文摘要:

Federated learning is an emerging paradigm for decentralized training of machine learning models on distributed clients, without revealing the data to the central server. Most existing works have focused on horizontal or vertical data distributions, where each client possesses different samples with shared features, or each client fully shares only sample indices, respectively. However, the hybrid scheme is much less studied, even though it is much more common in the real world. Therefore, in this paper, we propose a generalized algorithm, FedGraph, that introduces a graph convolutional neural network to capture feature-sharing information while learning features from a subset of clients. We also develop a simple but effective clustering algorithm that aggregates features produced by the deep neural networks of each client while preserving data privacy.

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

↑