专栏首页小小挖掘机【技术干货】聊聊在大厂推荐场景中embedding都是怎么做的

【技术干货】聊聊在大厂推荐场景中embedding都是怎么做的

解读YouTube、Airbnb、Alibaba的三篇经典论文,总结Embedding在工业界的一些用法和技巧,这三篇论文亮点众多,提供的经验非常值得我们去细细品味和借鉴。这篇文章篇幅较多(2w字),几乎把三篇论文的重要内容都进行了解读和总结,需花点时间去研读,文中难免有错误和理解不对的地方,欢迎指正讨论!

作者:卢明冬,个人博客:https://lumingdong.cn/engineering-practice-of-embedding-in-recommendation-scenario.html

YouTube

Deep Neural Networks for YouTube Recommendations

YouTube的这篇论文发表在2016年9月的RecSys会议上,是一篇非常经典的推荐系统工程论文。在这篇论文中,不但尝试将DNN用在推荐系统上,更是讲述了众多工程经验,论文处处有亮点,处处可借鉴,如今该论文的中的一些工程经验已被各大互联网公司参考引入自己的推荐系统。

YouTube的推荐系统架构使用经典的**召回(Match 或 Candidate Generation)「和」排序(Ranking)**经典的二级架构:

  • 「召回阶段」通过i2i/u2i/u2u/user profile等方式完成候选视频的快速筛选[^1],这个过程将百万(millions)候选集降低到了几百(hundreds)的量级。
  • 「排序阶段」对召回后的视频采用更精细的特征计算 user-item 之间的排序得分,作为最终输出推荐结果的依据,这个过程将候选集从几百(hundreds)的量级降低至几十(dozens)的量级。

「不管是召回阶段还是排序阶段,模型架构上都是DNN的基本结构,不同的是输入特征和优化目标不同。」

「召回阶段」

召回阶段模型(也称候选生成模型,Candidate Generation Model)的架构如下图:

这个模型虽然是一个深度神经网络,但其实「可将DNN看作是普通的矩阵分解(MF)的一种泛化」,这里把推荐问题建模成一个“超大规模”多分类问题。即在时刻

t

,为用户

U

(上下文信息

C

)在视频库

V

中精准地预测出下一次观看(next watch)视频

i

的类别(每个具体的视频视为一个类别,

i

即为一个类别,观看完成的视频则视为正例),用一个Softmax公式表示这个条件概率:

P(w_t=i|U,C)=\frac{e^{v_iu}}{\sum_{j \in V}e^{v_ju}}

其中,向量

u \in R^N

是 <user, context> 信息的高维Embedding,而向量

v_j \in R^N

则表示每个候选视频

j

的Embedding。「所以DNN的目标就是在用户信息和上下文信息为输入条件下学习用户的Embedding向量」,用公式表达DNN就是在拟合函数

u=f_{DNN}(user\_info, context\_info)

接下来,我们总结在召回阶段使用的一些工程技巧。

  1. 「输入融合多种信息,使用全网数据而不是只使用用户浏览数据」 模型输入不仅仅只有用户浏览历史,还加入了用户的搜索历史,人口统计学信息,地理信息以及其余上下文信息,这些输入信息大部分都是先将变长的稀疏ID特征转换成固定宽度的Embedding向量化表示(固定维度的输入才能输入深度神经网络,也便于参数学习以及模型的工程化),除了Embedding类的特征,还有一些简单的二元特征以及连续型特征(比如用户性别、登录状态、年龄等)归一化到[0, 1]区间上的实数值,论文很巧妙地将「多种信息拼接(concat)成一个特别宽的输入向量」,然后传给后面的DNN模型的隐藏层,这种方式的带来的好处是「任何连续型特征和类别型特征都可以很方便地加进模型中,并且可以维持一致的宽度」。 另外,「多种信息的融合也带来几个额外的好处:」 「输入特征再展开来讲,里面又涉及很多特征处理经验:」
    • 「观看历史」:这里是通过非端到端的预训练方式先做了Embedding之后才作为DNN的输入,训练Embedding用的是用户历史观看视频的序列,得到视频的Embedding表示。而输入DNN模型的时候,是将用户观看视频历史所对应的视频Embedding序列进行向量平均得到固定维度的watch vector作为DNN的输入,注意这里向量平均的方式可以是简单的算术平均,也可以是「加权平均(可根据视频的重要程度和时间属性等进行加权)」
    • 「搜索历史」:搜索历史的处理和观看历史的处理方式类似,有些特别的是,这里先把历史搜索的关键词query分词得到词条token,训练得到token的Embedding向量,然后将用户历史搜索的token所对应的Embedding向量进行**(加权)平均**得到固定维度的search vector作为DNN的输入,这样便能反应用户搜索历史的整体状态。
    • 「人口统计学信息」:性别、年龄、地域等,「简单的二元特征以及连续型特征会归一化为[0, 1]区间上的实数值」
    • 「其他上下文信息」:设备、登录状态等。
    • 「样本年龄(Example Age)」:**这是一个很特别的特征,它通过捕捉用户兴趣的短期变化,用来拟合用户对新视频的偏好变化带来的偏差。**YouTube算法团队发现,用户特别偏爱那些新上的视频,尽管这些视频的内容可能与他们的偏好并不相关。一些新视频的流量攀升,一方面可能是用户本身喜欢新鲜感带来的一次传播,另一方面热门视频也存在着病毒式的二次传播现象,由此可见视频流行度其实是一个非常不稳定的分布,而机器学习很难捕捉到这种不稳定的变化。 由于机器学习模型通常是基于历史样本训练,来预测将来的行为,因此「机器学习模型常常会表现一种出趋向于过去行为的隐含偏差(implicit bias)」。同样的,在推荐系统模型上也存在这样的问题,推荐系统预测出的基于视频库产生的多项式分布,其实反映的是训练时间窗口内也就是近几周可能观看视频的平均情况,而不能真正获取到未来的信息。 为了修正这一偏差,YouTube算法团队在模型中加入了一个样本年龄(Example Age)的特征,文中并没有精确的定义什么是example age,不过按照文章的说法,example Age指的是训练样本的age,应该是以样本产生的时间(样本记录或样本日志标记的时间点)作为样本的“出生”时间,表示的是用户点击某个视频的时间,而不是视频发布的时间,样本产生时间到训练当前时间的距离作为样本年龄(example age)。除了修正趋于过去行为的这一偏差,「这个特征还可以一定程度上来捕捉用户的兴趣的短期变化」,比如模型可以学到example age越小的样本,也就是离训练这一时刻越近的样本,越符合用户的最近的兴趣,从而在模型中权重更大。也因此,「在线serving的时候这个特征会置零,表示模型在训练时间窗口的最末尾做预测,也就是要预测用户在这一刻的兴趣」「这是一种非常通用的来消除机器学习系统固有偏差的方法。另外训练和在线区别对待也非常值得借鉴,一些单侧特征,训练的时候把bias影响特征加进去,在线的时候则置0」,比如美团分享的消除广告position bias(位置给广告点击带来的偏差)也是类似的做法。[^2] 论文对这个特征进行了验证,对于给定视频,「Example Age这个特征能够很好的把视频的新鲜程度对流行度的影响引入模型中」,对比如下:

    从上图中我们也可以看到,在引入“Example Age”这个特征后,模型的预测效力更接近经验分布;而不引入Example Age的蓝线,模型在所有时间节点上的预测趋近于平均(baseline),这显然是不符合客观实际的。

    • 可以缓解冷启动问题
    • 为模型带来更全面的特征信息,提高用户Embedding的表征力度
    • 为推荐场景提供一些探索功能
  2. 「为每个用户生成固定数量的训练样本」 这也是一个非常实用的训练技巧,为每个用户固定样本数量上限,平等地对待每个用户,避免loss被少数活跃用户代表,能明显提升线上效果,这个方法有点类似协同过滤中对活跃用户的惩罚。
  3. 「丢弃用户观看视频以及历史搜索query的时序特征」 如果过多考虑时序的影响,用户的推荐结果将过多受最近观看或搜索的视频的影响,导致越推越窄,推荐结果也失去了多样性。所以为了综合考虑之前多次搜索和观看的信息,YouTube丢掉了时序信息,将用户近期的历史纪录对应的Embedding向量做平均或加权平均。
  4. 「输入数据只使用标签之前的历史信息,避免产生数据穿越」 「用户观看视频的行为存在着明显的不对称共同浏览(asymmetric co-watch)现象,即用户浏览视频的时候,往往是序列式的」,有些前后观看行为甚至存在一些因果关联,比如用户通常会在开始看一些比较流行的,后来会逐渐找到细分的视频,而对于那些连续剧集以及关联程度比较大的视频,这种现象表现得更加明显。 因此YouTube的推荐模型的训练样本构建并没有使用类似Word2Vec那样将前后上下文作为输入信息、中心词作为Label的留一法(held-out),而是将未来的信息完全分离,因为我们是预测用户的下一次观看,而非去预测一个随机留一的视频观看(a randomly held-out watch)。二者的区别见下图:

