计算机工程与设计2024,Vol.45Issue(3) :911-917.DOI:10.16208/j.issn1000-7024.2024.03.037

基于协同信号的知识图注意力网络推荐算法

Collaborative signal knowledge graph attention network for recommender algorithm

郭伟 裴帅华
计算机工程与设计2024,Vol.45Issue(3) :911-917.DOI:10.16208/j.issn1000-7024.2024.03.037

基于协同信号的知识图注意力网络推荐算法

Collaborative signal knowledge graph attention network for recommender algorithm

郭伟 1裴帅华1
扫码查看

作者信息

  • 1. 西安科技大学通信与信息工程学院,陕西西安 710600
  • 折叠

摘要

针对现有基于知识图谱的推荐算法大多采用随机采样的方法构建感受野导致部分重要信息丢失的问题,提出一种基于协同信号的知识图注意力网络推荐算法(CKGAN).设计协作传播层,将用户的历史项目和待推荐项目在知识图谱中进行传播;为区分不同关系下的实体对中心节点的重要性不同,在聚合邻域信息时引入图注意力网络;为解决用户的不同兴趣,在预测层中,面对不同的候选集动态建模用户的向量表示.通过在Last.FM、Book-Cross、MovieLens-1M这3个真实的公共数据集上与主流的推荐算法进行对比实验,其结果表明,CKGAN在CTR预测和top-K推荐中都有显著提升.

Abstract

The existing recommendation algorithm based on knowledge graph adopts the method of random sampling to construct the receptive field,which leads to the loss of some important information.Regarding the issues,a recommendation algorithm called collaborative signal knowledge graph attention network for recommender algorithm(CKGAN)was proposed.The collabo-rative propagation layer was designed to spread users'historical items and items to be recommended in KG.To distinguish the importance of entities under different relations to the central node,graph attention network was introduced in the propagation process.To address the different interests of users,in the prediction layer,the vector representation of users was dynamically modeled in the face of different candidate sets.Through comparative experiments with state-of-the-art algorithms on three real public datasets,Last.FM,Book-Cross and MovieLens-1 M,the results show that CKGAN significantly improves both CTR pre-diction and top-K recommendation.

关键词

知识图谱/随机采样/感受野/图注意力网络/异构网络/隐式反馈/推荐系统

Key words

knowledge graph/random sampling/receptive field/graph attention network/heterogeneous network/implicit feed-back/recommendation system

引用本文复制引用

基金项目

陕西省自然科学基础研究计划陕煤联合基金(2019JLZ-08)

陕西省自然科学基础研究计划(2020JM-522)

陕西省自然科学基础研究计划(2021JM-396)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
参考文献量19
段落导航相关论文