作者:jessiexyliu 腾讯PCG算法工程师
|导语 随着小说召回业务的需求发展,在注重点击率的同时,还需要关注阅读转化率的效果,故此我们开始了多目标召回模型的探索之旅。
一. 背景介绍
在一般的召回场景中,以点击为正样本来训练模型就基本可以满足需求,但是在小说推荐场景,点击其实只是用户很浅的小说行为,我们更关注的是如何让用户产生真正的阅读。在这样的背景下,我们希望可以搭建一个以阅读为目标的模型,无奈现实是骨感的,用户真实阅读的样本数远远小于推荐系统展现给用户的数量,同时有真实阅读的用户也仅占全量用户的一小部分,即存在数据稀疏问题(data sparsity,DS)。由此,我们考虑将点击和阅读相结合,用点击数据帮助训练阅读指标,多目标召回模型也就应运而生。
二. 多目标召回模型的演进
1. 初版模型:添加辅助loss
当我们要同时考虑点击和阅读效果时,最直观易行的方式就是先保持DSSM双塔结构不变,在原有的点击单目标loss基础上,加入阅读的辅助loss,希望模型在训练用户与点击书籍的向量相似度的同时,提升用户与阅读书籍的相似度。具体loss公式如下:
这样做的优点比较明显,操作简单易行,同时可以有效的防止过拟合。但模型的问题也显而易见,因为两者的embedding是完全共享的,我们的美好预期是一个embedding就可以既表达点击又表达阅读,但是实际由于点击和阅读向量本身会存在一定差异,强行用一个embedding表示,可能会同时折损两方信息,最后事倍功半。从离线效果评估来看,这版的指标也确实没有明显的效果,由此说明,还是要对模型结构做出一定调整,来适配多目标的需求。
离线指标:
2. 基于多塔结构的多目标召回模型:
由前面的离线验证效果可知,双塔结构一定程度会束缚模型对user和item的表达,这时很自然的就想到,需要用多塔来进行user和item的表征,但如何在“互相掣肘”和“互不干预”之间找到平衡,是多目标模型一直在力求解决的问题。我们这里同时尝试了多种模型构造方式。
1)四塔结构:这里是借鉴ESSM的思想,只共享底层embedding look-up table,上层用两个塔分别训练两个label对应的embedding向量。这里是希望在embedding层面完全将两个label的训练分开,只在计算loss的时候做两者的聚合,优点是不会再出现上面提到的“相互掣肘”的情况,缺点是没有解决阅读label样本稀疏问题,会导致对应user embedding的过拟合问题。
2)三塔结构:在上面四塔结构的基础上,将user合并为一个塔训练,item用两个塔分别训练两个label对应的embedding。这里的考虑还是阅读label的样本稀疏问题,所以user tower选择将两个目标合并训练,而相对来讲书籍的稀疏性并不明显,所以可以保留书籍侧的双塔双目标结构。
下面是两个结构的离线指标,可以看出两者在点击和阅读的hit ratio(即召回率指标)都有小幅的提升,但也都不显著。
3.基于MMOE的多目标召回模型
到目前为止,我们其实只是在“互相掣肘”和“互不干预”之间相互摇摆,还远没有达到“相辅相成”的程度。不论是Share Bottom Model的模型结构,还是soft参数共享的相关结构,都或多或少也存在现在的问题。那么该如何让两个目标既能相互作用,又能保留自己的独特信息呢?这时MMOE模型进入了我们的视野。
MMOE结构如下:
由于很多文章都已经介绍过MMOE模型,在此我就不再赘述了,对我们的场景进行MMOE的适配后,离线实验效果如下:
可以看到借鉴mmoe结构的离线效果最好,于是我们选取这版模型上线实验,最后从实验效果来看,这版模型也确实带来了阅读转化率的提升。
三. 后续优化
1. 多目标loss的不同组合方式:目前我们只采用了简单的线性加权,在确定模型结构后对权重值做微调,但其实这里的花样是很多的,比如用时长作为label的加权,或者ESSM一样的乘积方式等等,这里我们后续也会做更多的尝试。
2.模型结构的尝试:我们现有的优化逻辑,都是基于最初的单目标版本层层改进,但其实很多其他类型的模型结构都没有尝试过,比如阿里从MIND到ComiRec的Capsule网络系列等。
多目标召回领域还有很多东西可以实践探索,非常欢迎感兴趣的朋友们一起交流共同进步。
参考文献:
[1] Huang, Po-Sen, et al. "Learning deep structured semantic models for web search using clickthrough data." Proceedings of the 22nd ACM international conference on Information & Knowledge Management. 2013.
[2] Ma, Jiaqi, et al. "Modeling task relationships in multi-task learning with multi-gate mixture-of-experts." Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2018.
[3] Ma, Xiao, et al. "Entire space multi-task model: An effective approach for estimating post-click conversion rate." The 41st International ACM SIGIR Conference on Research & Development in Information Retrieval. 2018.
近期热文
浅谈SWOT分析法
《QQ炫舞》十二年成熟游戏的求变思新之路
基于Scrapy的爬虫解决方案
让我知道你在看