而实际上, **传统的协同过滤类算法,都是隐含地采用上面图(a) 的 heldout 方式,其实这种方式泄露了将来的信息(future information),忽略了用户观看视频存在着的时序性 **。论文中也提到,上面图(b)的方式在线上 A/B test中表现更好。

  1. 「召回模型在离线训练时,使用负例采样来解决Softmax的低效问题」 这个优化在上一篇文章中也提到过,传统的Softmax在大规模分类问题中效率非常低下,它通常有两种优化替代方式,一种是负例采样(negative sampling),一种是层次Softmax(hierarchical softmax),这里使用的是负例采样,并且按照重要性加权(importance weighting)来对采样进行校正(calibration)。因此对于每个样本有正负两种Label,最后的学习目标是最小化交叉熵损失。论文中提到,YouTube召回模型会抽样上千个负样本,这种方法可以比传统的Softmax快100倍,他们也尝试了另外一种层次Softmax的优化方式,但效果不如负例采样好。
  2. 「召回模型在线Serving过程中采用一种最近邻搜索的方法来提高效率」 因为召回阶段并不需要精确的打分校准,为了提高性能,YouTube在线召回阶段并没有直接采用训练好的模型进行预测,而是线上直接利用最近邻搜索的方法,根据 User 向量来为用户快速查找出最相关的 N 个视频。近邻搜索(NN)通常分为精确近邻搜索和近似搜索,精确搜索常见的方法是基于树结构的最近邻搜索,如经典的KD树算法,近似搜索也就是近似最近邻搜索(Approximate Nearest Neighbor,ANN),近似搜索搜索精确度降低但搜索时间减少,非常适合一些场景的工程应用,常见的近似近邻搜索方法有哈希散列和矢量量化,经典的**局部敏感哈希(Locality-Sensitive Hashing, LSH)**算法就是一种哈希散列方法,论文中可能用的就是这种方法,可以用现成的工具faiss来实现。 其实在亿级人脸识别的场景下也是用的类似LSH的方法来实现快速搜索,另外注意,因为LSH是在同一个空间里找近邻,而这里其实是求向量内积,所以这里需要归一化到单位向量,这两个操作就是完全等价的。这也是人脸Loss讨论里提到的特征归一化和权重归一化。 最近邻搜索会涉及 User Embedding 和 Video Embedding两个Embedding,我们分析一下它们是怎么得到的。
    • User Embedding:这个很好理解,其实就是把最后一层隐层输出作为用户向量;
    • 「Video Embedding」:这个比较难理解,也存在一些争议。从论文以及我查找的资料来看,「这里应该是用Softmax的权重来表示Video的Embedding向量的」,论文中有提到"The softmax layer outputs a multinomial distribution over the same 1M video classes with a dimension of 256 (which can be thought of as a separate output video embedding)" 。那么这个时候,就出现了两个Video Embedding,第一个是输入时的根据观看历史在session中的共现视频序列训练出的Embedding,第二个是NN侧输出,也就是最后一层Softmax的权重,作为Video Embedding的代表,只是比较有疑问的是,在线阶段的这个向量还会更新吗?我个人觉得在线Serving的时候就不更新了,毕竟是Item Embedding,没必要那么频繁地更新,而如此看论文图中的那条连向数据库的线,就是将Video Embedding向量存入数据库以方便线上得到实时用户向量进行近邻搜索。从这里来看,此处和矩阵分解又非常相似,模型最后可以得到User Embedding和Video Embedding,在训练时,Softmax概率最大的正好对应User Embedding和Video Embedding向量内积值最大的那个,所以预测的时候直接计算内积(可以转换等价于LSH),也就没必要再用Softmax具体计算概率了。

「排序阶段」

排序阶段的模型结构和召回阶段非常类似,可见下图:

  • impression video ID:喜欢的视频ID列表
  • watched video ID:观看的视频ID列表
  • user language:用户语言
  • video language:视频语言
  • time since last watch:上次观看的时序特征
  • #previous impressions:长期兴趣特征

与召回阶段不同的是,排序阶段需要处理计算的数据量仅仅是百数量级的,为了提高预测精度,排序阶段使用了更多精细的特征。除此之外,排序阶段本身就可以整合多源召回,上面提到的召回模型可能仅仅是一种召回策略,通常召回阶段的来源往往很多。

我们来看看排序阶段这篇论文又提到了哪些经典的工程经验。

  1. 「以观看时长作为优化目标,而非简单地用CTR或者播放率(Play Rate)」 具体而言,论文采用了每次曝光预期播放时间(expected watch time per impression)作为优化目标。确实,观看时长更能反映用户的真实兴趣,而类似CTR的指标则带有迷惑性,尤其是一些靠标题党或者视频封面来吸引人的视频,很容易诱导用户点击,但其内容不一定是用户真正喜欢的。 优化目标的设定其实非常重要,是算法模型的根本性问题,也是最初指明方向的一个过程,如果方向错了,后面再怎么努力也会和真正的目标南辕北辙。目标的确定,除了要考虑业务目标,也要考虑到商业目标,比如YouTube推荐的视频被用户看得时间越长,消费用户的注意力越多,就越有商业广告的价值。目标的设定和调整,也可以根据线上 A/B test进行微调。
  2. 「排序阶段重要的特征工程」 深度学习模型虽然能够减少一部分特征工程的工作,但有些原始数据根本就不能直接输入到前馈神经网络(Feedforward Neural Network,FNN),与图像、NLP不同的是,推荐系统对特征工程格外依赖,好的特征能够起到非常关键的作用,就比如召回阶段提到的Example Age。因此,在推荐场景下我们仍然需要花费很大一部分精力来做特征工程,将原始数据转换成有用的特征。其中「最主要的挑战是如何建模用户时序行为(temporal sequence of user actions),并且将这些行为和要排序的 item 相关联。」
    • YouTube 发现「最重要的特征是描述用户与商品本身或相似商品之间交互的特征」,这与 Facebook 在 2014 年提出 LR + GBDT 模型的论文 ( Practical Lessons from Predicting Clicks on Ads at Facebook )中得到的结论是一致的。 因此YouTube考虑了用户与视频频道(或主题)的关系,包括数量特征,即浏览该频道视频的数量,以及时间特征,即最近一次浏览该频道距离现在的时间。这两个连续特征的最大好处是具备非常强的泛化能力。另外除了这两个偏正向的特征,用户对于视频所在频道的一些 PV 但不点击的行为,即负反馈 Signal 同样非常重要。
    • 把召回阶段的信息,比如「推荐来源和所在来源的分数,传播到排序阶段」同样能取得很好的提升效果。
    • 为了产生很好的响应式推荐,通常会在推荐的时候引入“搅动(churn)”的功能(连续的请求不会返回相同的列表),而使用描述视频「历史曝光频率的特征,可以对“搅动”起到很好的辅助效果」,比如一个用户最近被推荐了某个视频,但没有观看它,接着模型将自然地在下一页加载时降级该曝光(impression)。
    • 「对类别特征进行Embedding向量化」 NN 更适合处理连续特征 , 因此稀疏的特别是高基数空间的离散特征(类别型特征可能的取值非常多,这种特征也称为高基数类别特征)需要 Embedding 到稠密的向量中。每个唯一的ID空间(比如user、item、vocabulary)都具有一个单独学到的Embedding空间,「一般来说Embedding的空间维度基本与唯一值(去重后的值)数目的对数接近」。这些Embedding通常会在DNN模型训练前构建,并形成简单的look-up table。 YouTube在对Video进行Embedding的时候,直接将大量长尾Video进行了断掉。**在基于点击曝光的频率排序后,只选择了TopN的视频进行Embedding,其余全部置为0向量。**其实这也是工程和算法的一个权衡(trade-off),截断掉大量长尾可以节省online serving中宝贵的内存资源,当然低频video的Embedding的准确性不佳是另一个“截断掉也不那么可惜”的理由。这个经验也是值得借鉴的,除了置为0向量,也可以替换为所属类别的Embedding,或者用一个固定的Embedding统一表示,类似NLP中用UNK(unknown)表示长尾词。 另外,和召回阶段一样,在输入时,「多值类别特征的Embedding先进行平均处理」「重要的是,相同ID空间的类别型特征,也共享着底层的Embedding」。例如,存着单个关于视频ID的全局Embedding,供许多不同的特征使用(曝光的视频ID,该用户观看的最近视频ID,作为推荐系统”种子”的视频ID等等)。尽管共享Embedding,但依然需要每个特征独自输入到网络中,以使得上面的层可以学到每个特征的特定表征(representation)。共享嵌入(sharing emdeddings)对于提升泛化、加速训练、及减小内存等相当重要。
    • 「对连续特征进行归一化」 众所周知, NN 对输入特征的尺度和分布都是非常敏感的,实际上 ,基本上除了 TreeBased 的模型( 比如 GBDT/RF),机器学习的大多算法都如此。归一化方法对收敛很关键,论文推荐了一种排序分位归一到[0, 1)区间的方法,即
    \overline x=\int_{-\infty}^xdf

    ,累计分位点,该积分与特征值的分位数的线性插值相近似。 除此之外,我们还把归一化后的

    \overline x

    的平方根

    \sqrt {\overline x}

    和平方

    \overline x^2

    作为网络输入,以期使网络能够更容易得到特征的次线性(sub-linear)和( super-linear )超线性函数。「通过引入特征的不同阶来引入了特征的非线性」

  3. 排序模型没有采用经典的Logistic Regression当做输出层,而是采用了Weighted Logistic Regression。 这样做的原因主要是为了融合优化目标,因为模型使用期望观看时长(expected watch time per impression)作为优化目标,如果简单使用LR就无法引入正样本的观看时长信息。因此采用weighted LR。 YouTube使用几率(odds)的方式来引入权重信息,相当于对模型sigmoid层作了一个小小的改动。 关于几率,我们在《Logistic回归》一文中其实已经有过详细介绍,而且Logistic回归本身就是从几率推导出来的,所以Logistic回归又称对数几率回归。我们简单回顾一下。 一个事件的几率,是指该事件发生的概率与该事件不发生的概率的比值,用公式表示:

