Federated Learning-Based Cross-Enterprise Recommendation With Graph Neural Networks
Zheng Li,Muhammad Bilal,Xiaolong Xu,Jielin Jiang,Yan Cui
DOI: https://doi.org/10.1109/tii.2022.3203395
IF: 12.3
2022-11-13
IEEE Transactions on Industrial Informatics
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.
automation & control systems,computer science, interdisciplinary applications,engineering, industrial