链载Ai

标题: 推荐系统-重排序-阿里SRGA [打印本页]

作者: 链载Ai    时间: 1 小时前
标题: 推荐系统-重排序-阿里SRGA

目录


推荐阅读⭐️⭐️⭐️⭐️⭐️

《Scope-aware Re-ranking with Gated Attention in Feed》

这篇文章是蚂蚁公开的一种信息流商品重排序的方法。建模时考虑了用户在信息流商品推荐场景下的浏览行为,总结出两种行为特征:

现有的重排序模型大都忽略了这两种行为特征,本文提出的建模思路:

一、模型整体结构

二、1.1 Input Layer

模型的第一部分输入为用户侧特征,经过一个前馈神经网络层得到用户侧特征向量:。

模型的第二部分输入为长度为k的商品列表,每个商品经过前馈神经网络层并与对应的位置特征向量融合后得到商品向量:,其中为第i个商品对应的特征。

三、1.2 PI Layer

这里主要计算用户u对商品i的个性化偏好表示(用户向量和商品向量两两拼接后经过两个前馈神经网络层得到相应的向量表示):

四、1.3 Gated Attention Layer

这一部分是文章的重点,主要包含前文提到的以下三点:

其中GSA和LSA采用的是self-attention的结构,只是在计算注意力权重时分别做对应的特殊处理,self-attention结构常见的公式表示:

GSA计算逻辑:

LSA计算逻辑:

Attention(Q,K,V)的计算逻辑用图来表示:

GSA和LSA的基本逻辑可以认为是在计算softmax之前对注意力权重矩阵中的相应位置做对应的特殊处理。

Gate Module的主要作用就是融合GSA和LSA的输出:

五、Output Layer

最终的输出为用户u对商品i的偏好概率:

六、损失函数

模型训练时的损失函数为:








欢迎光临 链载Ai (https://www.lianzai.com/) Powered by Discuz! X3.5