取对数得到logit函数:

logit函数是广义线性的连接函数,因此令:

稍作转换,就得到了我们熟悉的sigmoid的函数:

这里将观看时长

作为正样本的权重,负样本权重指定为单位权重1,这样将权重引入几率后,由于正样本对预估的影响提升了

倍(权重是),而负样本没有变化(权重是1),因此对于指定样本,其引入权重后的几率是原来几率的

倍:

❝@张轩 @Shaohua Yang 严格的说,Weighted LR中的单个样本的weight,并不是让这个样本发生的概率变成了weight倍,而是让这个样本,对预估的影响(也就是loss)提升了weight倍。这种影响的反馈,通过推导可以看到,无论是在更新梯度时作用(乘以weight),还是直接将此样本在训练集里面扩充到weight个,大体上都是一样的效果(更新策略决定)。非weight的odds可以直接看成N+/N-,因为weighted的lr中,N+变成了weight倍,N-没变,还是1倍,所以直接可得后来的odds是之前odds的weight倍(严格的说应该是

倍)。 ❞ 由于在视频推荐场景中,对于整个视频资源库来说,用户打开一个视频的概率p往往是一个很小的值,因此上式可以简化为:

由于

就是用户打开视频的概率,

是观看时长,因此

就是用户观看某视频的期望时长,由此便能理解训练阶段的输出层为什么使用Weight LR了。 而对于线上Serving的输出层,则是使用

,它又是怎么来的呢? 其实还是几率(odds),我们把logit函数稍微做个转换变可得到:

因此在线上Serving中使用

做预测可以直接得到期望观看时长(expected watch time)的近似。 $既然预估的是期望观看时长,那为什么不用多分类或者是回归的模型来替代Weight LR? ❝@张相於 回归有一个问题在于值域是负无穷到正无穷,在视频推荐这样一个大量观看时间为0的数据场景,为了优化MSE,很可能会把观看时间预测为负值,而在其他数据场景下又可能预测为超大正值。逻辑回归在这方面的优势在于值域在0到1,对于数据兼容性比较好,尤其对于推荐这种小概率事件(rare event)的场景,相比回归会更加适合。而且odds和LR的公式是同源的,且值域也是非负的,更符合watch time的物理意义。 另一种方案是把观看时间离散化成k个bucket然后做多分类,但是多分类输出粒度不够细,不适合用来做排序。此外多分类的参数数量也比二分类多很多,同样的样本量下训练效果可能不如二分类效果好。❞

训练Weighted LR一般来说有两种办法:[^3] $这两种训练方法得到的结果有没有不同? ❝@张相於 这两种训练方法得到的结果是不一样的,比如要抽样10倍,对于第一种方法,就是把一条样本重复10倍,这样优化的过程中,每遇到一条这个样本,就会用梯度更新一下参数,然后用更新后的参数再去计算下一条样本上的梯度,如果逐步计算并更新梯度10次;但对于第二种方法,则是一次性更新了单条梯度乘以10这么多的梯度,是一种一次到位的做法。 直观一些来讲,第一种方法更像是给予一条样本10倍的关注,愿意花更多时间和精力来对待这条样本,是一种更细致的方法,第二种则比较粗暴了,不愿意花太多功夫,直接给你10倍权重。 @王喆 非常准确的理解,但我觉得这两种方法虽然形式上差别很大,但其实结果上差别不太大,因为重复抽样十次相当于算十次梯度再加起来,跟算一次梯度乘以10的结果一样。当然,重复采样逐次算梯度时还要考虑当前最优点移动的影响,但我觉得对于SGD来说影响不会特别大。 @张相於 当前参数点的移动确实会对结果造成影响,尤其是当把这10条样本打散分布在所有样本中时,也就是说不是连续使用这10条样本做优化时,影响应该会更大。但具体对最终效果的影响还是得试验后才知道。 @王喆 确实打散后的影响不好评估,结果确实会不同,还是跑一组AB test来的实在了。 ❞

  1. 将正样本按照weight做重复sampling,然后输入模型进行训练;
  2. 在训练的梯度下降过程中,通过改变梯度的weight来得到Weighted LR。

Airbnb

Real-time Personalization using Embeddings for Search Ranking at Airbnb

Airbnb这篇论文获得了KDD 2018的Best Paper,和上一小节介绍的YouTube论文类似,也是一篇充满工程实践经验的论文,其中Embedding在Airbnb业务中的整合和灵活应用,非常值得我们解读借鉴。

先来了解一下Airbnb的业务背景。

Airbnb是全球最大的短租平台,平台包含数百万种不同的房源,为了能够帮助用户找到自己心仪的短租屋,Airbnb主要通过两种方式来为用户推荐合适的住房。

一个是「搜索推荐」,即用户输入地点、日期、人数进行搜索,系统通过复杂的机器学习模型使用上百种信号对搜索结果中的房源进行排序,然后根据你的需求推荐你可能会感兴趣的房源:

另外一个是「相似房源推荐」,接入在房源详情页底部,会推荐和当前房源相似的房源,以图文滑动窗格的形式展示:

Airbnb表示,平台 99% 的房源预订来自于搜索推荐和相似房源推荐[^4],由此可见算法在业务中起到的关键作用。

Airbnb论文中将短租房源的item称为listing,其Embedding称为「房源嵌入(Listing Embedding)」,为了与论文保持一致,本文会延用该称谓,大家当作item理解即可。

接下来我们详细看看论文中提到的一些优秀的工程经验。

  1. 「将用户行为进行合理的Session切分,然后进行Embedding训练。」 这个方法在上一节Graph Embedding也提到过,即基于session 的用户行为(session-based)的Embedding。基于session用户行为学习到的Embedding,更容易捕获到用户的短期兴趣。session需要从用户行为历史中获取,因为用户历史行为并不是一次性产生的,用户每次的访问,其偏好都可能不同,因此,需要「对用户行为历史进行session切分,这样可以避免非相关序列的产生,也能够清洗噪声点和负反馈信号」。 Airbnb采用了click session数据对listing进行Embedding,其中click session指的是一个用户在一次搜索过程中,点击的listing的序列,这个序列需要满足两个条件,「一个是只有停留时间超过30s的listing page才被算作序列中的一个数据点(过滤噪声),二是如果用户超过30分钟没有动作,那么这个序列会断掉,不再是一个序列(避免非相关序列)。」
  1. 「使用负例采样的Skip-gram模型,并根据自己的业务特点进行了两项优化」。 这一部分是论文的一个核心,我们先从目标函数开始,逐步详细推导。 假设,给定从
N

个用户中获取的

S

个点击sessions的一个集合

S

,其中每个session

s = (l_1, ..., l_M) \in S

被定义成:一个关于该用户点击的

M

个listing ids连续序列,这里的session就是上面我们提到的已经切分后的session。对于给定该数据集,目标是为每个唯一的listing

l_i

