
摘要
基于会话的推荐问题旨在根据匿名会话预测用户行为。以往的方法将一个会话建模为序列,并估计用户表示和项目表示以进行推荐。尽管这些方法取得了令人鼓舞的结果,但它们在获取会话中的准确用户向量方面仍显不足,并且忽略了项目的复杂转换关系。为了获得准确的项目嵌入并考虑项目的复杂转换关系,我们提出了一种新的方法,即基于图神经网络的会话推荐(Session-based Recommendation with Graph Neural Networks,简称SR-GNN)。在该方法中,会话序列被建模为图结构数据。基于会话图,图神经网络可以捕捉到项目之间的复杂转换关系,而这些关系是以往传统的顺序方法难以揭示的。然后,通过注意力网络将每个会话表示为全局偏好和当前兴趣的组合。在两个真实数据集上进行的大量实验表明,SR-GNN 显著且一致地优于现有的最先进的基于会话的推荐方法。
代码仓库
hkust-knowcomp/sessioncqa
pytorch
GitHub 中提及
rithinch/session-based-vehicle-recommendations
pytorch
GitHub 中提及
herrbilbo/hse-recsys-project
pytorch
GitHub 中提及
DiMarzioRock7/SR-GNN
pytorch
GitHub 中提及
CRIPAC-DIG/SR-GNN
官方
tf
GitHub 中提及
xiaominglalala/Session_based_Recommendation
paddle
GitHub 中提及
userbehavioranalysis/SR-GNN_PyTorch-Geometric
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| session-based-recommendations-on-diginetica | SR-GNN | Hit@20: 50.73 MRR@20: 17.59 |
| session-based-recommendations-on-gowalla | SR-GNN | HR@20: 50.32 MRR@20: 24.25 |
| session-based-recommendations-on-last-fm | SR-GNN | HR@20: 22.33 MRR@20: 8.23 |
| session-based-recommendations-on-yoochoose1 | SR-GNN | MRR@20: 30.94 Precision@20: 70.57 |
| session-based-recommendations-on-yoochoose1-1 | SR-GNN | HR@20: 70.57 MRR@20: 30.94 |
| session-based-recommendations-on-yoochoose1-4 | SR-GNN | HR@20: 71.36 MRR@20: 31.89 |