FedGTA: Topology-aware Averaging for Federated Graph Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Li, Xunkai, Wu, Zhengyu, Zhang, Wentao, Zhu, Yinlin, Li, Rong-Hua, Wang, Guoren
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917572180443136
author Li, Xunkai
Wu, Zhengyu
Zhang, Wentao
Zhu, Yinlin
Li, Rong-Hua
Wang, Guoren
author_facet Li, Xunkai
Wu, Zhengyu
Zhang, Wentao
Zhu, Yinlin
Li, Rong-Hua
Wang, Guoren
contents Federated Graph Learning (FGL) is a distributed machine learning paradigm that enables collaborative training on large-scale subgraphs across multiple local systems. Existing FGL studies fall into two categories: (i) FGL Optimization, which improves multi-client training in existing machine learning models; (ii) FGL Model, which enhances performance with complex local models and multi-client interactions. However, most FGL optimization strategies are designed specifically for the computer vision domain and ignore graph structure, presenting dissatisfied performance and slow convergence. Meanwhile, complex local model architectures in FGL Models studies lack scalability for handling large-scale subgraphs and have deployment limitations. To address these issues, we propose Federated Graph Topology-aware Aggregation (FedGTA), a personalized optimization strategy that optimizes through topology-aware local smoothing confidence and mixed neighbor features. During experiments, we deploy FedGTA in 12 multi-scale real-world datasets with the Louvain and Metis split. This allows us to evaluate the performance and robustness of FedGTA across a range of scenarios. Extensive experiments demonstrate that FedGTA achieves state-of-the-art performance while exhibiting high scalability and efficiency. The experiment includes ogbn-papers100M, the most representative large-scale graph database so that we can verify the applicability of our method to large-scale graph learning. To the best of our knowledge, our study is the first to bridge large-scale graph learning with FGL using this optimization strategy, contributing to the development of efficient and scalable FGL methods.
format Preprint
id arxiv_https___arxiv_org_abs_2401_11755
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FedGTA: Topology-aware Averaging for Federated Graph Learning
Li, Xunkai
Wu, Zhengyu
Zhang, Wentao
Zhu, Yinlin
Li, Rong-Hua
Wang, Guoren
Machine Learning
Artificial Intelligence
Databases
Social and Information Networks
Federated Graph Learning (FGL) is a distributed machine learning paradigm that enables collaborative training on large-scale subgraphs across multiple local systems. Existing FGL studies fall into two categories: (i) FGL Optimization, which improves multi-client training in existing machine learning models; (ii) FGL Model, which enhances performance with complex local models and multi-client interactions. However, most FGL optimization strategies are designed specifically for the computer vision domain and ignore graph structure, presenting dissatisfied performance and slow convergence. Meanwhile, complex local model architectures in FGL Models studies lack scalability for handling large-scale subgraphs and have deployment limitations. To address these issues, we propose Federated Graph Topology-aware Aggregation (FedGTA), a personalized optimization strategy that optimizes through topology-aware local smoothing confidence and mixed neighbor features. During experiments, we deploy FedGTA in 12 multi-scale real-world datasets with the Louvain and Metis split. This allows us to evaluate the performance and robustness of FedGTA across a range of scenarios. Extensive experiments demonstrate that FedGTA achieves state-of-the-art performance while exhibiting high scalability and efficiency. The experiment includes ogbn-papers100M, the most representative large-scale graph database so that we can verify the applicability of our method to large-scale graph learning. To the best of our knowledge, our study is the first to bridge large-scale graph learning with FGL using this optimization strategy, contributing to the development of efficient and scalable FGL methods.
title FedGTA: Topology-aware Averaging for Federated Graph Learning
topic Machine Learning
Artificial Intelligence
Databases
Social and Information Networks
url https://arxiv.org/abs/2401.11755