学习一个d维Embedding表示:

v_{l_i} \in R^d

,以使相似的listing在该Embedding空间中更接近。论文中的Embedding维度d设为「32」,并且在论文末尾进行了试验,验证这样的设置可以平衡离线性能和在线搜索服务器内存中存储向量所需的空间,能够更好地进行实时相似度的计算。 学习Embedding使用的是Skip-gram模型,通过最大化搜索sessions的集合

S

的目标函数L来学习listing表示,L定义如下:

L= \sum\limits_{s \in S} \sum\limits_{l_i \in s} (\bbox[white,5px,border:2px dotted red]{ \sum\limits_{-m \leq j \leq m, i \neq 0} log P(l_{i+j} | l_i)})\tag{1}

从被点击的listing

l_i

的上下文邻居上观察一个listing

l_{i+j}

的概率

P(l_{i+j} \mid l_{i})

,使用Softmax定义:

P(l_{i+j}| l_i) = \frac{exp(v_{l_i}^T v_{l_{i+j}}')} {\sum\limits_{l=1}^{|V|} exp(v_{l_i}^T v_l')} \tag{2}

其中

v_l

v_l'

是关于listing

l

的输入和输出的向量表示,超参数m被定义成对于一个点击listing的forward looking和backward looking上下文长度,V被定义成在数据集中唯一listings的词汇表。 可以发现,上面的方法基本和Word2Vec或Item2Vec的思想一致,当然梯度更新也会受到Softmax性能的影响,因此采用负例采样进行优化。 我们会生成一个positive pairs

(l, c)

的集合

D_p

,其中

l

表示点击的listings,

c

表示它的上下文,然后从整个词典V中随机抽取n个listings来组成negative pairs

(l, c)

的集合

D_n

。优化的目标函数变为:

\arg \underset{\theta}{\max} \sum\limits_{(l,c) \in D_p} log \frac{1}{1+e^{-v_c'v_l}} + \sum\limits_{(l,c) \in D_n} log \frac{1}{1+e^{v_c'v_l}} \tag{3}

其中要学的参数

\theta

是:

v_l

v_c

l, c \in V

。优化通过随机梯度上升法(SGA)完成。 公式1、2、3都是最基本的用法,接下来我们来看看Airbnb如何根据自己的业务特点来优化Embedding学习的目标函数「优化一:使用最终预订的房源作为全局上下文 (Global Context)」 这一优化主要使用以用户预订房源(booked listing)作为结束的会话,加入预定不仅可以预测相邻的点击房源(clicked listings),还会预测最终预订的房源。当窗口滑动时,一些房源会进入和离开窗口,而预订的房源始终作为全局上下文(图中虚线)保留在其中,并用于更新中心房源(central listing)向量。

「优化二:适配聚集搜索的情况(优化负样本的选择)」 在线旅行预订网站的用户通常仅在他们的旅行目的地内进行搜索。因此,对于给定的中心房源,正相关的房源主要包括来自相同目的地的房源,而负相关房源主要包括来自不同目的地的房源,因为它们是从整个房源列表中随机抽样的,这种不平衡会导致在一个目的地内相似性不是最优的。为了解决这个问题,我们添加了一组从中央房源的目的地中抽样选择的随机负例样本集

D_{m_n}

「论文中将像目的地这种限定因素称之为市场market,其实应用时并不限定于目的地的地理位置信息,还可以拓展到其他限定因素,只要能够从这些限定因素中找到更合适的负例样本添加到目标函数即可。」 因此,经过上面的两个优化,「最终优化后的目标函数」变为:

\arg \underset{\theta}{\max} \sum\limits_{(l,c) \in D_p} log \frac{1}{1+e^{-v_c'v_l}} + \sum\limits_{(l,c) \in D_n} log \frac{1}{1+e^{v_c'v_l}} + log \frac{1}{1+ e^{-v_{l_b}' v_l}} + \sum\limits_{(l,m_n) \in D_{m_n}} log \frac{1}{1+e^{v_{m_n}'v_l}} \tag{4}

其中: 另外需要注意的一点,上面目标函数中,关于“优化一”的部分并没有加和符号

\sum

,这是为什么呢? 其实公式3和公式4都是公式1的红框部分,即对给定session的优化目标函数,并不是整体目标函数L。而对于每个预定session,只会有一个预定,它不像公式其他几个部分是样本对集合,自然就没有加和符号了。 利用上面所描述的优化目标函数,Airbnb使用超过 8 亿次的搜索点击会话,训练了 450 万个有效房源Embedding嵌入表示,从而获得了高质量的房源推荐和展示。

  • 要学习的参数
\theta

有:

v_l

v_c

l,c \in V

D_p

是一对正例样本对

(l, c)

,表示(中心房源,相关房源)元组,其向量在优化中会被互相推近

D_n

是一对负例样本对

(l, c)

,表示(中心房源,随机房源)元组,其向量在优化中会被互相推离

l_b

是最终被预订的房源,被视为全局上下文并被推向中心房源向量

D_{m_n}

是一对目的地维度的负例样本对

(l, m_n)

,代表(中央房源,来自同一目的地的随机列表)元组,其向量被推离彼此

  1. 「使用相近房源Embedding平均值来解决冷启动问题」。 每天在 Airbnb 上都有新的房源提供。这些房源在新建时还不在训练数据集中,所以没有嵌入信息。为给新房源创建Embedding,Airbnb的方法是找到 3 个地理位置最接近、房源类别和价格区间相同的已存在的房源,并计算这些房源嵌入的向量平均值来作为新房源的嵌入值
  2. 「Embedding的评估经验」 对于Embedding质量的评估,Airbnb使用了很多方法,主要的思想还是通过Embedding相似度的方法来进行的,论文中提到了以下几种方式:
    • 评估地理位置相似性是否被包含,对Embedding进行了 k 均值聚类 (k-means clustering)。下面的图显示了美国加州产生的 100 个聚类,确定了来自近似位置的房源基本能够聚集在一起
    • 评估房源特征相似性是否被包含。论文评估了不同类型(整套房源,独立房间,共享房间)和价格范围的房源之间的平均余弦相似性 (cosine similarity) ,并确认相同类型和价格范围的房源之间的余弦相似性远高于不同类型和不同价格的房源之间的相似性。因此我们可以得出结论,房源特征也被很好的包括在训练好的Embedding中了。
    • 评估一些隐式房源特征。部分类型的房源特征(例如建筑风格,样式和感觉)很难提取为房源特征的形式。为了评估这些特性并能够在Embedding空间中进行快速简便的探索,Airbnb内部开发了一个相似性探索工具 (Similarity Exploration Tool),下图的示例表明,通过Embedding能够找到相同独特建筑风格的相似房源,包括船屋、树屋、城堡等,说明即使我们不利用图片信息,也能从用户的click session中挖掘出相似风格的房源。
    • 论文还提出一种离线评估方法,测试通过用户最近的点击来推荐的房源,有多大可能最终会产生预订。 更具体地,假设已经获得了最近点击的房源和需要排序的房源候选列表,其中包括用户最终预订的房源;通过计算点击房源和候选房源在Embedding空间的余弦相似度,对候选房源进行排序,并观察最终被预订的房源在排序中的位置。 下图显示了一个此类评估的结果,搜索中的房源根据Embedding空间的相似性进行了重新排序,并且最终被预订房源的排序是按照每次预定前的点击的平均值来计算,追溯到预定前的 17 次点击。同时,下图也对比了几种Embedding学习的优化方式,可以看到优化起到的效果。
  3. 「利用User-Type和Listing-Type Embedding来解决数据稀疏问题」 上面学习到的Listing Embedding是使用clicked sessions进行训练的,它们更适合短期(short-term)、session内(in-session)的个性化的需求,虽然这个Embedding能够很好地给用户展示与在搜索session期间点击的listing相似的listings,但并不包含用户的长期兴趣信息。基于用户长期兴趣的信息对于个性化搜索来说非常有用,比如用户当前session搜索的是洛杉矶的短租屋,但是用户之前在纽约、伦敦住过的短租屋也代表着用户的兴趣偏好。那如何学习含有长期兴趣信息的Embedding呢? Airbnb首先想到的是使用了booking session序列,每个用户按预定(booking)的时间顺序排列的一个listings序列构成booking session
s_b = (l_{b1}, ..., l_{b_M})

