arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.05641cs.IRcs.AI
收藏

SR-HetGNN:基于异构图神经网络的会话推荐

SR-HetGNN:Session-based Recommendation with Heterogeneous Graph Neural Network

Beijing University of Posts and Telecommunications(北京邮电大学)

Jinpeng Chen, Haiyang Li, Xudong Zhang, Fan Zhang, Senzhang Wang, Kaimin Wei, Jiaqi Ji

AI总结:

针对现有会话推荐忽略用户画像等信息的问题,提出基于异构图神经网络的SR-HetGNN方法,通过构建异构图、融合长短期偏好学习会话嵌入,在两个真实数据集上性能优于现有最优方法。

摘要:

基于会话的推荐系统旨在根据用户此前的会话序列预测其下一次点击行为。现有研究通常依据用户会话序列中的物品转换关系学习用户偏好,但会话序列中的其他有效信息(如用户画像)在很大程度上被忽略,可能导致模型无法学习到用户的特定偏好。本文提出SR-HetGNN,这是一种新颖的会话推荐方法,采用异构图神经网络(HetGNN)学习会话嵌入并捕捉匿名用户的特定偏好。具体而言,SR-HetGNN首先根据会话序列构建包含多种类型节点的异构图,可捕捉物品、用户与会话之间的依赖关系;其次,HetGNN捕捉物品间的复杂转换关系,学习包含用户信息的物品嵌入;最后,结合注意力网络融合局部与全局会话嵌入,同时考虑用户长短期偏好的影响,得到最终的会话嵌入。在Diginetica和Tmall两个真实大规模数据集上开展的大量实验表明,SR-HetGNN的性能优于现有最先进的基于会话的推荐方法。

英文摘要:

The Session-Based Recommendation System aims to predict the user's next click based on their previous session sequence. The current studies generally learn user preferences according to the transitions of items in the user's session sequence. However, other effective information in the session sequence, such as user profiles, are largely ignored which may lead to the model unable to learn the user's specific preferences. In this paper, we propose SR-HetGNN, a novel session recommendation method that uses a heterogeneous graph neural network (HetGNN) to learn session embeddings and capture the specific preferences of anonymous users. Specifically, SR-HetGNN first constructs heterogeneous graphs containing various types of nodes according to the session sequence, which can capture the dependencies among items, users, and sessions. Second, HetGNN captures the complex transitions between items and learns the item embeddings containing user information. Finally, local and global session embeddings are combined with the attentional network to obtain the final session embedding, considering the influence of users' long and short-term preferences. SR-HetGNN is shown to be superior to the existing state-of-the-art session-based recommendation methods through extensive experiments over two real large datasets Diginetica and Tmall.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