Rong Zhang, Yuan Liu, Yang Li
School of Internet of Things Engineering, Jiangsu Vocational College of Information Technology, Wuxi, 214153, China.
School of Artificial Intelligence and Computer, Jiangnan University, WuXi, 214122, China.
Sci Rep. 2024 Oct 4;14(1):23051. doi: 10.1038/s41598-024-74516-z.
Integrating the Knowledge Graphs (KGs) into recommendation systems enhances personalization and accuracy. However, the long-tail distribution of knowledge graphs often leads to data sparsity, which limits the effectiveness in practical applications. To address this challenge, this study proposes a knowledge-aware recommendation algorithm framework that incorporates multi-level contrastive learning. This framework enhances the Collaborative Knowledge Graph (CKG) through a random edge dropout method, which constructs feature representations at three levels: user-user interactions, item-item interactions and user-item interactions. A dynamic attention mechanism is employed in the Graph Attention Networks (GAT) for modeling the KG. Combined with the nonlinear transformation and Momentum Contrast (Moco) strategy for contrastive learning, it can effectively extract high-quality feature information. Additionally, multi-level contrastive learning, as an auxiliary self-supervised task, is jointly trained with the primary supervised task, which further enhances recommendation performance. Experimental results on the MovieLens and Amazon-books datasets demonstrate that this framework effectively improves the performance of knowledge graph-based recommendations, addresses the issue of data sparsity, and outperforms other baseline models across multiple evaluation metrics.
将知识图谱(KGs)集成到推荐系统中可增强个性化和准确性。然而,知识图谱的长尾分布常常导致数据稀疏性,这限制了其在实际应用中的有效性。为应对这一挑战,本研究提出了一种纳入多级对比学习的知识感知推荐算法框架。该框架通过随机边丢弃方法增强协同知识图谱(CKG),在用户-用户交互、物品-物品交互和用户-物品交互三个层面构建特征表示。图注意力网络(GAT)中采用动态注意力机制对知识图谱进行建模。结合用于对比学习的非线性变换和动量对比(Moco)策略,它能够有效提取高质量特征信息。此外,多级对比学习作为辅助自监督任务,与主要监督任务联合训练,进一步提升推荐性能。在MovieLens和Amazon-books数据集上的实验结果表明,该框架有效提高了基于知识图谱的推荐性能,解决了数据稀疏性问题,并且在多个评估指标上优于其他基线模型。