,然而想要从booking session学习到相应的Embedding,会遇到诸多挑战: 1)booking sessions数据集比click sessions数据集要小很多,因为预定本身是低频事件; 2)许多用户在过去只预定单个listing,我们不能从长度只有1的session中学习Embedding; 3)为了上下文信息中的任意实体学习一个有意义的Embedding,至少需要该实体出现5-10次,然而在平台中的许多listing_ids会低于5-10次。 4)最后,由同用户的两个连续预定可能会有很长时间的间隔,这时候,用户偏好( 比如:价格点)可能会随很多因素发生了变化,比如职业发展影响等。 5)必须满足供需双方,给租户的最优化推荐还需要考虑房东的对租客的偏好 Airbnb其实是一个双边市场平台,它的房源大都来自于个人的民宿,所以平台也会考虑房东对租客的偏好,和预定酒店等类似业务不同,Airbnb房源的房东是可以拒绝租客的,大致的业务流程如下图所示:

从上图可以看出,booking过程其实存在房东的显式反馈信息(接受或拒绝租客的预定),这个信息也体现出房东的偏好,对房东来说,拒绝预定的部分原因可能是:租客较差的星级评分习惯、用户资料不完整或空白、没有资料头像等等。所以,为了提高租客的预定成功率,平台不仅要考虑租客的租住偏好,还要考虑房东对租客的一些基本要求和偏好,尽量避免给用户推荐的房源被房东拒绝。 面对这些挑战,如何才能学习到有意义的Embedding呢? 首先,针对第四个挑战,也就是为了为了捕获用户随时间变化的偏好信息,Airbnb除了Listing Embedding,又加了一个User Embedding的学习。而对于数据稀疏的问题,Airbnb使用了一种非常巧妙的方法,即「基于某些属性规则做相似user和相似listing的聚合,这种方法是通过Type级别来学习Embedding而非ID级别。」 我们接下来看具体的实现过程。 「第一步,通过基于规则(rule-based)的方式定义的属性类型分桶(buckets),将ID类映射到type类。」 「第二步,用type类数据生成booking session序列。」 如何生成同type类的booking session序列呢? 一种直观方法是直接把user_type当作原来的user_id,生成一个由listing_type组成的booking session。这种方法能虽然够解决数据稀疏性的问题,却无法直接得到User-Type Embedding。**为了学习在相同向量空间中的user_type和listing_type的Embeddings,Airbnb的做法是将user_type插入到booking sessions中,形成一个(user_type, listing_type)组成的元组序列,这样就可以让user_type和listing_type的在session中的相对位置保持一致了。**如下图所示,user_type 和 listing_type相对位置保持一致(不同颜色区分),根据中心Item的类型来选择对应上下文的类型:

具体来讲,对于给定user_id按时间排序的booking session

s_b = (l_{b1}, ..., l_{b_M})

,用(user_type, listing_type)组成的元组替换掉原来的session中的listing_id,因此这个booking session序列就变成了:

s_b = ((u_{type_1}, l_{type_1}),(u_{type_2}, l_{type_2}), ..., (u_{type_M},l_{type_M})) \in S_b

这里

l_{type_M}

指的就是预定的listing

l_{b_M}

对应的listing_type,

u_{type_M}

指的是该user在预定listing

l_{b_M}

时的user_type,「由于某一user的user_type会随着时间变化,所以每个session虽然由相同user_id的bookings组成,但对于单个user_id来说,他们的user_type在序列中的不同位置则不一定相同。」 「第三步,设计优化目标函数,进行Embedding学习。」 有了type类的booking session序列,训练Embedding的方式基本和上面的一致,也是采用负例采样的方式。当然,此处Airbnb也考虑了根据自身业务对负例采样进行进一步的优化工作。由于定义中的booking sessions几乎包含了来自不同markets的listings,因此这里没有必要从相同market中抽样额外的负样本作为booked listing。但为了解决上面提到的第五个挑战,也就是尽量避免推荐被拒,在此处的目标函数中引入了房东拒绝订单(rejection)的显式负例样本。如下图所示,对于同一用户,通过一个正号标记预定成功的listing,通过负号标记被房东拒单的listing。

当在对于另一个listing的成功预定(通过一个正号标记)之后主人拒绝(通过一个负号-标记)。 其实在User Type映射表中,就包含了租客的一些个人信息以及行为信息,通过引入房东的拒单信息,就可以捕捉到房东(所对应的房源)对租客的偏好,这样学习到的Embedding在进行相似推荐时便可以减小拒绝率,最大化预订成功的概率。 我们对rejections看成是显式负样本,以如下方式公式化。除了集合

D_{booking}

D_{neg}

,我们会生成一个集合

D_{rej}

,它由涉及到rejection事件的user_type和listing_type的

pairs(u_t, l_t)

组成。如图5b所示,我们特别关注,对于同一用户,当在对于另一个listing的成功预定(通过一个正号标记)之后主人拒绝(通过一个负号-标记)。 于是,最后改进的新目标函数为: User Type:

