Fork me on GitHub

王冬月:京东推荐算法精排技术实践

图片

分享嘉宾:王冬月 京东 机构负责人
编辑整理:吴祺尧
出品平台:DataFunTalk

导读: 京东主站推荐的主要业务场景覆盖APP中用户购买前、购买中和购买后的全流程,日均大概有7000万DAU。不同业务场景下用户需求不同,我们相应的优化目标也不同。首页feeds和“我的京东”就是典型的购买前的场景,我们主要去优化点击以及浏览深度;购物车和商品详情页是购买中的场景,主要优化点击率以及转化率;而订单页就是用户购买后的主要场景,会重点优化跨品点击和转化。

在技术上,我们的排序链路主要分为召回、粗排、精排和重排等四个模块,不同模块的优化方向不同。今天我会主要介绍精排模型的相关实践与优化,包括:

  • 业务&场景
  • 多目标优化
  • 用户行为序列建模
  • 精细行为建模
  • 多模态特征

01 业务&场景

首先和大家分享下京东推荐中的排序链路以及主要挑战。

图片

召回主要是包括主流的i2i和u2i召回,它的排序是数十亿到万的量级;粗排采用双塔模型,主要兼顾精准性和系统延时,排序量是从万到千的量级;在精排阶段,我们会使用更复杂的特征和模型结构,保证结果的精准性,它的排序量级是从千到百;重排模块则更多从整个序列的角度考虑问题,同时兼顾多样性、新颖性,量级是从百到十。

精排模块的挑战主要有三个:

  • 多目标:推荐系统通常有多个业务目标需要同时优化,包括点击率、转化率、GMV、浏览深度和品类丰富度等。
  • 用户表达:用户的偏好是多样性的,精准表征用户当前的意图是比较有挑战的。
  • 商品表达:商品信息包括品牌、品类等属性,也包括用户行为的反馈信息,还有商品的主图和标题等。如何利用不同模态的数据也是目前比热门的问题。

我们采用了多目标学习、用户行为序列建模和多模态特征等业界主流方法来优化这三个问题,并针对电商推荐的特性进行了一定改进和创新。

02 多目标优化

图片

推荐是一个多目标优化问题,实际上在线上我们有大约七到八个目标需要优化。为了简化呈现,这里就以点击和转化为例。上图右侧是传统的多目标优化模型。最底层是模型的输入特征,中间是多目标模型,之后连接了一个combination层将不同的目标预测值结合到一起,最后使用其进行排序。

推荐领域多目标学习的主要问题有:知识冲突,效果较好的MMoE/PLE等模型,主要通过更灵活的参数共享机制,为不同学习目标提供可选的稀疏结构来缓解知识冲突;融合参数的优化,融合参数主要是用来组合不同目标的输出用于排序,常用做法是通过离线grid search或者在线自动调参来搜索一组固定参数。

在多目标融合排序上,我们认为固定参数有一定的问题。用户在不同场景有不同的诉求,如有明确的购物意图或者仅仅是闲逛,固定的融合参数就无法使模型整体效果达到最优。因此我们提出了一个个性化多目标融合的方法,可以根据用户状态自动输出不同目标的融合权重。

图片

上图右侧是我们整体的模型架构,主网络是一个传统的MMoE结构;我们额外加入了一个融合网络,通过一个Gate结构学习如何组合不同目标。融合网络的输入是与主网络相同的底层特征,输出的是多目标融合参数向量,其最终会与主网络不同目标的预测值进行点积,作为最终的排序值。

图片

融合网络的损失函数与主网络类似,主网络主要通过优化不同学习目标的参数来优化整体损失函数,而融合网络主要学习如何组合不同目标来使得模型在整体损失方向上达到最优。在模型训练上,我们采用了交替训练的方法。每10个batch训练一次融合网络,而每个batch都会去训练主网络。除此之外,我们还采用了梯度阻断的技术,融合网络的梯度不反传回底层的用户行为序列建模和embedding layer,从而保证主网络的训练不受融合网络训练的影响。

图片

通过加入融合网络的方法,我们希望模型在用户有明确的购物欲望的时候,模型可以给转化的输出更多的权重,而当用户没有明确的购物欲望的时候,模型可以给点击更大的权重。为了验证这个问题,我们进行了多组实验。

首先通过分析融合网络的均值和标准差,我们发现模型端到端学习出的权重值与我们离线通过参数搜索得到的最优值比较接近,结果较为合理;此外融合网络的输出具有样本级别差异,到达了个性化效果。然后我们对比了闲逛和购买场景的输出差异,在点击session下,点击融合权重平均是0.92,订单session中对应的权重下降为0.89;而在订单session中,订单的融合权重对比点击session会有比较大的提升。最终线上的效果也证明了模型可以精准识别用户属于偏买还是偏逛的类型,转化率有了大幅提升。除了首页之外,我们还将该方法应用到直播和购物车等不同排序模型上,其转化率都得到了明显提升,证明了方法的有效性和通用性。

03 用户行为序列建模

