首页 | 官方网站   微博 | 高级检索  
     

融合知识图谱的双线性图注意力网络推荐算法
引用本文:潘承瑞,何灵敏,胥智杰,王修晖,宋承文.融合知识图谱的双线性图注意力网络推荐算法[J].计算机工程与应用,2021,57(1):29-37.
作者姓名:潘承瑞  何灵敏  胥智杰  王修晖  宋承文
作者单位:1.中国计量大学 信息工程学院,杭州 310000 2.中国计量大学 浙江省电磁波信息技术与计量检测重点实验室,杭州 310000
摘    要:知识图谱可有效缓解传统协同过滤中的数据稀疏和冷启动问题,因此,近年来在推荐系统中融入知识图谱的方法成为重要的探索方向。然而现有的方法大多将知识图谱的网络结构划分为单独路径或仅利用了一阶邻居信息,造成无法建立整个图上的高阶连通性问题。为解决该问题,提出融合知识图谱和图注意力网络的KG-BGAT模型,并设计了双线性采集器。双线性采集器能够在信息采集阶段获取节点间的特征交互信息,丰富节点表示;图注意力网络通过递归嵌入传播算法将各个节点表示沿图进行传播,能够捕获图中的高阶连通性。在MovieLens-1M数据集上进行了Top-K推荐实验,在推荐列表长度为20时,精确率、召回率和归一化折损累计增益分别为29.4%、24.9%、67.4%,超过了目前主流的CKE、RippleNet、KGCN等融合知识图谱的推荐算法。实验证明提出的方法能够有效提高推荐结果的准确性。

关 键 词:推荐系统  知识图谱  特征交互  图注意力网络  

Fusion Knowledge Graph and Bilinear Graph Attention Network Recommendation Algorithm
PAN Chengrui,HE Lingmin,XU Zhijie,WANG Xiuhui,SONG Chengwen.Fusion Knowledge Graph and Bilinear Graph Attention Network Recommendation Algorithm[J].Computer Engineering and Applications,2021,57(1):29-37.
Authors:PAN Chengrui  HE Lingmin  XU Zhijie  WANG Xiuhui  SONG Chengwen
Affiliation:1.College of Information Engineering, China JiLiang University, Hangzhou 310000, China 2.Key Laboratory of Electromagnetic Wave Information Technology and Metrology of Zhejiang Province, China JiLiang  University, Hangzhou 310000, China
Abstract:Knowledge graph can effectively alleviate problems of data sparse and cold start in traditional collaborative filtering.Therefore,it has become an important exploration direction to integrate knowledge graph into the recommender system.However,most of the methods divide the network structure of knowledge graph into separate paths or only use the first-order neighbor information,which make it impossible to establish the high-order connectivity on the whole graph.To solve the problem,this paper proposes a KG-BGAT model which combines knowledge graph and graph attention network,and designs a bilinear collector.The bilinear collector can obtain the feature interaction message between nodes during the stage of information collection,and enrich the representation of nodes.The graph attention network propagates each node representation along the graph through the recursive embedding propagation algorithm,which can capture the high order connectivity in the graph.Top-K recommendation experiments are tested on the Movielens-1M dataset.When the length of the recommendation list is 20,the accuracy rate,recall rate and normalized discounted cumulative gain are 29.4%,24.9%and 67.4%respectively,which exceeds the current mainstream recommendation algorithms such as CKE,RippleNet and KGCN.Experiments show that the proposed method can improve the accuracy of the recommended results.
Keywords:recommendation system  knowledge graph  feature interaction  graph attention network
本文献已被 维普 万方数据 等数据库收录!
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号