\arg \underset{\theta}{\max} \sum_{(u_t,c) \in D_{book}} log \frac{1} {1+e^{-v_c'v_{u_t}}} + \sum_{(u_t,c) \in D_{neg}} log \frac{1} {1 + e^{v_c'v_{u_t}}} + \sum_{(u_t,l_t) \in D_{reject}} log \frac{1} {1+e^{v_{l_t}' v_{u_t}}}

Listing Type:

\arg \underset{\theta}{\max} \sum_{(l_t,c) \in D_{book}} log \frac{1} {1+e^{-v_c'v_{l_t}}} + \sum_{(l_t,c) \in D_{neg}} log \frac{1} {1 + e^{v_c'v_{l_t}}} + \sum_{(l_t,u_t) \in D_{reject}} log \frac{1}{1+e^{v_{u_t}' v_{l_t}}}

其中,

u_t

为user_type,

l_t

为listing_type,

D_{book}

为预定集合,

D_{neg}

为随机负例集合,

D_{reject}

为拒单集合,

D_{reject}

由涉及拒单行为的user_type和listing_type的pairs

(u_t, l_t)

组成。 「第四步,基于cosine相似度进行推荐(Embedding评估)。」 对于所有user_types和listing_types所学到的Embeddings,我们可以根据用户当前的User-Type Embedding和Listing-type Embedding,基于cosine相似度给用户推荐最相关的listings。

上表展示的type信息以及对应的cosine相似度能够看出: 根据某用户对应的typeID user_type = SF_lg1_dt1_fp1_pp1_nb3_ppn5_ppg5_c4_nr3_l5s3_g5s3来看,该用户通常会预定高质量、宽敞、好评率高、并且在美国有多个不同listing_type的listing。可以观察到,最匹配这些用户偏好listing_type(例如,整租,好评多,大于平均价等),具有较高cosine相似度;而其它不太匹配用户偏好的listing_type(例如:空间少,低价,好评少),则具有较低cosine相似度。 「最后,我们来总结一下Airbnb是如何解决上面提到的各种问题的。」 首先,通过clicked session学习到的Embedding只包含短期偏好信息,不包含用户的长期偏好。所以为了学习有着用户长期偏好信息Embedding,Airbnb的方案是利用booking session数据。 长期信息有什么用呢?仅仅是为了通过相似度推荐吗?其实不是,我们下面会讲到,其实Airbnb并没有直接根据Embedding相似度排序作为搜索推荐结果,而是在后面还有一个排序模型,Embedding相似度其实是作为这个排序模型的一个特征来用的。 另外要想训练booking session数据其实并不容易,因为booking session有着严重数据稀疏问题(上述挑战1、2、3),对于长度为1的booking session(挑战2),Airbnb的做法很可能是直接丢弃这一类的数据。而对于数据集小以及训练实体出现次数不多的情况(挑战2、3),Airbnb的做法则是将ID类Item映射聚合到Type类,这样训练实体出现次数会变多,之前很可能完全没有关系的ID类Item则被看做同一类型的Type类Item,重复出现的机会变多了,对所有用户而言,booking session中的交集和关联也变多了。因此原本稀疏不可学习Embedding的数据变成了可用的数据。 除此之外,用户兴趣也会随着时间变化(挑战4),为了获取这部分信息,Airbnb特意学习了一个与Listing-Type Embedding在同一空间的User-Type Embedding。而为了在Embedding中体现房东对租客的偏好,优化预定成功的概率(挑战5),Airbnb在User-Type Embedding和Listing-Type Embedding的优化目标函数中加入了房东拒单的显示负样本。 到此,以上提到的问题基本都得到了解决。

  • Listing Type 映射

所有listing_id都可以根据自己的属性通过上面的分桶映射成一个listing_type,分桶设计是以数据驱动的,要尽可能地覆盖所有的类型。 例如,一个来自US的Entire Home listing(lt1),它是一个二人间(c2),1床(b1),一个卧室(bd2),1个浴室(bt2),每晚平均价格为60.8美元(pn3),每晚每个客人的平均价格为29.3美元(pg3),5个评价(r3),所有均5星好评(5s4),100%的新客接受率(nu3)。因此该listing根据上表规则可以映射为:listing_type = US_lt1_pn3_pg3_r3_5s4_c2_b1_bd2_bt2_nu3「从listing_id到listing_type的映射是一个多对一的映射,这意味着许多listings可能会被映射到相同的listing_type,因此这把这个过程称之为聚合。」

  • User Type 映射

**User-Type Embedding是与Listing-Type Embedding在相同的向量空间学习的,学习User-Type Embedding主要为了解释用户随时间变化的偏好。**User属性的映射方式同Listing类似, 不过数据除了利用User之前预定记录留下的一些行为信息,同时引入了User本身的一些属性信息,利用前5行的映射一定程度上可以缓解新用户的冷启动问题。 例如,对于一个用户,他来自San Francisco(SF),带有MacBook笔记本(dt1),说英文(lg1),具有用户照片资料(pp1),83.4%平均5星率(l5s3),他在过去有3个预订(nb1),其中关于订单(booked listings)的平均消费统计为:每晚平均价格(Price Per Night)为52.52美元 (ppn2),每晚单客户平均价格(Price Per Night Per Guest)为31.85美元(ppg3),预定房源的平均房间数(Capacity)是2.33(c2),平均浏览数(Reviews,此处存疑,看数量感觉这里的Reviews和Listing中的Reviews可能不一样)是 8.24(nr3)、5星好评率(Listing 5 star rating)是76.1%(l5s3)。因此该用户根据上表规则可以映射为:user_type=SF_lg1_dt1_fp1_pp1_nb1_ppn2_ppg3_c2_nr3_l5s3_g5s3

  1. 「Embedding的最终用法」 我们上面提到过Airbnb的推荐场景有两个,一个是房源详情页的相似房源推荐,另外一个是在个性化的搜索推荐结果。我们分别来看这两个推荐应用场景下是如何应用学习到的Embedding的。 「相似房源推荐」 相似房源推荐用的应该是Clicked Session学习到的Listing Embedding,用法也非常简单,就是通过K近邻找到与给定房源(点击的是该房源的详情页)最相似的K个房源,Airbnb中的K设定是12,并且对候选房源做了一定的筛选,比如只计算与给定房源在同一城市的房源,还有就是如果用户设置了入住和退房日期,则筛选在这个时段内的可预订的房源。 Airbnb通过A/B测试显示,基于Embedding的解决方案使「相似房源的点击率增加了21%」,最终「通过相似房源推荐场景产生的预订增加了 4.9%」「基于Embedding的实时个性化搜索」 Airbnb在搜索排序中用了一个排序模型,从论文描述来看应该就是我们之前文章讲到过的「LambdaMART」,还给出一个开源实现(GitHub地址),此处对于Embedding应用,更多是在特征工程上。 根据前面得到的Listing Embeddings和User-Type Embedding、Listing-Type Embeddings生成的Feature如下表所示:

上表中第一区域内(前6个特征)的都是短期特征(short-term feature),首先利用kafka来实时收集和维护过去两周内的用户行为数据,然后通过对应的Listing Embedding计算相关的相似度,下面是这些特征描述中对应符号的具体含义:

其他应该都很好理解,这里重点说一下

H_s

,它表示用户在过去2周内跳过的房源 ID,这里将跳过的房源定义为排序较靠前的房源,但用户跳过了此房源并点击了排序较靠后的房源。 来看看这六个短期特征具体是如何计算的,用户在搜索时会返回候选房源(无序),对于每个候选房源

l_i

,计算其对应Embedding与用户两周内的各个行为(上面的六个

H_*

)所产生的房源的平均Embedding相似度,以EmbClickSim为例,计算公式为:

EmbClickSim(l_i,H_c)=\underset {m \in M} \max cos(v_{l_i}, \sum_\limits{l_h \in m, l_h \in H_c} v_{l_h})

这里关键是对于两周内点击的房源(listing)所对应Embedding的处理(得先处理成一个,然后才可以和候选listing做相似度计算),一般是直接取平均就行了,但Airbnb的做法考虑了房源所在地的因素,即是先对用户两周内点击的房源按照其所在地(market)分为

M

个market集合,然后分别对每个market集合

m

内的所有Listing Embedding向量求平均并与候选房源

l_i

计算cosine相似度,最后从M个cosine相似度中取最大的那个值作为最终的特征值。 剩余的五个

H_*

计算方法与上面一样。 第二个区域的特征也是一个短期特征,不过它是单独增加的一个重要特征,即计算候选房源

l_i

与最后一次long-clicked对应房源的Embedding相似度,通过这个特征可以得到用户“实时”在意的反馈信息,公式如下:

EmbLastLongClickSim(l_i, H_{lc})=cos(v_{l_i}, v_{l_{last}})

第三个区域就是Type类Embedding特征了,他们属于长期特征(long-term feature),首先根据候选房源

l_i

查找到对应的listing type

l_t

以及user type

u_t

,然后计算二者对应的Embedding向量的cosine相似度,公式如下:

UserTypeListingTypeSim(u_t, l_t) = cos(v_{u_t},v_{l_t})

论文给出了以上特征在模型中的特征重要度,见下表:

上面的特征由多种Embedding计算得来,所以包含了短期session内的(In-session)信息(例如,喜欢(liked)、点击(clicks)、与主人的联系(host contacts)等),也包含用户的长期兴趣信息,Airbnb就是借助于这些特征来实现「实时个性化搜索推荐」的, 目标是给用户展示与In-session信号相似的多个listings。Airbnb称这种方法为「Real-Time Personalization」「In-session Personalization」,这也是论文标题中Real-Time Personalization所表达的意思。 论文最后给出了模型效果的评估:

  1. 搜索Query Embedding的应用 另外,Airbnb对搜索的Query也进行了Embedding,和普通搜索引擎的Embedding不太相同的是,这里的Embedding不是用自然语言中的语料库去训练的,而是用Search Session作为关系训练数据,训练方式更类似于Item2Vec,Airbnb中Queue Embedding的一个很重要的作用是捕获用户模糊查询与相关目的地的关联,这样做的好处是可以使搜索结果不再仅仅是简单地进行关键字匹配,而是通过更深层次的语义和关系来找到关联信息。比如下图所示的使用Query Embedding之前和之后的两个示例(Airbnb非常人性化地在搜索栏的添加了自动补全,通过算法去“猜想”用户的真实目的,大大提高了用户的检索体验):

为了让Embedding包含这种模糊查询和相关目的地的关联信息,利用Search Session构建训练序列的时候,就会非常注重这种关系的序列构成,所以这个Search Session序列构建的过程其实是通过用户查询关键词Query以及点击目的地来构建的,Airbnb将目的地用geo_ID形式来表示,所以Search Session就成为如下图所示的样子:

有了Search Session序列,训练方式基本和之前谈到的方法差不多,这里不再赘述。

Alibaba

Learning and Transferring IDs Representation in E-commerce

这篇论文也是发表在KDD 2018上的,论文的重点是介绍了一种ID类特征的表示方法。该方法基于item2vec方式,同时考虑了不同ID类特征之间的连接结构,在盒马鲜生app上进行了迁移应用且取得了不错的效果。

「ID类特征是一种比较特殊的数据,通常可能只是一个长整型数。表面上来看并不带任何信息,但实际上通过行为的汇总其承载了丰富的信息,所以ID类特征的处理其实是特征工程中非常重要的一部分」。ID类最初的处理方法是使用One-Hot编码,这种方式有两个缺点:一个是高维带来的稀疏问题,模型具有足够置信度的话需要的样本量指数级增加;另一个是无法表示ID之间的关系,比如通过One-Hot编码无法度量两个ID对应item的相似性。[^5]

因此,Embedding理所当然的成了替代One-Hot的最佳方案。

接下来,我们来看看这篇文论中比较tricks的亮点经验。

  1. 使用 log-uniform 负例采样 这篇论文同样采用负例采样的Skip-gram模型进行Embedding的学习,不过采样方法与Word2Vec中的不太一样,用的是对数均匀负采样(Log-uniform Negative-sampling)。log-uniform 负例采样基于Zipfian分布,物理学和社会科学中许多类型的数据都可近似看作是Zipfian分布。 我们来看一下log-uniform负例采样的具体过程。 首先,为了进一步加速负采样过程,先将D个物品按照其出现的频率进行降序排序,那么排名越靠前的物品,其出现的频率越高。 每个物品采样到的概率如下:
p(index)=\frac{\log(index+2)-\log(index+1)}{\log(D+1)}

由上式可以看出,由于分母都是一样的,分子依次为

\log(2/1), \log(3/2),\cdots,\log(D+1/D)

,是顺次减小的,同时求和为1。那么排名越靠前即出现频率越高的商品,被采样到的概率是越大的。 那么,该分布的累积分布函数为:

\begin{align}F(x)&=p(x \le index)\\ &= \sum_{i=0}^{index} \frac{\log(i+2)-\log(i+1)}{\log(D+1)} \\&=\frac{\log(index+2)}{\log(D+1)} \end{align}

F(x)=r

r

是随机产生的一个

(0,1]

之间的随机数,就可以通过下面的转换快速得到对应的index,也就可以快速采样到index对应的负例了:

index = [(D+1)^r] - 2

可以看到,采样公式非常简单,利用这个近似,对数均匀负例采样具有很高的计算效率。

  1. 多种ID分层联合嵌入模型(Jointly Embedding model) 像电商类的场景数据会有很多ID类特征,大的分类有物品ID和用户ID,特别的,物品ID除了本身的ID特征,还有一些属性ID,如下: 这么多的ID类型,又通过Item ID有结构关联(如下图),如何在训练Item Embedding的时候,将这些信息利用起来呢?

论文作者提出一种层级Embedding模型来联合训练得到item ID和attribute ID的低维表示。这个联合Embedding的模型结构如下图所示:

其中,item ID是核心的交互单元,若item ID之间存在共现关系,则表明其对应attribute ID也存在共现关系。 假设有K种类型的ID特征,这里定义

{\rm item}_i

的ID组

{\rm IDs}({\rm item}_i)

如下:

{\rm IDs}({\rm item}_i ) =[{\rm id}_1({\rm item}_i), {\rm id}_2({\rm item}_i), \cdots, {\rm id}_k ({\rm item_i}), . . . , {\rm id}_K ({\rm item_i})]

其中,

{\rm id}_1({\rm item}_i)

表示item ID,

{\rm id}_2({\rm item}_i)

表示product ID,

{\rm id}_3({\rm item}_i)

表示store ID,其他类似(论文中用了6个attribute ID,剩下的按顺序应该是Brand ID, cate-level1 ID,cate-level2 ID,cate-level3 ID)。 因此,Skip-gram模型的概率计算则可写成如下方式:

\begin{align} &p({\rm IDs}({\rm item}_j ) \mid{\rm IDs}({\rm item}_i ) \\= &\sigma \left( \sum_{k=1}^K(w_{jk} e'_{jk})^T(w_{ik} e_{ik})\right) \prod_{s=1}^S \left( -\sum_{k=1}^K(w_{sk} e'_{sk})^T(w_{ik} e_{ik})\right) \end{align}

其中,

e'_{\cdot k}\in {\rm E}'_k(\subset \R^{m_k \times D_k})

以及

e_{\cdot k}\in {\rm E}_k(\subset \R^{m_k \times D_k})

分别为第k个类型ID的上下文矩阵和目标矩阵。

m_k

是第

k

类ID的Embedding向量维度,

D_k

是第

k

类ID构成的的字典大小,需要注意的是,不同类型的IDs,Embedding的维度也可以不同,也就是说不同的ID类其实是映射到了不同的语义空间中。

w_{ik}

{\rm id}_k({\rm item}_i)

的权重,定义如下:

w_{ik}=\frac{1}{\sum_{j=1}^D I({\rm id}_k({\rm item}_i)={\rm id}_k({\rm item}_j))}

举例来理解,比如

w_{i1}=1

,因为

{\rm item}_i

是唯一的,而如果

{\rm item_i}

对应的产品有10种item的话,那么

w_{i2}=\frac{1}{10}

。 另外,item ID和attribute ID间存在着结构联系,这意味着可对模型增加限制。「两个item ID的Embedding接近,除了本身的共现性,还可能跟它们拥有共同的product ID、 store ID、brand ID或 category ID等有关,因此,attribute ID也应该吸收item ID包含的信息,以商店id为例,对于某一store ID的Embedding应该是该商店出售的所有Item ID的一个恰当的概括。」 所以可通过attribute ID来预测item ID。形式化表达如下:

p({\rm item}_i|{\rm IDs}({\rm item}_i)) = \sigma \big( \sum_{k=2}^K w_{ik}e^T_{i1} M_k e_{ik} \big)

其中,

M_k \in \R^{m_1 \times m_k}(k=2\cdots K)

是为了使

e_{i1}

变换到

e_{ik}

相同维度的矩阵,因为不同类型attribute ID的Embedding长度可能不一样,无法直接相乘。 然后最大化:

J = \frac{1}{N} \sum_{n=1}^N \big( \sum_{-C\leq j \leq C}^{1 \leq n+j \leq N,j\neq 0} \log p({\rm IDs}({\rm item}_{n+j}) | {\rm IDs} ({\rm item}_n)) + \alpha \log p({\rm item}_n| {\rm IDs}({\rm item}_n)) - \beta \sum_{k=1}^K \Vert M_k \Vert_2 \big)

其中,

\alpha

{\rm IDs}

之间的约束强度系数,

\beta

是变换矩阵L2正则化系数。 「通过这种方法共同训练可以加将Item ID 和attribute ID 嵌入到同一个语义空间」,这样便于在实践应用中使用或迁移这些向量化表示。由于Item id及其attribute ID比较稳定,所以联合嵌入模型所学的item ID和attribute ID的Embedding通常是每周更新一次。 用户Embedding则使用简单平均法来计算,不使用RNN结构是因为盒马鲜生的场景用户兴趣变化非常快,采用RNN结构训练会明显耗时,所以采用了「直接对近期交互的T个item ID平均的方式」来快速更新用户的Embedding:

{\rm Embedding}(u) = \frac{1}{T} \sum_{t=1}^T e_t

其中,

e_t

{\rm item}_t

的Embedding,因为用户兴趣变化要更快,用户的Embedding是要尽量实时更新的。 最后罗列一下模型训练时的一些超参设置:

  • 所有参数先用 Xavier 方式进行初始化,用Adam进行参数更新;
  • 滑动窗口C = 4;
  • 负例样本数S = 2;
  • 1个item ID和其余6个attribute ID的Embedding维度分别为[100, 100, 10, 20, 10, 10, 20];
  • α = 1.0,β = 0.1;
  • batch_size = 128 以及 epochs = 5 。
  • 「物品ID及其属性ID」:物品item ID是电商领域的核心,每一个物品有对应的属性ID,比如产品product ID,店铺store ID,品牌brand ID,品类category ID等,品类ID又可能分级产生多种品类ID,如分三级的话有一级品类、二级品类和三级品类。举例,比如“iPhone X”手机,在不同的店铺卖,共用相同的product ID,但不同店铺卖的iPhone X手机,它们的item ID是不同的,当然,其对应的store ID也不同。又因为是同一款产品,所以其对应的brand ID,category ID也会基本一致。
  • 「用户ID」:用户身份可以通过ID进行识别,这里的ID可以有多种形式,如cookie、设备IMEI、uuid、用户昵称等等。

  1. Embedding的迁移应用
    • 「计算item相似度,做相似物品推荐」 这是一个很常见的做法,就不多说了,我们提一下用Embedding做相似推荐,相比于用传统item-based CF的优势。item-based CF方法需要明确的user-item关系,但很大比例的交互数据并没有用户识别信息。而skip-gram模型基于会话的,可以利用全量的数据。因此基于Embedding的方法比基于item-based的CF方法要更好一些,实验结果也说明其有更高的召回。
    • 「迁移到新的item」(Seen Items to Unseen Items) 推荐系统中冷启动是个很大的问题,新物品由于没有交互信息,此时item-based CF方法也失效。一般可通过基于image或者text的辅助信息来进行推荐。这里解决冷启动的方法出发点是和新物品连接的attribute ID通常有历史记录(比如产品已经在其他店铺卖过或者该店铺卖过其他产品,也就是说很多attribute ID已经确定了),因此可以通过这些attribute ID的Embedding来预测得到新物品item ID的Embedding,这个时候我们就可以利用上面提到的公式了,最大化条件概率,即对数似然:
    \max p({\rm item}_i|{\rm IDs}({\rm item}_i)) = \sigma \big( \sum_{k=2}^K w_{ik}e^T_{i1} M_k e_{ik} \big)

    因为sigmoid函数是一个单调递增函数,因此有如下近似关系:

    \begin{align} &p({\rm item}_i|{\rm IDs}({\rm item}_i)) \\= &\sigma \big( \sum_{k=2}^K w_{ik}e^T_{i1} M_k e_{ik} \big)\\ \varpropto & \sum_{k=2}^K w_{ik}e^T_{i1} M_k e_{ik} \\ =& e^T_{i1}\big( \sum_{k=2}^K w_{ik} M_k e_{ik} \big) \end{align}

    最大化

    p({\rm item}_i|{\rm IDs}({\rm item}_i))

    就是使概率尽可能趋近于1,也就要求

    \sigma(x)

    中的

    x

    尽可能地大,即:

    \begin{align} &p({\rm item}_i|{\rm IDs}({\rm item}_i)) \to 1 \\ \Rightarrow& e^T_{i1}\big( \sum_{k=2}^K w_{ik} M_k e_{ik} \big) \;\; \text{is relatively large} \\ \Rightarrow &e_{i1} \approx \sum_{k=2}^K w_{ik} e_{ik}^T M_k ^T \end{align}

    最后这一步需要理解一下(不一定正确),如果我们把括号里面的向量表示成另一个向量e的话,当两个向量长度固定的时候,什么时候内积取得最大值?是二者同向且共线的时候,那么

    e_{i1}

    e

    应该是线性关系,即使

    e

    是真实的

    e_{i1}

    的n倍的话,在计算该物品与其他物品的相似度的时候,是不会产生影响的,因此

    e_{i1}

    可近似用

    e

    来代替。

    • 「迁移到新的领域」 盒马鲜生作为一个新平台,很大比例的用户是新的,因此做个性化的推荐较为困难。文章提出了一种将淘宝(source domain)用户的兴趣迁移到盒马(target domain)上的方法,其实这得益于盒马鲜生有很大一部分用户直接使用淘宝账户登录,可以共享用户信息。 迁移过程如下所示:

    1)设

    U^s

    U^t

    分别表示淘宝和盒马的用户,找到可以迁移的用户(即取交集)

    U^i = U^s \bigcap U^t

    ; 2)从淘宝获得

    U^s

    中用户计算好的Embedding; 3)基于嵌入向量的相似度,

    U^i

    中的用户使用k-means聚类为1000个组; 4)对于每个组,选择出 topN 个最热门的盒马item作为该组的候选集; 5)对于盒马上的一个新用户,如果它在淘宝上有交互记录,那么就取得他在淘宝上对应的用户向量,并计算该向量所属的用户组,然后将这个组所对应的候选集推荐给新用户; 5)基于得到的类别,将经过筛选和排序后的该类别的候选集中物品中推荐给用户。

    • 「迁移到不同任务」 另外一个迁移试验是将学习好的store ID对应的Embedding用于预测店铺销量的任务。销售预测在电商决策中非常重要,可以帮助管理人力分配,现金流,资源等,例如优化商品配送。在论文中的任务是预测每个商店第二天每30分钟的配送需求,这个预测结果可以指导配送人员数量的预先分配。 论文中使用输入来分析对比Embedding带来的效果,即仅使用历史数据、历史数据加店铺ID的One-Hot编码以及历史数据加Embedding,其中历史数据用的是过去7天每三十分钟的店铺配送量。

