万字详述推荐系统经典模型DCN,DCN_v2,PNN

公众号：ChallengeHub(持续分析机器学习，数据挖掘，推荐系统，NLP方面的知识)

前言：

之前文章介绍了特征交叉的一些方式和几种推荐系统中关于特征交叉的一些论文。上篇文章中主要介绍了wide&deep，deepFM，和NFM模型。其中wide&deep，和deepFM模型的特征交叉就一层，属于浅度的特征交叉，NFM可以实现更深的特征交叉。这篇文章将介绍两种经典的推荐模型，它们可以实现更深度的特征交叉。分别为DCN,DCN_v2和PNN。

1：DCN

论文名称：Deep & Cross Network for Ad Click Predictions
论文链接：https://arxiv.org/pdf/1708.05123.pdf
论文结构：

模型结构解读：
(1)：$x_0=[x_{embed,1}^T,...x_{embed,k}^T,x_{dense}^T]$
$x_0$代表的就是将输入特征处理之后进行concat起来的结果。每篇论文其实对于特征的处理形式都不一样，这篇论文是将类别的embedding，然后数值型的直接输入，将所有特征处理后拼接成一个大的embedding就是$x_0$。
(2)右边结构：$h_1=ReLU(W_{h,0}{x}_0+bh,0)$
这就是一个普通的DNN结构，激活函数采用的是ReLu并没有什么特殊之处。
(3)左侧结构：$x_{l+1}=x_0{x}_l^T{w}_l+b_l+x_l$
这个就是DCN的核心之处了，$x_0$是一个维度为$d$的列向量，$w_l$，$d_l$同样是一个$d$维度的向量，是模型的参数。公式可视化之后是这样的：

(4)上方结构：$p=\sigma ([x_{L_1}^T,h_{L_2}^T{w}_{logits}])$
将左侧和右侧得到的两个embedding直接concat起来成为一个大的embedding,经过一个浅层的神经网络输出一个一维数字，再经过一个sigmoid函数就是最终的结果了。

2:DCN_V2

论文名称：DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systems
论文地址：https://arxiv.org/pdf/2008.13535v2.pdf
看论文的名字，就知道该模型与DCN很相似了，该模型确实基于DCN的一个改进模型。与DCN最大的区别是特征交叉的地方有一点点地差异：

仔细一对比的话就是将DCN中一维向量$w_l$改变成了二维矩阵$W$。除此之外，这篇论文中还提出了DCN的两种结构：

一篇论文肯定不会这么简单嘛，工作量这么少的论文也发不了顶会呀。于是作者们又对DCN进行了改进，原因是DCN_V2中计算的复杂度太高了。

模型改进：

改进的地方同样来自于特征交叉的地方，论文中说的让人有点不能理解，我们直接看修改的部分吧:

(1)：由于将特征embedding之后再拼接起来成了一个$d$维的向量，这个$d$太大了，而矩阵$W$的维度是$d\times d$。所有导致这个计算的复杂度就很高了，于是我们可以$W=U{V}^T$，类似于矩阵分解的方法，将维度比较大的$d\times d$矩阵分解成两个维度小一些的$d\times r$矩阵。其中$r$远小于$d$。这种方法叫做矩阵的低阶分解，和SVD有点类似。这也我们的交叉公式就发生了变化:
原来 ：$x_{l+1}=x_0\odot (W_l{x}_l+b_l)+x_l$
现在：$x_{l+1}=x_0\odot (U_l(V_l^T{x}_l)+b_l)+x_l$
参数的数量和运算的复杂度都有效的变低了。

(2)：收到今年比较受欢迎的 Mixture-of-Experts (MoE)的影响，作者将这种结构也放入了交叉中之中。于是我们的交叉的公式又变为了以下的形式：
$$x_{l+1}=\sum _{i=1}^K{G}_i(x_l)E_i(x_l)+x_l$$
$$E_i(x_l)=x_0\odot (U_l^i(V_l^{iT}{x}_l)+b_l)$$
其中$G_i$就是一个门函数，可以是sigmoid或者是softmax。

(3)：增加非线性。
$$E_i(x_l)=x_0\odot (U_l^i\cdot g(C_l^i\cdot g(V_l^{iT}{x}_l))+b_l)$$
注：论文中用的是点乘，但是从浅梦大佬的代码中用的是正常的矩阵乘法。我也认为是矩阵乘法，否则最后$U_l^i\cdot g(C_l^i\cdot g(V_l^{iT}{x}_l))$的维度对不上，如果是矩阵乘法的话，那么$C_l^i$的维度则为$r\times r$。

改进的后模型据作者所说在降低了30%的复杂度的情况下，保留了模型的精度，模型可以称之为：the mixture of low-rank DCN(DCN-Mix)。

大佬的代码地址:https://github.com/shenweichen/DeepCTR-Torch/blob/bc881dcd417fec64f840b0cacce124bc86b3687c/deepctr_torch/layers/interaction.py#L464-L537

灵魂问答

