Li Mingqi, Ma Wenming, Chu Zihao
School of Computer and Control Engineering, Yantai University, YanTai, 264005, China.
Neural Netw. 2025 Apr;184:107116. doi: 10.1016/j.neunet.2024.107116. Epub 2025 Jan 4.
Recommender systems are widely used in various applications. Knowledge graphs are increasingly used to improve recommendation performance by extracting valuable information from user-item interactions. However, current methods do not effectively use fine-grained information within the knowledge graph. Additionally, some recommendation methods based on graph neural networks tend to overlook the importance of entities to users when performing aggregation operations. To alleviate these issues, we introduce a knowledge-graph-based graph neural network (PIFSA-GNN) for recommendation with two key components. The first component, user preference interaction fusion, incorporates user auxiliary information in the recommendation process. This enhances the influence of users on the recommendation model. The second component is an attention mechanism called user preference swap attention, which improves entity weight calculation for effectively aggregating neighboring entities. Our method was extensively tested on three real-world datasets. On the movie dataset, our method outperforms the best baseline by 1.3% in AUC and 2.8% in F1; Hit@1 increases by 0.7%, Hit@5 by 0.6%, and Hit@10 by 1.0%. On the restaurant dataset, AUC improves by 2.6% and F1 by 7.2%; Hit@1 increases by 1.3%, Hit@5 by 3.7%, and Hit@10 by 2.9%. On the music dataset, AUC improves by 0.9% and F1 by 0.4%; Hit@1 increases by 3.3%, Hit@5 by 1.2%, and Hit@10 by 0.2%. The results show that it outperforms baseline methods.
推荐系统广泛应用于各种应用场景。知识图谱越来越多地被用于通过从用户-物品交互中提取有价值的信息来提高推荐性能。然而,当前的方法并没有有效地利用知识图谱中的细粒度信息。此外,一些基于图神经网络的推荐方法在执行聚合操作时往往会忽略实体对用户的重要性。为了缓解这些问题,我们引入了一种基于知识图谱的图神经网络(PIFSA-GNN)用于推荐,它有两个关键组件。第一个组件是用户偏好交互融合,在推荐过程中纳入用户辅助信息。这增强了用户对推荐模型的影响。第二个组件是一种名为用户偏好交换注意力的注意力机制,它改进了实体权重计算以有效地聚合相邻实体。我们的方法在三个真实世界数据集上进行了广泛测试。在电影数据集上,我们的方法在AUC上比最佳基线方法高出1.3%,在F1上高出2.8%;Hit@1提高了0.7%,Hit@5提高了0.6%,Hit@10提高了1.0%。在餐厅数据集上,AUC提高了2.6%,F1提高了7.2%;Hit@1提高了1.3%,Hit@5提高了3.7%,Hit@10提高了2.9%。在音乐数据集上,AUC提高了0.9%,F1提高了0.4%;Hit@1提高了3.3%,Hit@5提高了1.2%,Hit@10提高了0.2%。结果表明它优于基线方法。