Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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.

Yinlin Zhu, Xunkai Li, Zhengyu Wu, Di Wu, Miao Hu, Rong-Hua Li• 2024

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy78.29
1215
Node ClassificationPubmed
Accuracy84.79
627
Node ClassificationwikiCS
Accuracy75.56
329
Node ClassificationAmazon Photo
Accuracy91.01
313
Node ClassificationOgbn-arxiv
Accuracy68.34
304
Node ClassificationOgbn-arxiv
Accuracy72.35
235
Node ClassificationAmazon Computers
Accuracy86.69
167
Federated LearningRoman-Empire
Time Consumption (s)4.88
84
Federated LearningCora
Time Consumption (s)4.91
84
Node ClassificationQuestions non-overlapping subgraph partitioning
AUC68.89
45
Showing 10 of 52 rows

Other info

Follow for update