Journal of Computer Applications ›› 2022, Vol. 42 ›› Issue (11): 3610-3616.DOI: 10.11772/j.issn.1001-9081.2021091696

• Artificial intelligence • Previous Articles    

Session recommendation method based on graph model and attention model

Weichao DANG, Zhiyu YAO(), Shangwang BAI, Gaimei GAO, Chunxia LIU   

  1. College of Computer Science and Technology,Taiyuan University of Science and Technology,Taiyuan Shanxi 030024,China
  • Received:2021-09-26 Revised:2022-03-07 Accepted:2022-03-21 Online:2022-11-14 Published:2022-11-10
  • Contact: Zhiyu YAO
  • About author:DANG Weichao, born in 1974, Ph. D., associate professor. His research interests include intelligent computing, software reliability.
    YAO Zhiyu, born in 1995, M. S. candidate. His research interests include session recommendation.
    BAI Shangwang, born in 1964, M. S., professor. His research interests include digital intelligent software system.
    GAO Gaimei, born in 1978, Ph. D., lecturer. Her research interests include network security, cryptography.
    LIU Chunxia, born in 1977, M. S., associate professor. Her research interests include software engineering, database.
  • Supported by:
    Natural Science Foundation of Shanxi Province(201901D111266)

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

党伟超, 姚志宇(), 白尚旺, 高改梅, 刘春霞   

  1. 太原科技大学 计算机科学与技术学院,太原 030024
  • 通讯作者: 姚志宇
  • 作者简介:党伟超(1974—),男,山西运城人,副教授,博士,CCF会员,主要研究方向:智能计算、软件可靠性
    姚志宇(1995—),男,山西朔州人,硕士研究生,主要研究方向:会话推荐 S20190660@stu.tyust.edu.cn
    白尚旺(1964—),男,山西吕梁人,教授,硕士,主要研究方向:智能软件系统
    高改梅(1978—),女,山西吕梁人,副教授,博士,CCF会员,主要研究方向:网络安全、密码学
    刘春霞(1977—),女,山西大同人,副教授,硕士,CCF会员,主要研究方向:软件工程、数据库。
  • 基金资助:
    山西省自然科学基金资助项目(201901D111266)

Abstract:

To solve the problem that representation of interest preferences based on the Recurrent Neural Network (RNN) is incomplete and inaccurate in session recommendation, a Session Recommendation method based on Graph Model and Attention Model (SR?GM?AM) was proposed. Firstly, the graph model used global graph and session graph to obtain neighborhood information and session information respectively, and used Graph Neural Network (GNN) to extract item graph features, which were passed through the global item representation layer and session item representation layer to obtain the global? level embedding and the session?level embedding, and the two levels of embedding were combined into graph embedding. Then, attention model used soft attention to fuse graph embedding and reverse position embedding, target attention activated the relevance of the target items, as well as attention model generated session embedding through linear transformation. Finally, SR?GM?AM outputted the recommended list of the N items for the next click through the prediction layer. Comparative experiments of SR?GM?AM and Lossless Edge?order preserving aggregation and Shortcut graph attention for Session?based Recommendation (LESSR) were conducted on two real public e?commerce datasets Yoochoose and Diginetica, and the results showed that SR?GM?AM had the highest P@20 of 72.41% and MRR@20 of 35.34%, verifying the effectiveness of it.

Key words: session recommendation, global graph, session graph, Graph Neural Network (GNN), neighborhood information

摘要:

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

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

CLC Number: