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

FedGNN: Federated Graph Neural Network for Privacy-Preserving Recommendation

About

Graph neural network (GNN) is widely used for recommendation to model high-order interactions between users and items. Existing GNN-based recommendation methods rely on centralized storage of user-item graphs and centralized model learning. However, user data is privacy-sensitive, and the centralized storage of user-item graphs may arouse privacy concerns and risk. In this paper, we propose a federated framework for privacy-preserving GNN-based recommendation, which can collectively train GNN models from decentralized user data and meanwhile exploit high-order user-item interaction information with privacy well protected. In our method, we locally train GNN model in each user client based on the user-item graph inferred from the local user-item interaction data. Each client uploads the local gradients of GNN to a server for aggregation, which are further sent to user clients for updating local GNN models. Since local gradients may contain private information, we apply local differential privacy techniques to the local gradients to protect user privacy. In addition, in order to protect the items that users have interactions with, we propose to incorporate randomly sampled items as pseudo interacted items for anonymity. To incorporate high-order user-item interactions, we propose a user-item graph expansion method that can find neighboring users with co-interacted items and exchange their embeddings for expanding the local user-item graphs in a privacy-preserving way. Extensive experiments on six benchmark datasets validate that our approach can achieve competitive results with existing centralized GNN-based recommendation methods and meanwhile effectively protect user privacy.

Chuhan Wu, Fangzhao Wu, Yang Cao, Yongfeng Huang, Xing Xie• 2021

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy81.31
1215
Node ClassificationPubmed
Accuracy83.84
396
Node ClassificationAmazon Photo
Accuracy90.35
191
Node ClassificationOgbn-arxiv
Accuracy67.71
170
Node ClassificationAmazon Computers
Accuracy87.86
120
RecommendationMovieLens-100K (test)
RMSE0.92
55
RecommendationMovieLens 1M (test)--
46
Node ClassificationCiteseer
Accuracy70.72
45
Node ClassificationMinesweeper overlapping subgraph partitioning
AUC68.59
39
Node ClassificationRoman-empire overlapping subgraph partitioning
Accuracy37.46
39
Showing 10 of 40 rows

Other info

Follow for update