总结

本文解读了三篇干货满满的Embedding应用实践的论文,总结他们在不同业务场景下一些优秀的Embedding应用实践经验,论文很多地方都充满了工程化实践的亮点。

文章最后,我们简单总结一下Embedding的应用经验

  1. 学习Embedding大部分使用基于负例采样的Skip-gram模型。
  2. Embedding的训练方式通常有两种,一种是预训练,是非端到端的方式,其中比较重要的过程是构造序列,基于Session切分的序列构造方法是一个非常好的方案。另外一种是初始化到DNN里,通过端到端的方式训练Embedding,可以看做是矩阵分解的泛化。
  3. 构造序列训练Embedding的时候,要分析业务特点,有些序列是发现item的共现关系,而有些则有一定的顺序关系,设计窗口采样样本对的时候,一定要确定是否可以进行数据穿越。
  4. Embedding可以通过拼接的方式来融合其他信息,共同作为DNN的输入。
  5. Embedding可以通过算术平均、加权平均等方式来得到上一级或者是相关的实体的Embedding向量概括表示,用户Embedding经常通过这样的方式来求得。
  6. Embedding非常便于计算相似度来衡量item间的相似性,也可以通过最近邻搜索在Serving阶段快速获得召回候选集,另外,在搜索场景中也有类似应用,可以有目的地训练Embedding,通过Query来找到更加隐性地关联内容。
  7. 除了正常的负例采样,还可以根据场景业务对Embedding的目标函数进行优化,比如添加在某些情景下的正例或负例,使学习好的Embedding在该情景下与目标相互靠近或远离。
  8. Embedding的冷启动是一个非常值得研究的问题,通常的做法有:对于端到端训练Embedding,可引入其他内容信息,这样在没有行为的时候,也可以训练Embedding。另外一个常用方法是将一定数量相似物品的Embedding向量的平均值来作为新物品的Embedding。
  9. Embedding评估更多的是通过物品相似性来判断,包括显式的特性以及隐式特性,可以通过不同的角度来对比测试。
  10. 行为较少的稀疏数据,可以通过规则映射聚合到上一级的类别ID,从而构造出更多的共现关系来学习Embedding。
  11. Embedding也可以作为其他模型的特征,方式大概有以下几种:直接输入,拼接聚合,相似度值等。
  12. ID类特征的行为汇总可以挖掘出丰富的信息,若业务场景下的物品ID还有众多属性ID关联,则可以将二者融合训练,互相补充彼此的信息,这样可以得到更好的效果。另外属性ID的Embedding可以一定程度上解决物品Embedding的冷启动问题。
  13. 预训练的Embedding,也可以迁移到其他场景实现其价值。

