论文《Memory Augmented Graph Neural Networks for Sequential Recommendation》阅读

论文概况

本文是2020年AAAI上的一篇论文，该篇文章聚焦于短期兴趣、长期兴趣、项目共现来解决会话推荐问题，提出了MAGNN（Memory Augmented Graph Neural Networks ）模型。

Introduction

作者提出了几个问题

• 现有方法往往只关注长期兴趣或者短期兴趣，而没有综合的将二者一起考虑
• 其他方法没有显式捕获项目序列中的项目共现模式（相似物品相近）

对于上述问题，作者提出了MAGNN模型：（1）捕获项目的短期上下文信息和长期依赖关系（2）在GNN框架中加入了一个门控机制，有效地融合短期和长期利益（3）使用双线性函数来捕获项目之间的特征相关性。

Method

A.Short-term Interest Modeling

将完整的用户会话分为l份，每个子会话包含|L|个项目，取最后一个子会话计算当前兴趣
会话图的构建：每个项目与之后3个连续的项目连接，一个项目的边的权重均分且相加等于1
然后根据邻接矩阵进行聚合更新，并于自身向量级联，此处存在两轮GNN。MAGNN中短期兴趣表示为第L个子会话中所有项目的向量均值。
$${\mathbf{h}}_i=\tanh \left({\mathbf{W}}^{(1)}\cdot \left[\sum _{k\in {\mathcal{N}}_i}{\mathbf{e}}_k{A}_{i,k};{\mathbf{e}}_i\right]\right),\forall i\in L_{u,l}\text{\hspace{1em}(1)}$$

B.Long-term Interest Modeling

之后利用前l-1个子会话来计算用户的长期兴趣（前面的子会话设为${\mathbf{H}}_{u,l}$）
首先使用PE函数将项目的位置信息考虑在内
$${\mathbf{H}}_{u,l}:={\mathbf{H}}_{u,l}+\mathrm{PE}\left(H_{u,l}\right)\text{\hspace{1em}(2)}$$
用用户嵌入来对${\mathbf{H}}_{u,l}$进行注意力计算
$${\mathbf{S}}_{u,l}=\mathrm{softmax}\left({\mathbf{W}}_a^{(3)}\tanh \left({\mathbf{W}}_a^{(1)}{\mathbf{H}}_{u,l}+\left({\mathbf{W}}_a^{(2)}{\mathbf{p}}_u\right)\otimes 1_{\varphi }\right)\right)\text{\hspace{1em}(3)}$$
${\mathbf{Z}}_{u,l}$的每一行都代表过去会话的一个方面
$${\mathbf{Z}}_{u,l}=\tanh \left({\mathbf{S}}_{u,l}\cdot {\mathbf{H}}_{u,l}^{\top }\right)\text{\hspace{1em}(4)}$$
此时${\mathbf{z}}_{u,l}$是对不同方面求平均值的组合查询嵌入
$${\mathbf{z}}_{u,l}=\mathrm{avg}\left({\mathbf{Z}}_{u,l}\right)\text{\hspace{1em}(5)}$$
然后我们利用${\mathbf{z}}_{u,l}$作为查询向量通过记忆网络来查找内存网络中共享用户潜在兴趣,将用户潜在兴趣与长期会话表征相加即可得到长期兴趣。
$$\begin{array}{l}{s}_i=\mathrm{softmax}\left({\mathbf{z}}_{u,l}^{\top }\cdot {\mathbf{k}}_i\right),\\ {\mathbf{o}}_{u,l}={\sum }_i{s}_i{\mathbf{v}}_i,\\ {\mathbf{p}}_{u,l}^H={\mathbf{z}}_{u,l}+{\mathbf{o}}_{u,l},\end{array}\text{\hspace{1em}(6)}$$

C.Interest Fusion

使用门控机制将长期兴趣与短期兴趣结合。
$${\mathbf{g}}_{u,l}=\sigma \left({\mathbf{W}}_g^{(1)}\cdot \frac{1}{|L|}\sum _{i\in L_{u,l}}{\mathbf{h}}_i+{\mathbf{W}}_g^{(2)}\cdot {\mathbf{p}}_{u,l}^H+{\mathbf{W}}_g^{(3)}\cdot \mathbf{p}u\right)\text{\hspace{1em}(7)}$$
$${\mathbf{P}}_{u,l}^C={\mathbf{g}}_{u,l}\odot \frac{1}{|L|}\sum _{i\in L_{u,l}}{\mathbf{h}}_i+\left(1_d-{\mathbf{g}}_{u,l}\right)\odot {\mathbf{p}}_{u,l}^H\text{\hspace{1em}(8)}$$

D.Prediction

预测层不只有用户的综合兴趣，还考量了用户嵌入，同时还适用双线性函数来捕获最后一个子会话与其他子会话之间的物品关系
$${\hat{r}}_{u,j}={\mathbf{p}}_u^{\top }\cdot {\mathbf{q}}_j+{\mathbf{p}}_{u,l}^{C\top }\cdot {\mathbf{q}}_j+\frac{1}{|L|}\sum _{i\in L_{u,l}}{\mathbf{e}}_i^{\top }{\mathbf{W}}_r{\mathbf{q}}_j\text{\hspace{1em}(9)}$$

总结

MAGNN将连续的会话切分为多个子会话，通过最后一个子会话计算短期兴趣，通过其余会话使用注意力机制以及记忆网络计算长期兴趣，在计算的过程中还将用户嵌入考虑在内，使用门控单元来融合长短期兴趣，最后也是综合兴趣、用户嵌入、最近的子会话结合起来进行预测。