AGREE：基于注意力机制的群组推荐（附代码）

本文内容的目录结构如下：

• 1、聊一聊群组推荐

• 2、模型介绍

• 2.1、符号表示含义和要解决的问题

• 2.2、基于注意力的群组表征学习

• 2.3、基于NCF的交互学习

• 2.4、模型优化

• 3、实验效果

• 4、代码解析

这篇文章主要分享的论文是2018年被CCF收录的一篇论文：Attentive Group Recommendation（基于注意力机制的群组推荐），第一作者是湖南大学的曹达老师，二作是论文Neural Collaborative Filtering的作者何老师。当然也会结合小编的工作来进行一些补充说明，写的不好，欢迎拍砖！

这篇文章结合了注意力机制和NCF框架，使得在群组推荐中有良好的效果。

关于群组推荐的概述可以参考： 论文｜组推荐系统及其应用研究

关于Neural Collaborative Filtering可以参考： 论文｜被“玩烂”了的协同过滤加上神经网络怎么搞

关于注意力的解释可以参考： 深度学习中的注意力机制

1、聊一聊群组推荐

群组推荐在工业推荐系统中应用十分广泛，其主要作用在新老用户上。

对于新用户来讲， 则是冷启动，常规的冷启动解决办法是，根据新用户的注册信息（性别、地域等，这个一般和产品设计进行结合）找到其所属的群组，将群组下的Top信息返回推荐给当前来访的新用户。

对于老用户来讲，可以作为一路召回使用。比如对所有用户进行群组划分，再统计群组下的Top内容作为群组下用户的一路召回内容。当然也可以按照某种属性进行群组划分。

所以群组推荐的实际应用中是很广泛和有效的，上面举的例子只是冰山一角，实际中要比这个复杂和广泛。但往往我们在得到群组下Top内容的时候采用的方法简单粗暴，这一点论文中也提到了群组推荐中的一个基本问题是如何汇总小组成员的偏好以推断小组的决定，因为群组中不同的成员扮演的角色是不一样的， 常见的计算方法有：

• 均值策略，即认为群组中每个人贡献是等价的，计算群组内top内容时，取对某内容贡献的均值。

• 最小化策略，即取群组内对某内容的最低贡献作为该内容的 排序 值

• 最大化策略，和最小化策略相反，即取群组内对某内容的最高贡献作为该内容的排序值

• 专家策略，根据用户在项目上的专业知识为其赋予了个性化的权重

但是，这些预定义的策略和数据无关，因为这些策略很难灵活的捕获群组成员的动态权重。因此提出了self-Attention Mechanism + NCF的思路，即Attentive Group Recommendation（AGREE）。

AGREE的主要贡献点有：

• 第一次将神经网络中的注意力机制用于群组推荐，基于输入数据动态地学习融合策略。

• 引入用户和内容之间的交互行为，提升群组推荐的效果，同时在一定程度上能够缓解冷启动问题（首先根据某种策略得到群组，新用户来访时匹配其对应的群组，根据群组内其他成员选取的top内容作为新用户的冷启动内容）。

• 基于数据集（爬取的和公开的）进行了大量的实验，并进行公开，促进学术研究。

2、模型介绍

AGREE模型包含两部：

• （1）：基于小组成员聚集和偏好表示的群组表示学习

• （2）：通过NCF进行交互学习，为用户和群组推荐项目

2.1、符号表示含义和要解决的问题

符号表示的规则

• 粗体大写字母（eg：）表示矩阵

• 粗体小写字母（eg：）表示向量

• 粗体小写字母（eg：）表示标量

• 花体字母（eg：）表示集合

如果未特别说明，所有的向量都是列向量。

下图说明了群组推荐任务中输入数据形态：

论文中给出的群组推荐问题定义的是：对于给定的群组，输出为该群组推荐的 Top-K物品。

整个模型训练的

• 输入是：用户集合，群组集合，物品集合，群组和物品的交互矩阵，用户和物品的交互矩阵

• 输出是：两个个性化的排序函数，（将一个物品映射给一个群组的分值）、（将一个物品映射给一个用户的分值）

2.2、基于注意力的群组表征学习

论文建议基于表征学习解决群组推荐问题，在该框架下，每个实体都被表示为一个向量，该向量对实体的固有属性（eg：单词的语义，用户的兴趣等）进行编码，且可以从数据中学习。物品推荐中一个众所周知的矩阵分解方法就是一个表征学习模型，该模型将用户和物品通过嵌入向量联系起来。

