Research output: Contribution to Journal/Magazine › Journal article › peer-review
<mark>Journal publication date</mark> | 31/01/2023 |
---|---|
<mark>Journal</mark> | IEEE Transactions on Industrial Informatics |
Issue number | 1 |
Volume | 19 |
Number of pages | 10 |
Pages (from-to) | 673-682 |
Publication Status | Published |
Early online date | 1/09/22 |
<mark>Original language</mark> | English |
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.