这篇文章比较有趣的是关于实现部分是采用问答的形式展开的，总共有5个问题，同样也给出了答案。担忧翻译问题于是我附上了英文。(文中对mDCN应该就是DCN-Mix的意思，命名有点混乱，希望我没弄错)
(1)：什么时候特征交互学习方法会比基于relu的DNN更有效？
(When would feature interaction learning methods become more efficient than ReLU-based DNN)

从论文后续的答案中，作者从三个方面回答了该问题(不过更像是回答怎样可以提升DCN_v2的效果)
a：提升交叉的复杂度
文中使用了三种二维的交叉方法($x_i$代表的是某个特征)，按照交叉的复杂程度有以下三种。

然后就是三种交叉的结果：

b：DCN中各个部分的作用

c：提升交叉的阶数

总结：总而言之，即使使用更深更大的网络，ReLU 在捕获显式特征交叉（乘法关系）方面效率低下。 当交叉模式变得更复杂时，精度会大大降低。 DCN 准确地捕获了简单的交叉模式，但无法捕捉更复杂的交叉模式。 另一方面，DCN-V2 对复杂的交叉模式保持准确和高效

(2)：如果去除掉DNN结构的话，各种交叉方法的表现是怎么样的？
(How does the feature-interaction component of each baseline perform without integrating with DNN)

总结：高阶方法表现出优于二阶方法的性能。 这表明高阶交叉在该数据集中是有意义的。在高阶方法中，交叉网络取得了最好的性能，与 DNN 相比略胜一筹。

(3)：DCN-Mix的于基础的DCN_v2相比表现如何，我们应该这么在模型的精确度和模型的消耗之间进行平衡(以DCN为例)。
(How does the proposed mDCN approaches compare to the baselines? Could we achieve healthier trade-off between model accuracy and cost through mDCN and the mixture of low-rank DCN)

总结：在所有方法中，DCN-V2效果最好，同时相比性能不差； DCN-Mix 进一步降低了模型消耗，在模型性能和效果之间取得了更好的平衡。(作者主要对比了DCN_v2和DCN-Mix与其它模型之间的效果)

(4)：DCN-Mix参与与模型效果之间的关系
(Q4 How does the settings in mDCN affect model quality)

上图左边展现的是DNN模型和Cross NetWork在不同网络层数下的效果。右侧是展现在不同的矩阵维度情况下的效果。(这个matrix rank就是在改进模型中$W_l,r\times d$中$r$的大小，也就是从高维分解为低维中，低维的数量)。

(5)：DCN-Mix是否捕捉到了重要的特征交叉？模型的解释性如何？
(Q5 Is mDCN capturing important feature crosses? Does the model provide good understandabilit)

作者直接从实验结果回答了该问题。左侧是DCN_v2中的$W$权重矩阵，右侧显示的特征交叉，颜色越深代表者交叉越重要，从两张图的重合程度可以看出还是学到了一些重要的显示交叉特征，比如gender-UserId和Movied-UserId。

作为近年来推荐系统中的SOTA模型，这篇论文还是有很多启发的。DCN的改进，高阶矩阵分解等操作还是很有学习的意义。并且实验阶段给我们展示了很多扩展性的工作。

PNN

论文地址:https://arxiv.org/pdf/1611.00144v1.pdf
模型结构：

首先看模型结构，从下面往上面看，第一层input就到embedding就不需要再解释了吧。
(1)$\mathbf{z}$：$z$就是简单地将所有feature地embedding直接复制过去。
其中$\mathbf{z}=({\mathbf{z}}_1,{\mathbf{z}}_2,...{\mathbf{z}}_{\mathbf{N}})===({\mathbf{f}}_1,{\mathbf{f}}_2,...{\mathbf{f}}_{\mathbf{N}})$，$.{\mathbf{f}}_{\mathbf{i}}$就是特征$i$的embedding。

(2) $\mathbf{p}$：$p_{i,j},i=1...N，j=1...N$,其中$p_{i,j}=g({\mathbf{f}}_{\mathbf{i}},{\mathbf{f}}_{\mathbf{j}})$，可以看出$p_{i,j}$就是特征$i,j$之间交叉的结果。至于怎么交叉，我们稍后再说。

(3)：从$z,p$层到达$L_1$我感觉是我看这篇论文没有想到的。
在我看来，应该就是将$\mathbf{z}$展开，再将$\mathbf{p}$展开，然后concat成一个一维的向量，然后再经过一层网络得到L1，但其实并不是这样的。我们介绍L1的构成：
$${\mathbf{l}}_1=relu({\mathbf{l}}_z+{\mathbf{l}}_p+{\mathbf{b}}_1)$$

可以看出${\mathbf{l}}_z,{\mathbf{l}}_p,{\mathbf{b}}_1$三个一维的向量加起来再经过relu得到的L1。与我看模型结构时的猜想并不一致。