参考资料

[^1]: 黄昕等. 推荐系统与深度学习. 清华大学出版社 [^2]: 王喆. YouTube深度学习推荐系统的十大工程问题. 知乎(https://zhuanlan.zhihu.com/p/52504407) [^3]: 王喆. 揭开YouTube深度推荐系统模型Serving之谜. 知乎(https://zhuanlan.zhihu.com/p/61827629) [^4]: Airbnb技术团队. Airbnb爱彼迎房源排序中的嵌入(Embedding)技术. 知乎(https://zhuanlan.zhihu.com/p/43295545) [^5]: 杨镒铭. 详解Embeddings at Alibaba(KDD 2018). 知乎(https://zhuanlan.zhihu.com/p/56119617)

码字很辛苦,有收获的话就请分享、点赞、在看三连吧?

本文分享自微信公众号 - 小小挖掘机(wAIsjwj)

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

原始发表时间:2020-08-17

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

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 推荐系统遇上深度学习(四十九)-九篇阿里推荐相关论文汇总!

    温故而知新!前面的四十八篇文章中,单单是阿里的文章咱们就写了九篇了。今天就来简单回顾一下!

    石晓文
  • 推荐系统遇上深度学习(四十四)-Airbnb实时搜索排序中的Embedding技巧

    本文介绍的论文题目是:《Real-time Personalization using Embeddings for Search Ranking at Air...

    石晓文
  • 算法工程师如何应对业务方和老板的灵魂拷问?

    潘乱老师的文章中有一段叙述,描述了 Robin 经常反馈 badcase 的情况:

    石晓文
  • 深度学习推荐系统中各类流行的Embedding方法(上)

    Embedding,中文直译为“嵌入”,常被翻译为“向量化”或者“向量映射”。在整个深度学习框架中都是十分重要的“基本操作”,不论是NLP(Natural La...

    OpenCV学堂
  • 58同城 | Embedding技术在房产推荐中的应用

    导读:在深度学习的应用过程中,Embedding 这样一种将离散变量转变为连续向量的方式为神经网络在各方面的应用带来了极大的扩展,有"万物皆可 Embeddin...

    week
  • python字符串类型

    字符串长度 >>> a = 'hello python'>>> len(a)12 字符串取值从0开始 >>> a[0]'h' >>> a[1]'e' >>> a...

    零月
  • [PHP] 数据结构-二叉树的创建PHP实现

    1.利用递归的原理,只不过在原来打印结点的地方,改成了生成结点,给结点赋值的操作 if(ch=='#'){*T=NULL;}else{malloc();(*T)...

    陶士涵
  • [译]Laravel 5.0 之 Middleware (Filter-Style)

    本文译自 Matt Stauffer 的系列文章. ---- 如果你有阅读我之前的 Laravel 5.0 系列文章,你可能已经注意到路由过滤器(route f...

    小李刀刀
  • Swift3.0 - 泛型

    酷走天涯
  • 解决VMware虚拟机搭建linux、win环境时遇到网络桥接无法使用、NAT网络正常访问的情况。

    使用NAT网络是VM虚拟出来的网段,可供直接上网。但在某些情况下需要虚拟机中的系统访问和物理机一样的局域网就要使用桥接的访问,让虚拟机中的系统也可以跟物理机一...

    96php.cn

扫码关注云+社区

领取腾讯云代金券