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

Li, Zheng and Bilal, Muhammad and Xu, Xiaolong and Jiang, Jielin and Cui, Yan (2023) Federated Learning-Based Cross-Enterprise Recommendation With Graph Neural Networks. IEEE Transactions on Industrial Informatics, 19 (1). pp. 673-682. ISSN 1551-3203

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.