表示用户向量，表示物品向量，我们的目标是通过群组成员的兴趣偏好为每个群组学习到一个向量，要从数据中动态学习聚合策略，可以将群组向量定义成依赖用户和物品的向量，抽象表示为：

表示群组中的用户。表示群组对物品的偏好向量。在AGREE模型中，包括两部分：用户向量聚合和群组偏好向量。形式化表示为：

上面公式中加号左侧为用户向量，加号右侧为群组偏好向量。

用户向量聚合

对群组中的成员进行加权求和，系数是一个可学习的参数，表示用户选择物品的影响力，很显然，如果一个用户在一个物品上的影响力比较大，则其对应的也会比较大。例如在决定一个群组该去哪里旅行的决定中，如果一个用户去过中国很多次，那么这个用户对是否去中国旅行的决定权就会更大一些，这也比较符合人的期望。

在表征学习框架下，表示用户的历史偏好，表示物品的属性，我们将设置为注意力神经网络，输入为和：

其中：

• 表示群组中用户的下标

• 和表示物品和用户对应的权重矩阵

• 表示偏置

• 表示激活函数

• 表示将结果映射为的权重向量

• 表示将结果归一化的概率函数

下图说明了用户向量聚合部分的模型设计。

群组偏好向量

除了对用户向量进行聚合，还对群组偏好向量进行了表示，目的是考虑群组的整体偏好。这里需要考虑的是当用户汇聚成一个小组时，他们追求的目标可能会与每个小组成员的偏好不同。例如：一个三口之家去看电影，小朋友喜欢看动画片，父母喜欢看浪漫电影，那么他们最终可能会考虑看一部教育电影。因此用一个通用的向量表示群组的普遍偏好是很必要的。这里直接使用群组成员的向量求和作为群组的向量表示，在后续的实验中也证明这是有效的。

2.3、基于NCF的交互学习

NCF是一个关于物品推荐的多层神经网络框架，可以参见之前的文章： Neural Collaborative Filtering 。其主要思路是将用户和物品的向量，输入到一个定制的神经网络中，基于历史数据学校用户和物品之间的交互，由于神经网络具有很强的表达能力，NCF比传统的矩阵分解模型效果要好很多。因此我们选择了NCF框架对用户、物品的表达向量和用户-物品之间的交互进行端到端的学习。

上图展示了论文使用的NCF解决方案，因为需要同时为群组和用户进行推荐，所以在输入的时候既有用户-物品对，还有群组项对，经过 Representation Layer 返回每个实体的表征向量（具体的生成方法参考：2.2、参考基于注意力的群组表征学习），然后输入到Pooling Layer（池化层）和Hidden Layer （隐藏层），（hidden layer被两个任务进行共享），最后获得预测分数。

pooling layer

假设输入的是群组-项目对，pooling layer（池化层）首先对输入的向量进行进行逐元素乘积，然后再与进行连接，表示如下：

之所以用这种方式表示，是因为内积能够有效的捕获到群组和物品之间的交互，同时连接群组和物品本身的信息能够避免信息丢失。同样用户-物品项之间的池化也采用这种方式。

hidden layer

池化层之上这是一个多层的隐藏层结构，能够捕获群组、用户、物品之间的非线性、高维的交互。

其中：

• 表示第层的权重矩阵

• 表示第层的偏置

• 表示第层的输出

• 表示激活函数

最后一层输出可以经过下面的公式转化为预测分：

其中

• 表示预测层的权重

• 表示 用户对物品的预测分

• 表示 群组对物品的预测分

这里需要强调的是，我们针对用户-物品和群组-物品两个预测任务共享了隐藏层，因为群组的embedding向量是通过用户的embedding向量汇聚而成的，他们在同一个向量空间中，且有利于两个任务的相互补充。

2.4、模型优化

目标函数

群组推荐本质也是做排序，所以这里使用的pairwise方法（关于pairwise、pointwise、listwise的区别可以参考： 怎么理解基于机器学习“四大支柱”划分的学习排序方法 ）。

用户-物品：

其中：

• 表示训练集

• 表示用户对物品有行为，但对物品没有行为 （是负采样产生）

• 表示观察到的交互和为观察到的交互之间的差值分数

这里认为有行为交互分值为1，没有交互分值为0，所以，又因为用户-物品的损失函数中为平方和，所以最后形式转为了。

