Home > Research > Publications & Outputs > Federated Learning-Based Cross-Enterprise Recom...

Links

Text available via DOI:

View graph of relations

Federated Learning-Based Cross-Enterprise Recommendation With Graph Neural Networks

Research output: Contribution to Journal/MagazineJournal articlepeer-review

Published
Close
<mark>Journal publication date</mark>31/01/2023
<mark>Journal</mark>IEEE Transactions on Industrial Informatics
Issue number1
Volume19
Number of pages10
Pages (from-to)673-682
Publication StatusPublished
Early online date1/09/22
<mark>Original language</mark>English

Abstract

Recommender systems are technology-driven marketing solutions for businesses that analyze user behavior data. However, collaborative data sharing between enterprises is often prohibited by privacy protection regulations, leading to insufficient data for graph neural networks (GNNs) training. Fortunately, federated learning (FL), a collaborative training framework without exposing source data, can be applied congruently. Nevertheless, most of FL-based GNN model training methods adopt federated averaging, which performs poorly on highly heterogeneous graph data. To solve this problem, a FL-based GNN Model Training framework for cross-enterprise recommendation, named FL-GMT, is proposed. Specifically, a GNN-based recommendation model is deployed as the local training model. Then, considering the performance inequity caused by uneven sample quality, a loss-based federated aggregation algorithm is designed, effectively improving the performance of disadvantaged participants. To improve the system stability at the end of the aggregation, a dynamic update method of loss attention is designed. Extensive experiments on benchmark datasets demonstrate that FL-GMT outperforms baselines in terms of system fairness, stability, and accuracy.