图片

接下来介绍用户行为序列建模的工作。整个用户行为序列建模的目的是想从用户的原始行为序列中学习用户的偏好,这是个性化推荐中最重要的一个技术分支。它的整体技术框架如上图所示,模型的输入是用户原始点击或者下单行为的序列,之后通过一个User Interest 模型来输出表征用户偏好的稠密向量。User Interest模型可以是pooling-based也可以是model-based,一般与排序任务一起端到端训练。

用户行为序列建模的挑战主要由以下几点:

  • 用户的行为以及兴趣是多种多样的,我们需要从用户的原始行为序列中将不同的兴趣提取出来;
  • 用户的行为是随着时间不断变化的,比如在一个月前用户感兴趣的商品现在就失去兴趣了,即具有衰变性和演变性;
  • 数据具有噪音,因为不像NLP领域,文字的排列会受基本语法的约束,用户的行为本身并没有约束,导致序列中可能会包含噪音行为,即没有特别关联性的行为。比如昨天我还在浏览手机这一品类,今天我突然就去浏览水果。
  • 模型应用在线上服务时存在比较明显的性能问题,我们需要优化预测的时延。

图片

用户行为序列建模在最近几年发展较快。第一个主要工作就是YoutubeDNN的Pooling Basded的方法,主要问题是信息丢失严重,兴趣表征与待排商品也没有任何相关性。为了解决这些问题,基于attention的模型(如DIN)被提出,它可以从用户行为序列中挑选出与待排商品比较相关的行为特征,主要问题是无法建模衰变性或者演变性,不同时间或者次序的行为表征的兴趣没有差异性。之后基于RNN-based的模型解决了这一问题,但是这类模型受顺序推理的限制,无法建模较长的序列。在工业界RNN-based的序列模型通常只能对大约50个行为进行建模。目前业界应用最广泛的是基于Transformer的方法,它对用户行为序列的表征能力更强,且在一定程度上可以无视序列的长度。此外,虽然模型结构比RNN更复杂,但是由于模型自身已于并行的特点,它在工程上更好被优化。

图片

我们的行为序列建模模块主要以Transformer为基础,并针对电商领域用户行为的特性进行了相应优化。整体框架如上图右侧所示,采用一个encoder-decoder的结构来对用户兴趣进行表征和提取。用户的偏好是多样性的,如何建模不同时期的偏好以及相应的演变是一个非常有挑战的工作。原始NLP Transformer采用position encoding来补偿模型所丢失的位置信息。借鉴这一想法,我们在相同位置进行了时序编码,除了位置信息(position encoding),还增加了时间差(recency encoding)的编码信息,并通过直接相加来融合。Position和recency embedding与排序模型一起,端到端学习用户行为次序和时间对兴趣强弱的影响。

模型中encoder模块主要负责对用户偏好进行表达,与原始Transformer类似,采用一个两层的self-attention + ffn的网络结构。相同的商品在不同的用户行为序列中代表的兴趣有所不同,我们需要借助用户历史中其他行为来表征每个行为背后的精准意图。与原始Transformer不同的是,我们借鉴特征交叉的思想,同时保留了一阶和二阶兴趣表征向量。在decoder端,我们使用multi-head target attention,从encoder输出的一阶和二阶向量中提取出与待排商品相关的一阶和二阶偏好。

图片

在工程上,由于模型计算复杂性的大幅提升,线上延时增加了约40毫秒,无法满足上线要求。为了解决这个问题,我们通过合并计算和并行推理来优化线上延时。在离线训练时每个batch的样本属于不同的用户,那么我们需要对batch中每个的行为序列进行独立的encoder推理,其时间复杂度是立方级的;但是在线上推理时,同一用户一次请求,一个batch内的encoder推理结果相同,那么我们只要计算一次即可,时间复杂度就可以降低为平方级。此外,通过调整计算图的执行顺序使得多个图可以同时去推理。最终从线上效果来看,tp99累计可以下降至33毫秒,使得模型满足全量上线要求。

图片

最后,我们对模型进行了离线分析,希望研究时序编码是否在端到端学习中学习到足够的信息。从业务上来讲,我们认为用户越近期的行为对当前的用户偏好影响越大。从上图右侧的结果展示了模型的多个head学习的信息。我们可以看到行为次序越靠后,纵坐标表示的attention值会有某种比较明显的倾向性,表明模型确实学习到了兴趣的演变特性。模型上线后,推荐结果的精准性、丰富性和新颖性均得到大幅提升。在首页feed流,人均点击提升了3.5%,这也是我们使用深度模型后单次优化得到的最大提升。

04 精细行为建模

图片

现在介绍一下我们在精细行为建模方向的工作。用户的行为中往往包含较多的噪声,包括一些误点击操作,需要从充满噪声的序列中识别用户的真实意图,因此我们提出了基于精细行为的兴趣建模来优化这一问题。

精细行为是用户点击后在商品详情页的统计量和子行为信息,包括停留时长、操作次数、查看主图、浏览评论等。

图片

