• •    

基于图模型和注意力模型的会话推荐

党伟超1,姚志宇2,白尚旺1,高改梅2,刘春霞3   

  1. 1. 太原科技大学计算机科学与技术学院
    2. 太原科技大学
    3. 太原科技大学 计算机科学与技术学院,太原 030024
  • 收稿日期:2021-09-29 修回日期:2022-03-07 发布日期:2022-04-15
  • 通讯作者: 姚志宇
  • 基金资助:
    山西省自然科学基金

Session recommendation based on graph model and attention model

  • Received:2021-09-29 Revised:2022-03-07 Online:2022-04-15
  • Supported by:
    Natural Science Foundation of Shanxi Province

摘要: 为解决基于循环神经网络会话推荐方法的兴趣偏好表示不全面以及不准确问题,提出基于图模型和注意力模型的会话推荐方法(SR-GM-AM)。首先,利用全局图模型和会话图模型分别获取邻域信息和会话信息,经过图神经网络,提取项目图特征,全局项目表示和会话项目表示分别获取两个级别的表示,结合生成图嵌入;然后,使用软注意力进行图嵌入和反向位置嵌入融合,目标注意力激活目标项目相关性,通过线性转换生成会话嵌入;最后,经过预测层,输出下次点击的N项推荐列表。在公共数据集上进行对比实验,实验结果表明,P@20最高达到了72.41%,MRR@20最高达到了35.34%,相较最优的基于无损边缘保留聚合和快捷图注意力的推荐方法(LESSR),分别提升了0.5个百分点和2个百分点,验证了该方法的有效性。

关键词: 会话推荐, 全局图, 会话图, 图神经网络, 邻域信息

Abstract: To solve the problem that representation of interest preferences based on the recurrent neural network is incomplete and inaccurate in session recommendation, a session recommendation based on graph model and attention model(SR-GM-AM) method was proposed. Firstly, global graph model and session graph model were used to obtain neighborhood information and session information respectively, the item graph features were extracted through the graph neural network, global item representation and session item representation obtain two-level representations, which were combined with the graph embedding. Then, soft attention was used to fuse graph embedding and reverse position embedding, target attention activated the relevance of the target items, session embedding was generated through linear transformation. Finally, the recommended list of the N items for the next click was outputted through the prediction layer. The comparison experiments were carried out in the public data sets. Experimental results show that, P@20 reaches 72.41% and MRR@20 reaches 35.34%, compared with the optimal Lossless Edge-order preserving aggregation and Shortcut graph attention for Session-based Recommendation(LESSR) method, it is improved by 0.5 percentage point and 2 percentage point respectively, verifying the effectiveness of the proposed method.

Key words: session recommendation, global graph, session graph, graph neural network, neighborhood information

中图分类号: