FM 模型结构

FM 模型用于排序时，模型的公式定义如下：

$$\hat{y}(\mathbf{x}):=w_0+\sum _{i=1}^n{w}_i{x}_i+\sum _{i=1}^n\sum _{j=i+1}^n⟨{\mathbf{v}}_i,{\mathbf{v}}_j⟩x_i{x}_j$$

• 其中，$i$ 表示特征的序号，$n$ 表示特征的数量；$x_i\in \mathbb{R}$ 表示第 $i$ 个特征的值。
• $v_i,v_j \in \mathbb {R}^{k} $ 分别表示特征 $x_i,x_j$ 对应的隐语义向量（Embedding 向量）， $⟨{\mathbf{v}}_i,{\mathbf{v}}_j⟩:=\sum _{f=1}^k{v}_{i,f}\cdot v_{j,f}$ 。
• $w_0,w_i\in \mathbb{R}$ 均表示需要学习的参数。

FM 的一阶特征交互

在 FM 的表达式中，前两项为特征的一阶交互项。将其拆分为用户特征和物品特征的一阶特征交互项，如下：

$$\begin{array}{rl}& w_0+\sum _{i=1}^n{w}_i{x}_i\\ & =w_0+\sum _{t\in I}{w}_t{x}_t+\sum _{u\in U}{w}_u{x}_u\end{array}$$

• 其中，$U$ 表示用户相关特征集合，$I$ 表示物品相关特征集合。

FM 的二阶特征交互

观察 FM 的二阶特征交互项，可知其计算复杂度为 $O\left(k{n}^2\right)$ 。为了降低计算复杂度，按照如下公式进行变换。

$$\begin{array}{rl}& \sum _{i=1}^n\sum _{j=i+1}^n⟨{\mathbf{v}}_i,{\mathbf{v}}_j⟩x_i{x}_j\\ =& \frac{1}{2}\sum _{i=1}^n\sum _{j=1}^n⟨{\mathbf{v}}_i,{\mathbf{v}}_j⟩x_i{x}_j-\frac{1}{2}\sum _{i=1}^n⟨{\mathbf{v}}_i,{\mathbf{v}}_i⟩x_i{x}_i\\ =& \frac{1}{2}(\sum _{i=1}^n\sum _{j=1}^n\sum _{f=1}^k{v}_{i,f}{v}_{j,f}{x}_i{x}_j-\sum _{i=1}^n\sum _{f=1}^k{v}_{i,f}{v}_{i,f}{x}_i{x}_i)\\ =& \frac{1}{2}\sum _{f=1}^k({\left(\sum _{i=1}^n{v}_{i,f}{x}_i\right)}^{}\left(\sum _{j=1}^n{v}_{j,f}{x}_j\right)-\sum _{i=1}^n{v}_{i,f}^2{x}_i^2)\\ =& \frac{1}{2}\sum _{f=1}^k({\left(\sum _{i=1}^n{v}_{i,f}{x}_i\right)}^2-\sum _{i=1}^n{v}_{i,f}^2{x}_i^2)\end{array}$$

• 公式变换后，计算复杂度由 $O\left(k{n}^2\right)$ 降到 $O\left(kn\right)$。

由于本文章需要将 FM 模型用在召回，故将二阶特征交互项拆分为用户和物品项。有：

$$\begin{array}{rl}& \frac{1}{2}\sum _{f=1}^k({\left(\sum _{i=1}^n{v}_{i,f}{x}_i\right)}^2-\sum _{i=1}^n{v}_{i,f}^2{x}_i^2)\\ =& \frac{1}{2}\sum _{f=1}^k({(\sum _{u\in U}{v}_{u,f}{x}_u+\sum _{t\in I}{v}_{t,f}{x}_t)}^2-\sum _{u\in U}{v}_{u,f}^2{x}_u^2-\sum _{t\in I}{v}_{t,f}^2{x}_t^2)\\ =& \frac{1}{2}\sum _{f=1}^k({\left(\sum _{u\in U}{v}_{u,f}{x}_u\right)}^2+{\left(\sum _{t\in I}{v}_{t,f}{x}_t\right)}^2+2\sum _{u\in U}{v}_{u,f}{x}_u\sum _{t\in I}{v}_{t,f}{x}_t-\sum _{u\in U}{v}_{u,f}^2{x}_u^2-\sum _{t\in I}{v}_{t,f}^2{x}_t^2)\end{array}$$

• 其中，$U$ 表示用户相关特征集合，$I$ 表示物品相关特征集合。

FM 用于召回

基于 FM 召回，我们可以将 $\hat{y}(\mathbf{x}):=w_0+\sum _{i=1}^n{w}_i{x}_i+\sum _{i=1}^n\sum _{j=i+1}^n⟨{\mathbf{v}}_i,{\mathbf{v}}_j⟩x_i{x}_j$ 作为用户和物品之间的匹配分。

• 在上一小节中，对于 FM 的一阶、二阶特征交互项，已将其拆分为用户项和物品项。
• 对于同一用户，即便其与不同物品进行交互，但用户特征内部之间的一阶、二阶交互项得分都是相同的。
• 这就意味着，在比较用户与不同物品之间的匹配分时，只需要比较：（1）物品内部之间的特征交互得分；（2）用户和物品之间的特征交互得分。

FM 的一阶特征交互

• 将全局偏置和用户一阶特征交互项进行丢弃，有：$$F{M}_{\mathrm{一}\mathrm{阶}}=\sum _{t\in I}{w}_t{x}_t$$

FM 的二阶特征交互

• 将用户特征内部的特征交互项进行丢弃，有：$$\begin{array}{rl}& F{M}_{\mathrm{二}\mathrm{阶}}=\frac{1}{2}\sum _{f=1}^k({\left(\sum _{t\in I}{v}_{t,f}{x}_t\right)}^2+2\sum _{u\in U}{v}_{u,f}{x}_u\sum _{t\in I}{v}_{t,f}{x}_t-\sum _{t\in I}{v}_{t,f}^2{x}_t^2)\\ & =\frac{1}{2}\sum _{f=1}^k({\left(\sum _{t\in I}{v}_{t,f}{x}_t\right)}^2-\sum _{t\in I}{v}_{t,f}^2{x}_t^2)+\sum _{f=1}^k\left(\sum _{u\in U}{v}_{u,f}{x}_u\sum _{t\in I}{v}_{t,f}{x}_t\right)\end{array}$$

合并 FM 的一阶、二阶特征交互项，得到基于 FM 召回的匹配分计算公式：

$${\text{MatchScore}}_{FM}=\sum _{t\in I}{w}_t{x}_t+\frac{1}{2}\sum _{f=1}^k({\left(\sum _{t\in I}{v}_{t,f}{x}_t\right)}^2-\sum _{t\in I}{v}_{t,f}^2{x}_t^2)+\sum _{f=1}^k\left(\sum _{u\in U}{v}_{u,f}{x}_u\sum _{t\in I}{v}_{t,f}{x}_t\right)$$

在基于向量的召回模型中，为了 ANN（近似最近邻算法） 或 Faiss 加速查找与用户兴趣度匹配的物品。基于向量的召回模型，一般最后都会得到用户和物品的特征向量表示，然后通过向量之间的内积或者余弦相似度表示用户对物品的兴趣程度。

基于 FM 模型的召回算法，也是向量召回算法的一种。所以下面，将 ${\text{MatchScore}}_{FM}$ 化简为用户向量和物品向量的内积形式，如下：

$${\text{MatchScore}}_{FM}=V_{item}{V}_{user}^T$$

• 用户向量：

  $$V_{user}=[1;\sum _{u\in U}{v}_u{x}_u]$$
  • 用户向量由两项表达式拼接得到。
  • 第一项为常数 $1$，第二项是将用户相关的特征向量进行 sum pooling 。

• 物品向量：

  $$V_{item}=[\sum _{t\in I}{w}_t{x}_t+\frac{1}{2}\sum _{f=1}^k({\left(\sum _{t\in I}{v}_{t,f}{x}_t\right)}^2-\sum _{t\in I}{v}_{t,f}^2{x}_t^2);\sum _{t\in I}{v}_t{x}_t]$$
  • 第一项表示物品相关特征向量的一阶、二阶特征交互。
  • 第二项是将物品相关的特征向量进行 sum pooling 。

思考题

1. 为什么不直接将 FM 中学习到的 User Embedding： $\sum _{u\in U}{v}_u{x}_u$ 和 Item Embedding： $\sum _{t\in I}{v}_t{x}_t$ 的内积做召回呢？

   答：这样做，也不是不行，但是效果不是特别好。因为用户喜欢的，未必一定是与自身最匹配的，也包括一些自身性质极佳的 item（e.g., 热门 item），所以，非常有必要将 "所有 Item 特征一阶权重之和" 和 “所有 Item 特征隐向量两两点积之和” 考虑进去，但是也还必须写成点积的形式。

代码实战

正在完善...

参考链接

• paper.dvi (ntu.edu.tw)
• FM：推荐算法中的瑞士军刀 - 知乎 (zhihu.com)