多任务学习在美图推荐排序的近期实践

一、前言

距笔者上次提出NFwFM点击率预估模型,并覆盖美图秀秀、美拍等推荐场景已经过去半年。算法钻研如逆水行舟,不进则退。用户和公司越来越高的期望也要求我们不断进步。

与此同时,美图社区内容推荐业务发展迅猛,有目共睹。社区内容推荐一方面是千人千面,精准连接用户与内容,另一方面是连接用户与用户。因此近半年来,我们的业务目标自然的从点击率预估发展到点击率和关注转化率的多目标预估模型。

本文介绍将多任务学习应用于点击率和关注转化率预估的一些思考和实践经验。这些实践已全量覆盖美图秀秀社区以及美拍热门等推荐场景。

二、算法选型

笔者管中窥豹, 将多目标预估分为两大类, 一类是多模型,一类是多任务学习。

多模型为每一个目标单独构造训练样本和特征,并采用合适的模型各自训练。其优点是可以为每个目标进行深度优化,且预估值融合阶段能够根据业务目标灵活控制各个目标的权重。缺点是需要配备多套人力和机器来存储和优化样本、特征以及调整融合公式。

学术界和业界近年来在多任务学习(Multi-task Learning, MTL)方面的研究和实践进展颇多。MTL是机器学习的一种范式,属于迁移学习的范畴,但与大家通常理解的迁移学习的区别在于多任务学习是要借助其它的任务来帮助提升所有任务而非某一个任务的学习效果。

当多个任务相关性强时,譬如点击率和关注转化率,多任务学习与迁移学习的本质区别便成为多任务的最大优点。此外,依笔者浅见,DNN灵活的网络结构,其在一个模型中能够包容多种不同分布的样本并各自优化其目标,更是释放了MTL的最大价值。其缺点是参数共享,当存在不那么相关的任务时,任务之间会相互扰乱,影响效果。

多任务学习本身一个很古老的概念[5]。对多任务学习做详细的介绍超出了本文的范围,感兴趣的同学可以参考文末的参考文献。

思虑至此,结合实际预估目标,我们很自然的选择多任务学习作为点击率预估和关注转化率预估的多目标预估模型。

三、Multi-task NFwFM

正如前面所说,多任务学习只是机器学习的一种范式,并不局限于特定的问题或者模型。因为有非常实际的工业需求,所以工业界和学术界的研究和实践成果还在不断增加。

比如2016年横空出世的YouTubeNet[1], KDD'18的MMoE[2]等学术研究;工业界如美团"猜你喜欢"引入Multi-task DNN后CTR提升1.23%[3], 知乎引入MMoE后互动率提升100%[4]。

在实践过程中为了追求训练效率高,模型可拓展性以及最重要的效果要好,工业界通常选择Shared-Bottom multi-task DNN[5]的网络结构。

如上图,Multi-task NFwFM的前4个隐含层是共享的,点击率预估任务和关注转化率预估任务共享特征表示。在最后即将预估多个目标时通过全连接层进行拆分,各自学习对应任务的参数,从而专注地拟合各自任务。

多任务学习的不同任务在共享层里的局部极小值位置是不同的,通过多任务之间不相关的部分的相互作用,有助于逃离局部极小值点;而多任务之前相关的部分则有利于底部共享层对通用特征表示的学习,因此通常多任务能够取得比单任务模型更好的效果。

在线上预估时,因为模型尺寸没有变化,推理效率和线上的点击率预估模型一致。考虑到我们是在点击率任务的基础上同时优化关注转化率,融合公式上体现为优先按照点击率排序再按照曝光→关注的转化率排序。

Multi-task NFwFM已在美图秀秀社区首页Feeds推荐、相关推荐下滑流全量上线。首页Feeds 点击率 +1.93%,关注转化率 +2.90%, 相关推荐下滑流人均浏览时长+10.33%, 关注转化率+9.30%。

四、样本reweight

