Suppr超能文献

基于多级对比学习的增强知识图谱推荐算法

Enhanced knowledge graph recommendation algorithm based on multi-level contrastive learning.

作者信息

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.

Abstract

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数据集上的实验结果表明,该框架有效提高了基于知识图谱的推荐性能,解决了数据稀疏性问题,并且在多个评估指标上优于其他基线模型。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/dfe5/11452630/bf51fe87c49c/41598_2024_74516_Fig1_HTML.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验