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
627
Node ClassificationAmazon Photo
Accuracy90.35
313
Node ClassificationOgbn-arxiv
Accuracy67.71
304
Node ClassificationAmazon Computers
Accuracy87.86
167
RecommendationMovieLens 1M (test)--
116
Federated LearningRoman-Empire
Time Consumption (s)3.33
84
Federated LearningCora
Time Consumption (s)2.28
84
RecommendationMovieLens-100K (test)
RMSE0.92
55
Node ClassificationCiteseer
Accuracy70.72
45
Showing 10 of 63 rows

Other info

Follow for update