推荐系统FM系列文章(一)-- FM模型

0. 写在前面

推荐系统相关从业人员肯定对FM(Factorization Machines)模型不会感到陌生,工业界及学术界在FM的基础上也提出了一系列优化模型,这些模型至今仍广泛应用于各类场景。本篇文章将带领大家重温FM模型,一探其中原理。

个人心得:

  1. 相比于LR,FM引入了二阶特征组合
  2. 通过矩阵分解引入隐向量的方式来估计特征参数,解决特征稀疏问题,同时大大减少参数数目
  3. 相比于MF(Matrix Factorization),FM能够引入用户特征、物品特征以外的其他特征,应用范围更广

论文地址:

https://www.csie.ntu.edu.tw/~b97053/paper/Rendle2010FM.pdf

1. 背景

在推荐系统实现个性化推荐初期,协同过滤取得了广泛的应用。协同过滤基于相似物品的相同受众或者相似用户的相同偏好的思想实现个性化推荐,为了解决协同过滤UI矩阵(用户-物品矩阵)的稀疏问题,MF被提出用于估计用户与未发生交互的物品之间的得分。然而,协同过滤只能够考虑用户与物品的交互特征,用户及物品本身的特征,模型并不能够有效利用。为了充分利用特征,人们将LR(Linear Regression)模型应用到推荐场景当中,但是众所周知,LR是线性模型,无法拟合二阶及二阶以上的特征。为了解决这个问题,并同时保证模型的可应用型,FM模型问世。

2. 模型架构

我们先来看在推荐场景下的一组典型特征,如下图所示。

image

对id类特征进行特征编码后,我们得到新的特征组合为

image

上面我们也提到,如果使用一阶线性模型LR来拟合特征的话,LR的表示为
y=w_{0}+\sum_{i=1}^{n}w_{i}x_{i}
这里的LR只考虑拟合样本一阶特征,没有考虑二阶特征的情形。然而,二阶组合特征,例如“性别=男,颜色=蓝色”,是非常有意义的。那么,考虑二阶组合特征的LR可以表示为
y=w_{0}+\sum_{i=1}^{n}w_{i}x_{i}+\sum_{i=1}^{n-1}\sum_{j=i+1}^{n}w_{ij}x_{i}x_{j}
相比于一阶LR,二阶LR多了二阶特征这一项,即特征两两组合并进行枚举。这个地方,会出现两个问题:

  1. 二阶特征参数数目为n^{2}, 参数数目呈指数级增加。
  2. 样本稀疏导致的学习问题。类别变量经编码后大部分位数为0,与其他特征进行组合后,取值仍然为0,会导致样本稀疏的情况。在LR的训练过程中,对于稀疏样本,值为0的特征对应的参数将很难被更新。

FM模型在有效完成二阶特征组合工作的同时,有效的解决了上述问题,因此被广泛应用于各类特征二阶交叉的场景。FM模型的实现形式如下式所示
y=w_{0}+\sum_{i=1}^{n}w_{i}x_{i}+\sum_{i=1}^{n-1}\sum_{j=i+1}^{n}< v_{i},v_{j} >x_{i}x_{j}

可以看到,除了二阶特征项与二阶LR模型有差别以外,其他部分完全相同。下面,我们就来详细讲解下这部分。

FM模型的二阶项依然是枚举特征两两交叉的情况,但是稍有区别的是,二阶交叉特征的参数由标量w_{i,j}变为了向量v_{i},v_{j}的点积。v_{i},v_{j}如下所示,每一维一阶特征x_{i}均有相对应的向量v_{i}
V_{n\times k}= \begin{bmatrix} a_{11}& a_{12}& \cdots & a_{1k}\\ a_{21}& a_{22}& \cdots & a_{2k}\\ \vdots & \vdots & \ddots & \vdots \\ a_{n1}& a_{n2}& \cdots & a_{nk} \end{bmatrix} =\begin{bmatrix} v_{1}\\ v_{2}\\ \vdots\\ v_{n} \end{bmatrix}
可以看到,v_{i},v_{j}是维度为k的向量,其中k是由模型使用者设定的超参数。那么,这里我们可以计算一下,FM使用到的参数数量为n*k,也就是上面向量矩阵的大小。其中,n为一阶特征数目,k为向量长度(k<<n)。相比于二阶LR中n*n的参数数目,FM模型可以说是极大的减少了模型参数数目,同时解决了上述的第一个问题。由于每个一阶特征均有其对应的向量v,并且每个向量在训练过程中的更新不再仅仅与某一个向量发生交互,而是与所有的向量发生交互,

举个例子,即便在所有样本中x_{i}x_{j}没有共同出现过,也不影响其权重w_{ij}的训练,因为v_{i}会在训练过程中根据v_{k}k != i,j)而更新,不再依赖于v_{j},同理,v_{j}也会根据v_{k}k != i,j)而更新,不再依赖于v_{i}

如此,FM模型解决了样本稀疏情况下参数的学习和更新问题。

3. 总结

FM模型作为推荐系统中的经典模型,解决了传统LR模型在特征二阶交叉后出现的参数爆炸及稀疏样本下的参数更新问题,因此被广泛应用于各类推荐场景。继FM模型后,学术界及工业界也尝试从不同的角度开展FM优化工作,涌现出了一批应用于推荐场景下的优秀模型。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,014评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,796评论 3 386
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,484评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,830评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,946评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,114评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,182评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,927评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,369评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,678评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,832评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,533评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,166评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,885评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,128评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,659评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,738评论 2 351

推荐阅读更多精彩内容