Attention 机制在深度学习推荐算法中的应用

阅读数:17 2019 年 11 月 29 日 11:40

Attention机制在深度学习推荐算法中的应用

1.AFM:Attentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networks
Attention机制在深度学习推荐算法中的应用
模型入上图所示,其中 sparse iput,embedding layer,pair-wise interaction layer 都和 FM 一样,后面加入了一个 attention net 生成一个关于特征交叉项的权重,将 FM 原来的二次项累加变成加权累加。这里的 attention net 其实很简单,就是简单的线性变换求出权重之后再和原来的输入进行加权求和。
Attention机制在深度学习推荐算法中的应用

然后加权计算得到 AFM 的输出:
Attention机制在深度学习推荐算法中的应用

缺点:AFM 只是在 FM 的基础上添加了 attention 的机制,但实际上只利用了加权的二次项,并没有加入更深的网络去学习更高阶的非线性交叉特征,所以它的上限和 FFM 很接近,没有完全发挥出 dnn 的优势。

  1. 阿里 DIN:Deep Interest Network for Click-Through Rate Prediction
    Attention机制在深度学习推荐算法中的应用

用户的历史行为数据是有多个的,在计算新的候选商品的 ctr 时,可以将用户多个历史行为的 embedding 加个 average pooling 平均一下,但是这样没有办法建模出不同历史行为对目标预测的影响程度。DIN 中加入了一种 Attention 机制来使得模型对用户相关的行为历史看重一些,对不相关的历史甚至可以忽略的效果。
Attention机制在深度学习推荐算法中的应用

上式中,[公式] 是用户的 embedding 向量,[公式] 是候选广告商品的 embedding 向量,[公式] 是用户 u 的第 i 次行为的 embedding 向量,因为这里用户的行为就是浏览商品或店铺,所以行为的 embedding 的向量就是那次浏览的商品或店铺的 embedding 向量。[公式] 从过去 [公式] 的加和变成了 [公式] 的加权和,[公式] 的权重 [公式] 就由 [公式] 与 [公式] 的关系决定,也就是上式中的 [公式]

那么怎么计算 g 呢,传统的做法可以直接将用户 embedding u 和候选商品 embedding v 做点积或者 uWv。DIN 的做法是将 u、v 和两者之间的差值这三个向量合并起来作为输入,输入到 mlp 中得到权重。

  1. 阿里 DIEN:Deep Interest Evolution Network

Attention机制在深度学习推荐算法中的应用
DIEN 先用 GRU 从用户的行为序列中去抽取兴趣,然后通过带 attention 的 GRU 单元来寻找与目标 item 相关的兴趣点。其中 attention 计算方式为:
Attention机制在深度学习推荐算法中的应用

其中 [公式] 表示一个目标 item 中不同 field 特征的 embedding 向量的 concat,可以看到这里的 attention 是一种很传统的 uWv 的方式,然后将求得的 attention 权重结合 GRU 一起使用。

  1. 阿里 DSIN:Deep Session Interest Network for Click-Through Rate Prediction
    Attention机制在深度学习推荐算法中的应用

有三处用到了 attention 的结构,首先在 Session Interest Extractor Layer 中,使用了 transformer 来处理每个 session,得到一个 d 维的向量。多个 session 的 d 维向量经过双向 LSTM 得到隐层状态 ht。

另外两处用到 attention 的地方在 Session Interest Activating Layer 中,使用了和 DIEN 中一样的 attention 结构来刻画用户的会话兴趣与目标物品的相关性、LSTM 得到隐层状态 ht 与目标物品的相关性。
Attention机制在深度学习推荐算法中的应用

最后把四部分向量:用户特征向量、待推荐物品向量、会话兴趣加权向量 UI、带上下文信息的会话兴趣加权向量 UH 进行 concat,输入到全连接层中,得到输出。

  1. 阿里:Behavior Sequence Transformer for E-commerce Recommendation in Alibaba

Attention机制在深度学习推荐算法中的应用
将用户的行为序列先通过 embedding 层处理,然后经过了一个 Transformer 层。

  1. 阿里:Deep Spatio-Temporal Neural Networks for Click-Through Rate Prediction

Attention机制在深度学习推荐算法中的应用
输入包括目标广告的信息、上下文广告信息、点击广告信息、曝光未点击广告信息,经过 embedding 层,然后有两种 attention 方式,第一种是 self-attention 方式:

Attention机制在深度学习推荐算法中的应用
其实就是比如上下文的 embedding 表示 [公式] 输入到一个函数 f 中,得到标量输出 [公式] ,然后通过 softmax 将每个上下文 item 的权重归一化到 0-1 之间,随后基于权重进行加权求和。缺点就是没有考虑 target ad 的信息。

另外一种 attention 方式是交互式的,并且加入了目标广告的信息 [公式] :

Attention机制在深度学习推荐算法中的应用
7. ATRank: An Attention-Based User Behavior Modeling Framework for Recommendation

Attention机制在深度学习推荐算法中的应用
分为原始特征空间 (raw feature spaces)、行为嵌入空间 (behavior embedding spaces)、隐语义空间 (latent semantic spaces)、行为交互层 (behavior interaction layers)、下游网络层 (downstream application network)

模型中用到 attention 的地方在 latent semantic spaces 层后面,latent semantic spaces 层的目的是将每个行为组中的向量通过线性变换到多个不同的语义空间,然后采用 self-attention 结构:

Attention机制在深度学习推荐算法中的应用
其中 S 是整个语义层拼接后的输出,Sk 是第 k 个语义空间上的投影。

Attention机制在深度学习推荐算法中的应用
经过 self-attention 后,第 K 个语义空间的向量表示为:

Attention机制在深度学习推荐算法中的应用
[公式] 是将 S 映射到第 k 个语义空间的另一个转换函数,文中用的一层的 MLP。

总结:可以看出深度学习推荐模型中,使用的 attention 可以分为两大类,一类是 self-attention(比如 ATRank),一类是传统的 soft attention(主要都是 uWv 形式,比如 DIEN,DSIN 等),u 表示用户行为历史中一个 item 的 embedding,v 是待推荐的目标 item。求出乘积,通过 softmax 函数就可以得到用户历史行为中每个 item 的权重。另外 DIN 的做法比较特殊,是将 u、v 和两者之间的差值这三个向量合并起来作为输入,输入到 mlp 中得到用户历史行为中每个 item 的权重。

参考文献:

https://arxiv.org/pdf/1708.04617.pdf
https://arxiv.org/pdf/1706.06978.pdf
https://zhuanlan.zhihu.com/p/51623339
https://arxiv.org/pdf/1809.03672.pdf
https://arxiv.org/abs/1905.06482
https://arxiv.org/pdf/1905.06874.pdf
https://arxiv.org/abs/1906.03776
https://arxiv.org/abs/1711.0663

本文转载自 Alex-zhai 知乎账号。

原文链接: https://zhuanlan.zhihu.com/p/78015378

评论

发布