FedTAD: Topology-aware Data-free Knowledge Distillation for Subgraph Federated Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhu, Yinlin, Li, Xunkai, Wu, Zhengyu, Wu, Di, Hu, Miao, Li, Rong-Hua
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910422755442688
author Zhu, Yinlin
Li, Xunkai
Wu, Zhengyu
Wu, Di
Hu, Miao
Li, Rong-Hua
author_facet Zhu, Yinlin
Li, Xunkai
Wu, Zhengyu
Wu, Di
Hu, Miao
Li, Rong-Hua
contents Subgraph federated learning (subgraph-FL) is a new distributed paradigm that facilitates the collaborative training of graph neural networks (GNNs) by multi-client subgraphs. Unfortunately, a significant challenge of subgraph-FL arises from subgraph heterogeneity, which stems from node and topology variation, causing the impaired performance of the global GNN. Despite various studies, they have not yet thoroughly investigated the impact mechanism of subgraph heterogeneity. To this end, we decouple node and topology variation, revealing that they correspond to differences in label distribution and structure homophily. Remarkably, these variations lead to significant differences in the class-wise knowledge reliability of multiple local GNNs, misguiding the model aggregation with varying degrees. Building on this insight, we propose topology-aware data-free knowledge distillation technology (FedTAD), enhancing reliable knowledge transfer from the local model to the global model. Extensive experiments on six public datasets consistently demonstrate the superiority of FedTAD over state-of-the-art baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2404_14061
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FedTAD: Topology-aware Data-free Knowledge Distillation for Subgraph Federated Learning
Zhu, Yinlin
Li, Xunkai
Wu, Zhengyu
Wu, Di
Hu, Miao
Li, Rong-Hua
Machine Learning
Artificial Intelligence
Databases
Social and Information Networks
Subgraph federated learning (subgraph-FL) is a new distributed paradigm that facilitates the collaborative training of graph neural networks (GNNs) by multi-client subgraphs. Unfortunately, a significant challenge of subgraph-FL arises from subgraph heterogeneity, which stems from node and topology variation, causing the impaired performance of the global GNN. Despite various studies, they have not yet thoroughly investigated the impact mechanism of subgraph heterogeneity. To this end, we decouple node and topology variation, revealing that they correspond to differences in label distribution and structure homophily. Remarkably, these variations lead to significant differences in the class-wise knowledge reliability of multiple local GNNs, misguiding the model aggregation with varying degrees. Building on this insight, we propose topology-aware data-free knowledge distillation technology (FedTAD), enhancing reliable knowledge transfer from the local model to the global model. Extensive experiments on six public datasets consistently demonstrate the superiority of FedTAD over state-of-the-art baselines.
title FedTAD: Topology-aware Data-free Knowledge Distillation for Subgraph Federated Learning
topic Machine Learning
Artificial Intelligence
Databases
Social and Information Networks
url https://arxiv.org/abs/2404.14061