上述普适版的Multi-task NFwFM带来了点击率和关注率转化率的稳定提升,这驱使我们进一步根据业务数据的特点压榨模型价值。

点击和关注转化样本生而不平等。在秀秀首页Feeds推荐场景下,点击样本数:关注样本数≈100:1。虽然Multi-task NFwFM包容了点击和关注转化这两种不同分布的样本,并支持各自优化其目标,然而点击样本和关注样本严重不平衡,这无疑会影响关注转化率预估目标的学习。

为了最大化模型的价值,reweight自然成为我们的选择。样本的研究和模型应用是同等重要的问题,相关研究仍然很活跃,不断有新算法被提出。代表性的工作有近期Kaiming He等人提出的Focus Loss[6], 其通过减少易学习样本的权重,使得模型在训练时更专注于难学习的样本。

Focus Loss的优点是可以通过超参数来控制两种样本的分布,缺点是在梯度下降类的优化算法中,样本无法随机shuffle,而这对梯度下降类算法的收敛速度至关重要。

回到实践本身,我们采用样本reweight的土办法,一方面,样本reweight天然的支持随机shuffle, 另一方面,从以往实践的经验来看,用更多、更高质量的数据进行训练几乎是总能带来提升的。

以下图为例,假设有4类feeds, A、B、C、D, 关注转化率的大小关系是A > C > B > D=0。实线部分表示4类feed的真实CTR,则虚线表示各类feed被关注样本reweight之后的虚拟CTR。

通过关注转化率reweight点击正样本,改变了CTR分布之后,模型的pCTR将倾向于C>D>A>B,保证在原始CTR接近的情况下,关注转化率较高的feed更容易被模型推荐出来,提升整体的关注转化率,并且因为C、D等点击率较高的feed不受分布改变后影响,因此整体CTR应不受太大影响。

基于上述关注样本reweight点击正样本的策略,美图秀秀社区首页Feeds推荐CTR -1.09%, 关注转化率 +12.03%。

五、Task-specific weighted Loss

样本reweight让我们在关注转化率目标上取得了显著的提升,但是CTR下降明显。笔者调整并灰度实验多个reweight候选值,尝试降低关注转化率对点击正样本的分布的影响,结果让人失望,CTR均低于线上。

很大可能的原因是上述图示对我们场景数据的假设太简单理想了。正如前文所述,多任务学习的缺点是参数是共享的,当多个任务存在不相关部分时,任务之间会相互扰乱,影响效果。

学术界将这个现象称为"share conflict",分析并解决这个问题的挑战较大,且业界和学术界可供参考的经验少。

另一种减小share conflict的曲线方式是gradient balancing method,在我们场景下对应的是加大multi-task NFwFM中点击率预估任务的重要性,让点击率预估任务来主导底部共享层的参数学习,进而影响整体模型的预估性能。

Yarin Gal在CVPR'18提出一种基于不确定估计的方法, Homoscedastic Uncertainty[7], 来设定多个任务的学习率,学习率越大对整体模型的学习主导性越强。

如下图,在秀秀社区首页Feed推荐场景中, Homoscedastic Uncertainty估计方法表明关注转化率的不确定性确实比点击率预估任务高,因此我们应该让点击率预估任务来主导整体模型的学习。

然而灰度实验期间,发现该方法估计出来的不确定性本身不稳定,参数容易学飘,甚至出现相反的结论,导致实验效果波动较大。在实践中,笔者采用效果正向提升的几天的不确定性值作为最终的参数。

上述改进的Multi-task NFwFM, 在美图秀秀社区首页Feeds推荐场景全量上线,点击率 -0.36%, 关注转化率+12.75%。

六、未来展望

对于多目标预估问题,我们已经实现了一套具有良好拓展性的系统,并在生产实践中取得了成功应用。纵观业界,不少公司也在生产中使用了各种多任务模型和算法。

