知识增强的图神经网络序列推荐模型  

Knowledge-Enhanced Graph Neural Network Model for Sequential Recommendation

作　　者：李盼 解庆 李琳[1] 刘永坚[1,2] LI Pan;XIE Qing;LI Lin;LIU Yongjian(School of Computer Science and Artificial Intelligence,Wuhan University of Technology,Wuhan 430070,China;Chongqing Research Institute,Wuhan University of Technology,Chongqing 401135,China)

机构地区：[1]武汉理工大学计算机与人工智能学院,武汉430070 [2]武汉理工大学重庆研究院,重庆401135

出　　处：《计算机工程》2023年第2期70-80,共11页Computer Engineering

基　　金：重庆市自然科学基金(cstc2021jcyj-msxmX1013);湖北省重点研发计划项目(2021BAA030)。

摘　　要：现有基于图神经网络的序列推荐模型大多仅关注用户与项目交互的结构性信息,序列偏好的学习仅涉及项目交互顺序,缺乏项目自身的内容信息,并且未有效利用用户信息及挖掘项目之间更深层的语义关系。提出一种知识增强的图神经网络序列推荐模型KGGNN,引入知识图谱,并结合用户交互数据构建协同知识图谱,学习得到项目语义关联辅助信息以及用户关联辅助信息。将交互序列构建成有向序列图,利用门控图神经网络以及用户关联辅助信息学习序列中项目节点的结构性信息。通过注意力机制组合项目向量作为全局序列偏好,将最近交互的项目作为当前兴趣偏好,融合两者形成最终序列偏好,并结合项目语义关联辅助信息进行模型预测。在Amazon-Book、Last-FM、Yelp2018这3个公开数据集上的实验结果表明,辅助信息能有效提升序列推荐的准确性,该模型在命中率(HIT@K)和归一化折损累计增益(NDCG@K)2个指标上相较于GRU4Rec、NARM、SASRec等模型均有显著提升。当评估指标K值选取10时,与KGSR模型相比,其HIT@10指标在3个数据集上分别提升12.9%、4.5%、6.9%,NDCG@10指标在3个数据集上分别提升29.4%、5.7%、16.7%。Existing sequential recommendation models based on the Graph Neural Network(GNN) focus on the structural information of users’ interaction with items,and the learning of sequence preference involves the sequence of interactive items,which lack content information and user information,and does not mine a deep semantic relationship between items.A knowledge-enhanced GNN sequence recommendation model(KGGNN) is proposed.First,a knowledge graph is introduced and combined with users’ interaction data to construct a collaborative knowledge graph,which can obtain the relevant auxiliary information of items and users.By transforming an interaction sequence into a directed sequence graph,the algorithm uses the Gated Graph Neural Network(GGNN) and relevant auxiliary information of the users to learn the structural information of the item node in the sequence.The model combines the vectors of the items as the global sequence preference using the attention mechanism,and the recently interactive item is considered as the current interest preference.The two preferences are fused to form the final sequence preference,which is used with the semantic relevant auxiliary information of the current item to predict the next item.The experimental results of Amazon Book,Last FM,and Yelp 2018 indicate that the auxiliary information can effectively help improve the accuracy of the hit rate(HIT@K)and normalized discounted cumulative gain(NDCG@K)metrics.Compared with GRU4Rec,NARM,SASRec,and other models,the aforementioned indicators have significantly improved.When the K value of the evaluation index is 10,compared with the KGSR model,HIT@10 is improved by 12.9%,4.5%,and 6.9%,and NDCG@10 is improved by 29.4%,5.7%,and 16.7%.

关 键 词：序列推荐 图神经网络 知识图谱 推荐系统 用户行为 深度学习 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]