推荐系统遇上深度学习 (十八)-- 探秘阿里之深度兴趣网络(DIN) 浅析及实现


本文地址:http://www.6aiq.com/article/1547894084444
知乎专栏 点击关注
本文版权归作者和AIQ共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出

石晓文的学习日记 
发表时间: 2018.06.26

阿里近几年公开的推荐领域算法有许多,既有传统领域的探索如 MLR 算法,还有深度学习领域的探索如 entire -space multi-task model,Deep Interest Network 等,同时跟清华大学合作展开了强化学习领域的探索,提出了 MARDPG 算法。

上一篇,我们介绍了 MLR 算法,通过分而治之的思想改进了传统的 LR 算法,使其能够拟合更复杂的线性关系。这一篇,我们来简单理解和实现一下阿里在去年提出的另一个重要的推荐系统模型 - 深度兴趣网络 (DIN,Deep Interest Network). 该方法由盖坤大神领导的阿里妈妈的精准定向检索及基础算法团队提出,充分利用 / 挖掘用户历史行为数据中的信息来提高 CTR 预估的性能。

1、背景

深度学习在 CTR 预估领域已经有了广泛的应用,常见的算法比如 Wide&Deep,DeepFM 等。这些方法一般的思路是:通过 Embedding 层,将高维离散特征转换为固定长度的连续特征,然后通过多个全联接层,最后通过一个 sigmoid 函数转化为 0-1 值,代表点击的概率。即Sparse Features -> Embedding Vector -> MLPs -> Sigmoid -> Output.

这种方法的优点在于:通过神经网络可以拟合高阶的非线性关系,同时减少了人工特征的工作量。

不过,阿里的研究者们通过观察收集到的线上数据,发现了用户行为数据中有两个很重要的特性:

Diversity:用户在浏览电商网站的过程中显示出的兴趣是十分多样性的。

Local activation: 由于用户兴趣的多样性,只有部分历史数据会影响到当次推荐的物品是否被点击,而不是所有的历史记录。

这两种特性是密不可分的。

举个简单的例子,观察下面的表格:
5997513d08c249ca9d54d7f33677da49.png

Diversity体现在年轻的母亲的历史记录中体现的兴趣十分广泛,涵盖羊毛衫、手提袋、耳环、童装、运动装等等。而爱好游泳的人同样兴趣广泛,历史记录涉及浴装、旅游手册、踏水板、马铃薯、冰激凌、坚果等等。

Local activation体现在,当我们给爱好游泳的人推荐 goggle(护目镜) 时,跟他之前是否购买过薯片、书籍、冰激凌的关系就不大了,而跟他游泳相关的历史记录如游泳帽的关系就比较密切。

针对上面提到的用户行为中存在的两种特性,阿里将其运用于自身的推荐系统中,推出了深度兴趣网路 DIN,接下来,我们就一起来看一下模型的一些实现细节,然后我们会给出一个简化版的 tensorflow 实现。

2、模型设计

整体框架

我们先来看一下推荐系统的整体框架:
2bb8f15a966b4d69ab0f6c8d31185df2.png

1. 检查用户历史行为数据

2. 使用 matching module 产生候选 ads。

3. 通过 ranking module 做 point-wise 的排序,即得到每个候选 ads 的点击概率,并根据概率排序得到推荐列表。

4. 记录下用户在当前展示广告下的反应 (点击与否),作为 label。

特征设计

本文将所涉及到的特征分为四个部分:用户特征、用户行为特征、广告特征、上下文特征,具体如下:
e72426d1abf74ca09988152fd5e08a10.png

的非 0 值个数是不一样的,常见的做法就是将 id 转换成 embedding 之后,加一层 pooling 层,比如 average-pooling,sum-pooling,max-pooling。DIN 中使用的是 weighted-sum,其实就是加权的 sum-pooling,权重经过一个 activation unit 计算得到。这里我们后面还会再介绍到。

BaseModel

在介绍 DIN 之前,我们先来看一下一个基准模型,结构如下:

216df3d5d3f740edb377ed4732c0c355.png