整体方案以前面提到的改进的Transformer为基础,通过精细行为来建模用户点击后的真正意图和兴趣强弱。如上图所示,我们通过一个MLP将精细行为映射到一个定长的向量上,之后在通过bitwise-add的方式与时空编码信息结合。停留时长和操作数等统计信息会进行离散化,并通过端到端学习。查看主图、浏览评论等精细行为直接学习ID表征,并通过sum-pooling聚合。

图片

最后我们分析了精细统计属性对兴趣提取的影响,与之前的模型结论类似,我们也能观察到停留时长和attention之间存在某种趋势,停留越久attention值就越大。最终效果,在首页feeds流,人均点击提升2%,转化也提升了1.7%。

05 多模态特征

图片

最后介绍一下我们在多模态方面的工作。影响用户点击决策的因素除了商品基本属性,还受商品的主图和标题的影响。此外,推荐系统存在冷启动问题,比如一些新品是缺乏用户反馈的,它们的特征embedding处于欠拟合状态。为了解决这个问题,我们引入了多模态特征。多模态特征可以带来一定的信息增益,商品的主图和标题中非结构化的视觉和语义信息可以比较好地表征商品的状态。另外,多模态特征对缓解冷启动问题效果较好,因为这一类特征不需要用户的反馈,它的特征覆盖率在我们的业务场景下较高。

图片

模型整体框架是通过预训练模型+微调分别得到图像和文本向量,并基于Transformer进行视觉和语义偏好的建模。目前图像多模态模型已经在京东推荐中全量上线,而文本模型还在实验中,所以这里主要介绍一下我们在图像方面的工作。

由于预训练模型采用的是resnet101,它的训练数据分布和电商领域的图片分布有较大的差异,所以我们使用了电商主图的产品词分类数据对模型进行微调,使得模型能够学习到电商商品的特征。最终,我们把商品主图输入至图像模型,将最后一层64维向量作为图像embedding,应用于排序模型。

在进行多模态的偏好建模过程中,有两个主要问题需要解决。首先,图像embedding和其他端到端训练的embedding来自于不同模型,这使得它们处于不同的特征空间。为了对齐向量空间,我们加入了一个transfer layer,采用了一个multi-gate multi-expert的结构,以商品的品类作为gate的输入,选取不同的映射expert,将图像特征与其他端到端的特征进行对齐。最终,我们会将转变后的图像embedding和sku embedding拼接到一起,并通过上文提到的Transformer进行统一编码。

图片

最后我们对图像预训练向量的质量进行了分析,具体做法是筛选一部分商品,观察他们向量的相似度。通过上图可以看到,感官上相近商品的相似度比较高,而当色差较大时相似度也会有相应的降低。如果是差异较大的商品,图像相似度也比较小。在线上效果方面,我们将该方法应用到了首页feeds和首页核心频道,效果比较显著,人均点击分别提升2.6%和5%。

06 精彩问答

Q1:多目标融合模型中实际在线上会有多少个目标?每个目标的量级不一样,如何保证模型不受某一个目标主导?

A:在首页,我们有6个目标。业务上,通常会有主要的优化目标,重要性不同;技术上,也会对不同目标进行相应的加权操作,使得他们的量级处于一个相对平衡的状态。

Q2:线上会同时使用多目标模型的多路输出吗?

A:会的。个性化多目标模型中就是把多个目标的输出融合后作为排序依据的。

Q3:能不能再详细介绍一下多模态向量表征中图像embedding的微调方法?

A:首先,我们的训练数据是商品的主图,对应的标签是产品词的分类。在京东的业务场景下,产品词大概有十万个类别。我们会把数据输入到预训练的模型中,冻结住除了最后一个分类层外的所有网络参数,只微调分类层。最后我们会将分类层输出的64维向量取出,作为商品的图像embedding。

Q4:多模态融合模型有考虑去做联合训练吗?

A:这是技术选型的问题,因为联合训练通常来说会是单独使用图像的模型去做联合训练,或者单独使用文本模型去做联合训练。但实际上我们要在线上使用的话,图像、文本以及其他一些模型输出的embedding都会被使用,这就涉及到了三到四种不同模型的联合训练。这种做法优化起来十分困难,对于线上服务以及后续迭代都会有很大的影响。所以,实际上我们不会去采用多模态联合训练的方式。

Q5:可以再详细介绍一下多目标融合模型中的损失函数的优化方式和设计思路吗?

A:多目标融合模型的主网络和融合网络分别对应着两个损失函数和两个优化器。我们会分别训练这两部分网络,其中主网络输出的是多目标的预测值,比如点击率和转化率;另一个网络输出的是多目标预测值的融合权重,比如点击率的权重和转化率的权重。最终,模型使用点积的方式将多个目标的输出融合为一个预测值。在优化思路上,我们希望主网络优化各个目标的参数来达到最优;融合网络不影响主网络的更新,它学习如何组合主网络不同目标的预测值,来使得最终的排序能在优化方向上最优。

分享嘉宾:

图片


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