本文已被:浏览 636次 下载 6415次
Received:October 28, 2021 Revised:November 29, 2021
Received:October 28, 2021 Revised:November 29, 2021
中文摘要: 为了解决推荐系统的冷启动和稀疏性问题, 本文提出了一种基于异质信息网络的推荐模型. 传统的推荐方法无法在知识图谱表示学习中融入隐含的路径信息, 这样使得知识推荐系统性能较为一般. 本文提出的模型在异质信息网络中设置元路径, 通过图神经网络融入到知识图谱表示学习中. 再利用注意力网络连接推荐任务和知识图谱表示任务, 其可以学习两个任务之中潜在的特征, 并且能够增强推荐系统中被推荐项和知识图谱中实体的相互作用. 最后在推荐任务中进行用户点击率预测. 模型在公开数据集Book-Crossing和通过DBLP数据集构建的图谱上进行了实验. 最后结果表明, 模型在AUC, 召回率和F1值3个指标上均比其他算法有更好的表现.
Abstract:To address the cold-start and sparsity problems of recommendation systems, this study proposes a recommendation model based on a heterogeneous information network. Previous approaches are unable to take into account both knowledge graph representation learning and implicit path information, which makes the performance of knowledge recommendation systems mediocre. The proposed method sets meta-paths in the heterogeneous information network and integrates them into knowledge graph representation learning by the graph neural network (GNN). Next, the attention network is used to connect a recommendation task with a knowledge graph representation task. It can not only learn the potential features of the two tasks but also enhance the interactions between the recommended items in the recommendation system and the entities in the knowledge graph. Finally, the user click rate is predicted in the recommendation task. The method is experimented on the open dataset Book-Crossing and the knowledge graph constructed with the DBLP dataset, and the results demonstrate that the proposed model achieves better performance than that of other algorithms in indexes of area under curve (AUC), recall, and F1-score.
keywords: recommendation system heterogeneous information network knowledge graph graph neural network attention network deep learning
文章编号: 中图分类号: 文献标志码:
基金项目:国家自然科学基金(62072323); 江苏省高校自然科学研究项目重大项目(19KJA610002); 中国大学生创新训练计划(202010285028Z)
引用文本:
陈可迪,赵雷,陈心怡,施科男.基于异质信息网络的推荐模型.计算机系统应用,2022,31(8):361-368
CHEN Ke-Di,ZHAO Lei,CHEN Xin-Yi,SHI Ke-Nan.Recommendation Model Based on Heterogeneous Information Network.COMPUTER SYSTEMS APPLICATIONS,2022,31(8):361-368
陈可迪,赵雷,陈心怡,施科男.基于异质信息网络的推荐模型.计算机系统应用,2022,31(8):361-368
CHEN Ke-Di,ZHAO Lei,CHEN Xin-Yi,SHI Ke-Nan.Recommendation Model Based on Heterogeneous Information Network.COMPUTER SYSTEMS APPLICATIONS,2022,31(8):361-368