这里 element-wise 的意思其实就是元素级别的加减,同时,可不要忽略广播的存在哟。一个元素和一个向量相乘,也可以看作 element-wise 的,因为这个元素会广播成和向量一样的长度嘛,嘻嘻。

可以看到,Base Model 首先吧 one-hot 或 multi-hot 特征转换为特定长度的 embedding,作为模型的输入,然后经过一个 DNN 的 part,得到最终的预估值。特别地,针对 multi-hot 的特征,做了一次 element-wise+ 的操作,这里其实就是 sum-pooling,这样,不管特征中有多少个非 0 值,经过转换之后的长度都是一样的!

Deep Interest Network

Base Model 有一个很大的问题,它对用户的历史行为是同等对待的,没有做任何处理,这显然是不合理的。一个很显然的例子,离现在越近的行为,越能反映你当前的兴趣。因此,对用户历史行为基于 Attention 机制进行一个加权,阿里提出了深度兴趣网络(Deep Interest Network),先来看一下模型结构:
1e6665c73e7f4f89929c5d7c09b8cd16.png

Attention 机制简单的理解就是,针对不同的广告,用户历史行为与该广告的权重是不同的。假设用户有 ABC 三个历史行为,对于广告 D,那么 ABC 的权重可能是 0.8、0.1、0.1;对于广告 E,那么 ABC 的权重可能是 0.3、0.6、0.1。这里的权重,就是 Attention 机制即上图中的 Activation Unit 所需要学习的。

为什么要引入这一个机制呢?难道仅仅是通过观察历史数据拍脑袋决定的么?当然不是,如果不用 Local activation 的话,将会出现下面的情况:假设用户的兴趣的 Embedding 是 Vu,候选广告的 Embedding 是 Va,用户兴趣和候选的广告的相关性可以写作 F(U,A) = Va * Vu。如果没有 Local activation 机制的话,那么同一个用户对于不同的广告,Vu 都是相同的。举例来说,如果有两个广告 A 和 B,用户兴趣和 A,B 的相似性都很高,那么在 Va 和 Vb 连线上的广告都会有很高的相似性。这样的限制使得模型非常难学习到有效的用户和广告的 embedidng 表示。

在加入 Activation Unit 之后,用户的兴趣表示计算如下:
bb4b25c396cc4c79bbae1bbab7feaba3.png

其中,Vi 表示 behavior id i 的嵌入向量,比如 good_id,shop_id 等。Vu 是所有 behavior ids 的加权和,表示的是用户兴趣;Va 是候选广告的嵌入向量;wi 是候选广告影响着每个 behavior id 的权重,也就是 Local Activation。wi 通过 Activation Unit 计算得出,这一块用函数去拟合,表示为 g(Vi,Va)。

3、模型细节

3.1 评价指标 GAUC

模型使用的评价指标是 GAUC,我们先来看一下 GAUC 的计算公式:
7e01580337a14f09872421cab6acdb93.png

我们首先要肯定的是,AUC 是要分用户看的,我们的模型的预测结果,只要能够保证对每个用户来说,他想要的结果排在前面就好了。

假设有两个用户 A 和 B,每个用户都有 10 个商品,10 个商品中有 5 个是正样本,我们分别用 TA,TB,FA,FB 来表示两个用户的正样本和负样本。也就是说,20 个商品中有 10 个是正样本。假设模型预测的结果大小排序依次为 TA,FA,TB,FB。如果把两个用户的结果混起来看,AUC 并不是很高,因为有 5 个正样本排在了后面,但是分开看的话,每个用户的正样本都排在了负样本之前,AUC 应该是 1。显然,分开看更容易体现模型的效果,这样消除了用户本身的差异。

但是上文中所说的差异是在用户点击数即样本数相同的情况下说的。还有一种差异是用户的展示次数或者点击数,如果一个用户有 1 个正样本,10 个负样本,另一个用户有 5 个正样本,50 个负样本,这种差异同样需要消除。那么 GAUC 的计算,不仅将每个用户的 AUC 分开计算,同时根据用户的展示数或者点击数来对每个用户的 AUC 进行加权处理。进一步消除了用户偏差对模型的影响。通过实验证明,GAUC 确实是一个更加合理的评价指标。



