Attentive sequential model based on graph neuralnetwork for next poi recommendation
基于图神经网络的注意力序列模型用于下一个兴趣点推荐
论文背景
Attentive sequential model based on graph neuralnetwork for next poi recommendation
基于图神经网络的注意力序列模型用于下一个兴趣点推荐
WWW21
PDF
关键词:推荐系统、序列推荐、兴趣点推荐、图神经网络、注意力机制
现有问题
传统推荐方法忽略了用户短时偏好的动态变化。另外,许多现有方法不能完全探索兴趣点签到序列中复杂的联系和转变形式。
架构
提出ASGNN。
ASGNN包括四部分:兴趣点签到序列图构建、特征表示学习、长短时偏好获取、兴趣点推荐
兴趣点签到序列图构建
G(V, E), V = (U, L),U是用户集,L是兴趣点集。E包括用户-兴趣点边和兴趣点-兴趣点边。
图中边的权重代表用户在兴趣点的签到次数。
特征表示学习
图构建好后,使用GNN学习到用户和兴趣点的低维表示。这避免了马尔科夫决策过程需要的大量状态。
为了提高效率更新节点,使用了GGNN。
长短时偏好获取
设计了两层注意力机制分别捕获长短时用户偏好。
兴趣点推荐
上一步得到的个性化用户偏好参数和兴趣点特征点乘,得到每个兴趣点分数,通过softmax标准化输出概率值。
训练的损失函数为交叉熵函数。
实验
围绕下列问题展开:
- ASGNN在序列兴趣点推荐任务上性能如何(基线对比)
- ASGNN的关键组件效果如何(组件实验)
- ASGNN的嵌入维度对推荐的影响(维度分析)
- ASGNN和基线在不同稀疏性的数据集上的性能如何(数据稀疏性影响)
- ASGNN学习兴趣点嵌入是否有效(可视化说明)
数据集
Gowalla, FourSquare, Brightkite
https://snap.stanford.edu/data/loc-gowalla.html
https://sites.google.com/site/yangdingqi/home/foursquare-dataset
https://snap.stanford.edu/data/loc-brightkite.html
基线
- POP
- BPR
- FPMC
- HRM
- CPAM
- SHAN
- SRGNN
评测指标
- 召回率Recall
- MRR
基线对比
组件实验
维度分析
数据稀疏性影响
可视化说明
贡献点
- 提出ASGNN,它将用户签到行为视为图,并使用GNN局部方式学习用户行为模式和他们的偏好用于下一个兴趣点推荐。
- 设计了一个个性化层级注意力机制捕捉用户长短时偏好,并将它们适应于序列推荐。
- 实验结果显示ASGNN超过基线和部分SOTA模型。