4 个月前

知识图谱卷积网络在推荐系统中的应用

知识图谱卷积网络在推荐系统中的应用

摘要

为了缓解基于协同过滤的推荐系统中存在的稀疏性和冷启动问题,研究人员和工程师通常会收集用户和项目的属性,并设计精巧的算法来利用这些额外的信息。一般来说,这些属性并不是孤立存在的,而是相互关联的,从而形成了一个知识图谱(Knowledge Graph, KG)。在本文中,我们提出了一种知识图谱卷积网络(Knowledge Graph Convolutional Networks, KGCN),这是一种端到端框架,通过挖掘项目在知识图谱上的相关属性,有效捕捉项目之间的关联性。为了自动发现知识图谱的高阶结构信息和语义信息,我们为知识图谱中的每个实体从其邻居中采样作为其感受野,然后在计算给定实体的表示时结合邻域信息和偏置。感受野可以扩展到多个跳数之外,以建模高阶近似信息并捕捉用户的潜在远距离兴趣。此外,我们将提出的 KGCN 实现在小批量模式下,这使得我们的模型能够在大型数据集和知识图谱上运行。我们将所提出的模型应用于电影、书籍和音乐推荐的三个数据集,并且实验结果表明我们的方法优于强大的推荐基线模型。

代码仓库

KanchiShimono/KGCN
tf
GitHub 中提及
youngch12/Cluster_KGCN
tf
GitHub 中提及
Ki-Seki/KGCN-pytorch-updated
pytorch
GitHub 中提及
mostsuperman/KGCN-ML
tf
GitHub 中提及
johnnyjana730/MVIN
tf
GitHub 中提及
zzaebok/KGCN-pytorch
pytorch
GitHub 中提及
hwwang55/KGCN
官方
tf
GitHub 中提及
mostsuperman/test
tf
GitHub 中提及

基准测试

基准方法指标
click-through-rate-prediction-on-bookKGCN-sum
AUC: 0.738
F1: 0.688
click-through-rate-prediction-on-lastfmKGCN-concat
AUC: 0.796
F1: 0.721
link-prediction-on-movielens-25mKGCN
Hits@10: 0.771
nDCG@10: 0.4699
link-prediction-on-yelpKGCN
HR@10: 0.8125
nDCG@10: 0.4668

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
知识图谱卷积网络在推荐系统中的应用 | 论文 | HyperAI超神经