youtubeDNN及其优化

最新推荐文章于 2024-04-27 15:03:07 发布

码一码码码

最新推荐文章于 2024-04-27 15:03:07 发布

阅读量711

点赞数 2

文章标签：深度学习机器学习自然语言处理

本文链接：https://blog.csdn.net/weixin_39586997/article/details/117522552

版权

1.YouTube推荐系统需要解决的三大难点

数据规模：YouTube 的用户和视频量级都是十亿级别的，需要分布式学习算法和高效的部署。
新颖性：推荐系统需要及时对新上传的视频和用户的新行为作出响应。
数据噪音：由于稀疏和外部因素影响，用户的历史行为很难预测。大部分 YouTube 视频只有隐式反馈（即用户对视频的观看行为），缺少显式反馈（即用户对视频的评分）。此外，视频的元信息不够有结构性。我们的算法需要对训练数据的这些因素稳健（robust）。

2.整体架构

召回模型的网络结构：

3.模型结构和算法原理

在给定用户历史行为与上下文的情况下，通过DNN网络，学习user embedding向量，作为输入送到softmax分类器，生成召回候选集。

负类采用通过重要性采样softmax，避免了softmax在计算时要便利所有视频计算指数的问题，提升了训练速度，并使用重要性加权的方式校正这个采样。对每个样本，对真实标签和采样得到的负类，最小化其交叉熵损失函数。

youtube召回模型采用了word embedding的技巧来计算每一个视频的embedding，然后将视频的embedding，用户搜索视频的embedding分别计算average，再加入用户的属性、视频质量等特征，采用两个完全相连的ReLU层和softmax函数来预测用户下一个看的视频是什么。

使用DNN的原因之一，在DNN中连续性变量和类别型变量都很容易输入到模型中，包括一些人口统计特征（Demographic features），对最终的效果起着十分重要的作用。用户的地域，设备等都可以作为embedding向量输入到DNN中去。简单的二值化特征（如性别）和数值型特征（如年龄）可以直接输入到DNN中，数值型需要经过归一化到[0,1]再输入到模型中。

MLP层使用常见的塔状设计，底层最宽，往上每层的神经元数目减半，直到 Softmax 输入层是 256 维（1024ReLU->512ReLU->256ReLU）。

在线服务阶段，通过最近邻查询的方法，找到每个用户向量最相似的topk个视频向量，做为召回结果。

4.核心代码

5.工程实践经验：

（1）把推荐问题转化为了多分类问题，会面临softmax计算低效的问题，youtubeNet通过负采样的方式，并用importance weighting的方法对采样进行校准。同时，还有做hierarchical softmax的方法来解决这一问题。

（2）候选集线上服务时，采用了最近邻搜索的方法，避免了对全候选集的逐一计算，满足了线上实时性的要求。只需要将item embedding和user embedding存到redis中即可。

（3）user embedding和vid embedding是如何生成的：

user embedding是通过user观影历史的多个item embedding，拼接其特征embedding，然后将前n个embedding融合（通常是pooling的效果最好），然后经过三层的DNN网络得到；

vid embedding有两种实现方式，一种是取了之前user embedding中用的item embedding拼接特征embedding，另一种是单独生成一个vid embedding matrix，与user embedding同时训练。目前博主用的是第一种方法实现。

本质上softmax就是在做广义矩阵分解，分出user embedding和item embedding。

（4）针对用户更偏好新视频的问题，youtube通过引入”example age”（即模型训练时间与消费行为发生时间之间的差）这个特征来解决。

（5）在youtubeNet的原始论文中，在对训练集预处理时没有采用原始的用户日志，二是针对每个用户提取了等数量的训练样本，这是为了减少高度活跃用户对于loss的过度影响。