首先我们来介绍${\mathbf{l}}_z$的构成：
$${\mathbf{l}}_z=(l_z^1,l_z^2,...l_z^{D_1})$$
$l_z^n={\mathbf{W}}_z^n\odot \mathbf{z}={\sum }_{i=1}^N{\sum }_{j=1}^M({\mathbf{W}}_z^n{)}_{i,j}{z}_{i,j}$可以看出，${\mathbf{W}}_z^n$时一个维度与$\mathbf{z}$完全相同的矩阵，最后${\mathbf{l}}_{\mathbf{z}}$的维度有多少($D_1$)，则${\mathbf{W}}_z^n$的个数有多少。

同理我们可以得到：
$l_p^n={\mathbf{W}}_p^n\odot \mathbf{p}={\sum }_{i=1}^N{\sum }_{j=1}^N({\mathbf{W}}_p^n{)}_{i,j}{p}_{i,j}$。但是在求$p_{i,j}$处作者提出了两种方法。

第一种是IPNN

这种就是最常见的交叉了，$p_{i,j}=<{\mathbf{f}}_{\mathbf{i}},{\mathbf{f}}_{\mathbf{j}}>={\mathbf{f}}_{\mathbf{i}}^{\mathbf{T}}{\mathbf{f}}_{\mathbf{j}}$
受到FM模型的启发，将${\mathbf{W}}_{\mathbf{p}}^{\mathbf{n}}$转化为两个向量。即：${\mathbf{W}}_{\mathbf{p}}^{\mathbf{n}}={\mathbf{\theta }}^{\mathbf{n}}{{\mathbf{\theta }}^{\mathbf{n}}}^T$
因此原来的等式就可以变为：
$$l_p^n={\mathbf{W}}_p^n\odot p=\sum _{i=1}^N\sum _{j=1}^N{\theta }_i^n{{\theta }_j^n}^T<{\mathbf{f}}_{\mathbf{i}},{\mathbf{f}}_{\mathbf{j}}>=<\sum _{i=1}^N{\delta }_i^n,\sum _{i=1}^N{\delta }_i^n>$$
此时：${\delta }_i^n={\theta }_i^n{\mathbf{f}}_{\mathbf{i}}$。可得$l_p^n=(||{\sum }_{i=1}{\delta }_i^1||,...||{\sum }_{i=1}{\delta }_i^n||)$。引入这种方法与FM的初衷也是一样，为了降低计算的复杂度，同时也可以减少参数的数量。

本来计算的时间复杂度为：$O(N^2(D_1+M)$($N$是特征的数量，$M$为特征embedding的维度，$D_1$为最后${\mathbf{l}}_{\mathbf{p}}$的维度)。由于$p_{i,j}$只需要计算一次，时间复杂度并不是直接的$O(N^2{D}_{1}M)$。通过降维后时间复杂度为$O(N{D}_{1}M)$。

这种通过矩阵分解的方式减少参数数量，降低计算的复杂度在推荐模型中应用还是很多的，上面的DCN-M同样也应用到了。

第二种是OPNN

这种交叉比较少见，${\mathbf{p}}_{\mathbf{i},\mathbf{j}}={\mathbf{f}}_{\mathbf{i}}{{\mathbf{f}}_{\mathbf{j}}}^T$。在IPNN中p_{i,j}是一个数字，但是在OPNN中，则是一个矩阵，${\mathbf{p}}_{\mathbf{i},\mathbf{j}}$的维度为$M\times M$。此时$\mathbf{p}$再也不是根据${\mathbf{p}}_{\mathbf{i},\mathbf{j}}$，而是这样的：

$$\mathbf{p}=\sum _{i=1}^N\sum _{j=1}^N{\mathbf{f}}_{\mathbf{i}}{{\mathbf{f}}_{\mathbf{i}}}^T$$
总结来说，有N个特征，那么就会交叉N的平方次，每次交叉都会形成一个矩阵，将N的平方和矩阵相加起来就可以得到$\mathbf{p}$了。直观来说就是将上面那个正方形的矩阵层层叠加，最后得到一个相同大小的矩阵。该矩阵的维度同样为$M\times M$。此时计算的复杂度达到惊人的$O(N^2{D}_1{M}^2)$。

同样上面的公式还是可以进行优化的：
$$\mathbf{p}={\mathbf{f}}_{\sum }{{\mathbf{f}}_{\sum }}^T$$

其中${\mathbf{f}}_{\sum }={\sum }_i^N{\mathbf{f}}_{\mathbf{i}}$。然后直接采用公式$l_p^n={\mathbf{W}}_p^n\odot \mathbf{p}$就可以得到我们要的结果。

以上就是关于推荐系统中经典模型DCN，DCN_v2，PNN模型的详细介绍，介绍得很详细，对于我自己来说相当于重新仔细地学习了一遍，收获不浅，更加了解了模型中的具体细节。同时也希望可以给有兴趣学习推荐模型的一些朋友有一定的帮助。

此外，喜欢数据挖掘，推荐系统，nlp的朋友可以关注我们公众号，然后回复加群添加我好友。