Personalized Federated Learning via Learning Dynamic Graphs

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhou, Ziran, Gao, Guanyu, Wu, Xiaohu, Lyu, Yan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910863697379328
author Zhou, Ziran
Gao, Guanyu
Wu, Xiaohu
Lyu, Yan
author_facet Zhou, Ziran
Gao, Guanyu
Wu, Xiaohu
Lyu, Yan
contents Personalized Federated Learning (PFL) aims to train a personalized model for each client that is tailored to its local data distribution, learning fails to perform well on individual clients due to variations in their local data distributions. Most existing PFL methods focus on personalizing the aggregated global model for each client, neglecting the fundamental aspect of federated learning: the regulation of how client models are aggregated. Additionally, almost all of them overlook the graph structure formed by clients in federated learning. In this paper, we propose a novel method, Personalized Federated Learning with Graph Attention Network (pFedGAT), which captures the latent graph structure between clients and dynamically determines the importance of other clients for each client, enabling fine-grained control over the aggregation process. We evaluate pFedGAT across multiple data distribution scenarios, comparing it with twelve state of the art methods on three datasets: Fashion MNIST, CIFAR-10, and CIFAR-100, and find that it consistently performs well.
format Preprint
id arxiv_https___arxiv_org_abs_2503_05474
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Personalized Federated Learning via Learning Dynamic Graphs
Zhou, Ziran
Gao, Guanyu
Wu, Xiaohu
Lyu, Yan
Machine Learning
Artificial Intelligence
Personalized Federated Learning (PFL) aims to train a personalized model for each client that is tailored to its local data distribution, learning fails to perform well on individual clients due to variations in their local data distributions. Most existing PFL methods focus on personalizing the aggregated global model for each client, neglecting the fundamental aspect of federated learning: the regulation of how client models are aggregated. Additionally, almost all of them overlook the graph structure formed by clients in federated learning. In this paper, we propose a novel method, Personalized Federated Learning with Graph Attention Network (pFedGAT), which captures the latent graph structure between clients and dynamically determines the importance of other clients for each client, enabling fine-grained control over the aggregation process. We evaluate pFedGAT across multiple data distribution scenarios, comparing it with twelve state of the art methods on three datasets: Fashion MNIST, CIFAR-10, and CIFAR-100, and find that it consistently performs well.
title Personalized Federated Learning via Learning Dynamic Graphs
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2503.05474