3.2 Dice 激活函数

从 Relu 到 PRelu

Relu 激活函数形式如下:

c5bd3b4f6cee4591b721c1b48a1a996e.png

Relu 激活函数在值大于 0 时原样输出,小于 0 时输出为 0。这样的话导致了许多网络节点的更新缓慢。因此又了 PRelu,也叫 Leaky Relu,形式如下:
2e79571f86414cc997f444336b4cc5a5.png

这样,及时值小于 0,网络的参数也得以更新,加快了收敛速度。

从 PReLU 到 Dice

尽管对 Relu 进行了修正得到了 PRelu,但是仍然有一个问题,即我们认为分割点都是 0,但实际上,分割点应该由数据决定,因此文中提出了 Dice 激活函数

Dice 激活函数的全称是 Data Dependent Activation Function,形式如下:
0f324f2a9a34410ca9ce32fda8e49dbc.png

其中,期望和方差的计算如下:
7c3d8f96d49449698e69b5ec6165fca7.png

可也看到,每一个 yi 对应了一个概率值 pi。pi 的计算主要分为两步:将 yi 进行标准化和进行 sigmoid 变换。

3.3 自适应正则 Adaptive Regularization

CTR 中输入稀疏而且维度高,通常的做法是加入 L1、L2、Dropout 等防止过拟合。但是论文中尝试后效果都不是很好。用户数据符合长尾定律 long-tail law,也就是说很多的 feature id 只出现了几次,而一小部分 feature id 出现很多次。这在训练过程中增加了很多噪声,并且加重了过拟合。

对于这个问题一个简单的处理办法就是:直接去掉出现次数比较少的 feature id。但是这样就人为的丢掉了一些信息,导致模型更加容易过拟合,同时阈值的设定作为一个新的超参数,也是需要大量的实验来选择的。

因此,阿里提出了自适应正则的做法,即:

1. 针对 feature id 出现的频率,来自适应的调整他们正则化的强度;

2. 对于出现频率高的,给与较小的正则化强度;

3. 对于出现频率低的,给予较大的正则化强度。

计算公式如下:
612f59ed99f7464aa201c6c32bab92cf.png

4、效果展示

下图是对 Local Activation 效果的一个展示,可以看到,对于候选的广告是一件衣服的时候,用户历史行为中跟衣服相关的权重较高,而非衣服的部分,权重较低。
b437e4aeb2f44905a021c7829564e39d.png

下图是对使用不同正则项的结果进行的展示,可以发现,使用自适应正则的情况下,模型的验证集误差和验证集 GAUC 均是最好的。
ede65b1e69644527828cbd05b47469d3.png

下图对比了 Base Model 和 DIN 的实验结果,可以看到,DIN 模型在加入 Dice 激活函数以及自适应正则之后,模型的效果有了一定的提升:
22d1191b4987410dba140e1d0caec67e.png

5、实战 DIN

本文的 github 地址为:https://github.com/princewen/tensorflow_practice/tree/master/recommendation/Basic-DIN-Demo

参考的 github 地址为:https://github.com/zhougr1993/DeepInterestNetwork/tree/master/din

这里我们只给出一些模型细节的实现,具体的数据处理以及其他方面的内容大家可以根据上面两个地址进行学习:

数据准备

按照下面的方法下载数据:
bdb1844b9b6d49f689f1277e5cd14528.png

Dice 激活函数

这里实现的 Dice 激活函数没有根据上一步的均值方差来计算这一步的均值方差,而是直接计算了这个 batch 的均值方差。我们可以根据计算出的均值方差对 x 进行标准化 (代码中被注释掉了),也可以直接调用 batch_normalization 来对输入进行标准化。

注意的一点是,alpha 也是需要训练的一个参数。

import tensorflow as tf