同样，群组-物品的损失函数为：

优化技巧

主要是在模型设计和训练上，有以下几点：

1、mini-batch。需要注意的首先需要将训练集进行shuffle，即混洗，保证数据之间是无序的。

2、pre-training。神经网络对于相关的初始化值是比较敏感的，因此在AGREE模型中，对网络的初始值进行了初始化，初始化的方法是去除了AGREE中的Attention机制，将得到的相关参数复用在AGREE中，而且在预训练模型中使用的是Adma优化方法，因为速度较快，而在AGREE模型中使用的则是SGD（关于SGD可参考： 梯度算法之批量梯度下降，随机梯度下降和小批量梯度下降 ）。

3、droupout。由于神经网络的拟合能力强，为了提升模型的泛化能力，这里采用了深度学习中比较常见的Droupout机制，不仅在第一层中去除了，而且隐藏层也进行了剔除。另外需要注意的是在训练的时候需要进行dropout，但是在预测的时候不需要dropout。

3、实验效果

实验采用的数据集有两个：

• 自己爬取和整理的马蜂窝数据

• CAMAa2011

这两个数据集都只有正样本，因此从缺失的数据中进行随机负采样构建负样本。而且适当的调整正负样本的比例能够增强实验的效果，所以论文中选择的负样本的采样率在4～6之间。

实验采用的方法是：留一法（即交叉验证法），为了评估算法的效果，使用了两种指标来进行评估：

• Hit Ratio（HR）

• Normalized Discounted Cumulative Gain (NDCG)

实验中超参数的设置：

• 负采样率：4

• 嵌入层使用Glorot初始化权重

• 隐藏层使用均值为0，标准差为0.1的高斯分布初始化权重

• 基于梯度优化的算法使用Adma优化，mini-batch设置为[128, 256, 512, 1024]，学习率设置为[0.001, 0.005, 0.01, 0.05, 0.1]

• 隐藏成第一层的维度和嵌入层维度相同，为32

• 激活函数：ReLU

• 每组实验参数进行5次实验，选取平均值

为了验证算法的效果，和以下几种方法进行了对比：

• NCF

• Popularity（根据物品的流行度进行推荐）

• COM（基于概率的方法生成群组推荐）

• GREE（AGREE去除注意力机制）

AGREE和GREE的实验效果对比结果如下：

同时为了证明从数据集中学习聚合策略的有效性，和以下实验做了对比：

• NCF+avg

• NCF+ the least misery strategy

• NCF + The maximum satisfaction strategy

• NCF + The expertise strategy

AGREE算法的有效性实验对比结果如下：

同时也拿AGREE和其变体（AGREE-U、AGREE-G）做了实验对比：

• AGREE-U：AGREE算法中只利用了用户的embedding向量

• AGREE-G：AGREE算法中只利用了群组的embedding向量

AGREE算法和其变体的实验效果对比结果如下：

同时也拿AGREE和冷启动的中一些算法进行了结果对比，这里冷启动的前提是用户已经加入了某群组，但是没有产生行为，实验结果如下：

综合来看，AGREE算法的效果是要比其他群组推荐算法好很多的。

4、代码解析

论文中也给出了代码的github 地址：https://github.com/LianHaiMiao/Attentive-Group-Recommendation

代码的环境是：

pytorch version: '0.3.0'
python version: '3.5'

由于目前pytorch已经更新到1.4版本了，所以对代码进行了更新，更新后的repo库为：https://github.com/Thinkgamer/Attentive-Group-Recommendation

运行代码，输出为：

num_group is: 290
num_users is: 602
num_items is: 7710
AGREE 的Embedding 维度为: 32, 迭代次数为: 30, NDCG、HR评估选择topK: 5
Iteration 0, loss is [1.0019 ]
Iteration 0, loss is [0.9933 ]
user and group training time is: [1201.2 s]
User Iteration 0 [1203.3 s]: HR = 0.2133, NDCG = 0.1389, [2.2 s]
Group Iteration 0 [1246.6 s]: HR = 0.1897, NDCG = 0.1229, [45.5 s]
Iteration 1, loss is [0.9112 ]
...

pytorch中文官方文档的地址为：

• https://pytorch-cn.readthedocs.io/zh/latest/

真正的努力， 都 不喧 嚣！

搜索与推荐Wiki

All In CTR、DL、ML、RL、NLP

原创不易，点个 “在 看” 鼓励鼓励吧