同多任务学习的广阔空间相比,目前我们的实践还很初级,未来我们一方面会继续发挥多任务学习的优势,根据业务形态设计多任务学习目标,另一方面还会尝试更为复杂的模型和算法。

END

参考文献

  1. Covington P, Adams J, Sargin E. Deep neural networks for youtube recommendations
  2. Ma J, Zhao Z, Yi X, et al. Modeling task relationships in multi-task learning with multi-gate mixture-of-experts
  3. 美团"猜你喜欢"深度学习排序模型实践
  4. 进击的下一代推荐系统:多目标学习如何让知乎用户互动率提升100%?
  5. Rich Caruana. 1998. Multitask learning. In Learning to learn
  6. Lin T Y, Goyal P, Girshick R, et al. Focal loss for dense object detection
  7. Kendall A, Gal Y, Cipolla R. Multi-task learning using uncertainty to weigh losses for scene geometry and semantics

-------------

作者介绍

陈文强,美图高级算法技术专家。曾就职阿里巴巴、腾讯。具有多年推荐算法的实际落地经验,长期深耕于最优化算法、推荐算法领域的研究。NFwFM提出者,在Neural Computing等期刊发表数篇论文。

本文分享自微信公众号 - 美图数据技术团队(gh_feb1d206d92b)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2019-06-25

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏算法与编程之美

Python涨知识|你懂is和==的神奇区别吗

在编程中我们经常会用到“is”和“==”来表示判断,那么我想问大家一个问题为什么python会出现两个“表面”意思相近的语句呢?

8220
来自专栏AI科技大本营的专栏

过关斩将打进Kaggle竞赛Top 0.3%,我是这样做的

从下图可以看出,融合后的模型性能最好,RMSE 仅为 0.075,该融合模型用于最终预测。

19920
来自专栏算法和应用

讨论覆盖函数中偏函数扩展的复杂性

摘要:覆盖函数是子模块函数的重要子类,可用于机器学习,博弈论,社交网络和设施位置。我们研究了覆盖函数的偏函数扩展的复杂性。也就是说,给定由[m]的子集族和每个点...

11360
来自专栏机器之心

图解NumPy,这是理解数组最形象的一份教程了

NumPy 软件包是 Python 生态系统中数据分析、机器学习和科学计算的主力军。它极大地简化了向量和矩阵的操作处理。Python 的一些主要软件包(如 sc...

11820
来自专栏MixLab科技+设计实验室

本周AI见闻 - NO.03

其一,充分挖掘各类有润滑油需求的4S店信息,也可以结合润滑油公司的自有CRM数据库,实现档案补齐,直接提供给润滑油公司销售一定量的销售线索,也可以根据不同行业的...

4820
来自专栏机器之心

3D重建传统算法对比深度学习,SFU谭平:更需要的是二者的融合

近日,机器之心发表文章《 3D 重建:硬派几何求解 vs 深度学习打天下?》,对几何与深度学习两种方法进行了对比。

22120
来自专栏机器之心

超全深度学习细粒度图像分析:项目、综述、教程一网打尽

计算机视觉(CV)是用机器来理解和分析图像的过程,是人工智能中一个重要分支。在 CV 的各个研究领域中,细粒度图像分析(fine-grained image a...

13110
来自专栏用户5637037的专栏

人工智能模型的语义解释能力研究

原文题目:On the Semantic Interpretability of Artificial Intelligence Models

12240
来自专栏小闫笔记

初识人工智能

刚开始接触人工智能的时候,大家肯定看到了一些名词:人工智能、深度学习、机器学习...what??哈哈,先来简单的给大家解释一下这三者的区别,再来谈论其他的问题。...

24640
来自专栏机器之心

为何Keras中的CNN是有问题的,如何修复它们?

上个星期我做了一些实验,用了在 CIFAR10 数据集上训练的 VGG16。我需要从零开始训练模型,所以没有使用在 ImageNet 上预训练的版本。

10120

扫码关注云+社区

领取腾讯云代金券

年度创作总结 领取年终奖励