def dice(_x,axis=-1,epsilon=0.0000001,name=''):

 alphas = tf.get_variable('alpha'+name,_x.get_shape()[-1],
 initializer = tf.constant_initializer(0.0),
 dtype=tf.float32)

 input_shape = list(_x.get_shape())
 reduction_axes = list(range(len(input_shape)))

 del reduction_axes[axis] # [0]

 broadcast_shape = [1] * len(input_shape) #[1,1]
 broadcast_shape[axis] = input_shape[axis] # [1 * hidden_unit_size]

 # case: train mode (uses stats of the current batch)
 mean = tf.reduce_mean(_x, axis=reduction_axes) # [1 * hidden_unit_size]
 brodcast_mean = tf.reshape(mean, broadcast_shape)
 std = tf.reduce_mean(tf.square(_x - brodcast_mean) + epsilon, axis=reduction_axes)
 std = tf.sqrt(std)
 brodcast_std = tf.reshape(std, broadcast_shape) #[1 * hidden_unit_size]
 # x_normed = (_x - brodcast_mean) / (brodcast_std + epsilon)
 x_normed = tf.layers.batch_normalization(_x, center=False, scale=False) # a simple way to use BN to calculate x_p
 x_p = tf.sigmoid(x_normed)

 return alphas * (1.0 - x_p) * _x + x_p * _x

Activation Unit

这里的输入有三个,候选广告 queries,用户历史行为 keys,以及 Batch 中每个行为的长度。这里为什么要输入一个 keys_length 呢,因为每个用户发生过的历史行为是不一样多的,但是输入的 keys 维度是固定的 (都是历史行为最大的长度),因此我们需要这个长度来计算一个 mask,告诉模型哪些行为是没用的,哪些是用来计算用户兴趣分布的。

经过以下几个步骤得到用户的兴趣分布:

  1. 将 queries 变为和 keys 同样的形状 B * T * H(B 指 batch 的大小,T 指用户历史行为的最大长度,H 指 embedding 的长度)

  2. 通过三层神经网络得到 queries 和 keys 中每个 key 的权重,并经过 softmax 进行标准化

  3. 通过 weighted sum 得到最终用户的历史行为分布

def attention(queries,keys,keys_length):
 '''
 queries: [B, H]
 keys: [B, T, H]
 keys_length: [B]
 '''

 queries_hidden_units = queries.get_shape().as_list()[-1]
 queries = tf.tile(queries,[1,tf.shape(keys)[1]])
 queries = tf.reshape(queries,[-1,tf.shape(keys)[1],queries_hidden_units])

 din_all = tf.concat([queries,keys,queries-keys,queries * keys],axis=-1) # B*T*4H
 # 三层全链接
 d_layer_1_all = tf.layers.dense(din_all, 80, activation=tf.nn.sigmoid, name='f1_att')
 d_layer_2_all = tf.layers.dense(d_layer_1_all, 40, activation=tf.nn.sigmoid, name='f2_att')
 d_layer_3_all = tf.layers.dense(d_layer_2_all, 1, activation=None, name='f3_att') #B*T*1

 outputs = tf.reshape(d_layer_3_all,[-1,1,tf.shape(keys)[1]]) #B*1*T
 # Mask
 key_masks = tf.sequence_mask(keys_length,tf.shape(keys)[1])
 key_masks = tf.expand_dims(key_masks,1) # B*1*T
 paddings = tf.ones_like(outputs) * (-2 ** 32 + 1) # 在补足的地方附上一个很小的值,而不是0
 outputs = tf.where(key_masks,outputs,paddings) # B * 1 * T
 # Scale
 outputs = outputs / (keys.get_shape().as_list()[-1] ** 0.5)
 # Activation
 outputs = tf.nn.softmax(outputs) # B * 1 * T
 # Weighted Sum
 outputs = tf.matmul(outputs,keys) # B * 1 * H 三维矩阵相乘,相乘发生在后两维,即 B * (( 1 * T ) * ( T * H ))
 return outputs

参考文献

1、盖坤演讲视频:http://www.itdks.com/dakalive/detail/3166

2、论文:Deep Interest Network for Click-Through Rate Prediction

3、github:https://github.com/zhougr1993/DeepInterestNetwork


本文地址:http://www.6aiq.com/article/1547894084444
知乎专栏 点击关注
本文版权归作者和AIQ共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出