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

Federated Graph Learning with Structure Proxy Alignment

About

Federated Graph Learning (FGL) aims to learn graph learning models over graph data distributed in multiple data owners, which has been applied in various applications such as social recommendation and financial fraud detection. Inherited from generic Federated Learning (FL), FGL similarly has the data heterogeneity issue where the label distribution may vary significantly for distributed graph data across clients. For instance, a client can have the majority of nodes from a class, while another client may have only a few nodes from the same class. This issue results in divergent local objectives and impairs FGL convergence for node-level tasks, especially for node classification. Moreover, FGL also encounters a unique challenge for the node classification task: the nodes from a minority class in a client are more likely to have biased neighboring information, which prevents FGL from learning expressive node embeddings with Graph Neural Networks (GNNs). To grapple with the challenge, we propose FedSpray, a novel FGL framework that learns local class-wise structure proxies in the latent space and aligns them to obtain global structure proxies in the server. Our goal is to obtain the aligned structure proxies that can serve as reliable, unbiased neighboring information for node classification. To achieve this, FedSpray trains a global feature-structure encoder and generates unbiased soft targets with structure proxies to regularize local training of GNN models in a personalized way. We conduct extensive experiments over four datasets, and experiment results validate the superiority of FedSpray compared with other baselines. Our code is available at https://github.com/xbfu/FedSpray.

Xingbo Fu, Zihan Chen, Binchi Zhang, Chen Chen, Jundong Li• 2024

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy79.21
1215
Node ClassificationPubmed
Accuracy84.58
627
Node ClassificationAmazon Photo
Accuracy92.59
313
Node ClassificationOgbn-arxiv
Accuracy64.77
304
Node ClassificationAmazon Computers
Accuracy89.46
167
Node ClassificationCiteseer
Accuracy75.62
45
Node ClassificationOgbn-arxiv
Overall F139.44
40
Node ClassificationPhysics
Overall F10.8412
34
Node ClassificationCS
Overall F180.22
34
Node ClassificationCora
Accuracy71.22
27
Showing 10 of 11 rows

Other info

Follow for update