FM 模型理论基础

1. 背景介绍

假设一个广告分类的问题,根据用户和广告位相关的特征,预测用户是否点击了广告。数据如下:

clicked? Country Day Ad_type
1 USA 26/11/15 Movie
0 China 19/2/15 Game
1 China 26/11/15 Game

由上表可知,3维特征和点击标签,对类别特征进行One-hot编码转化为数值特征,转化为数据如下:

clicked? Country=USA Country=China Day=26/11/15 Day=19/2/15 Ad_type=Movie Ad_type=Game
1 1 0 1 0 1 0
0 0 1 0 1 0 1
1 0 1 1 0 0 1

特征空间还是比较稀疏的,并且在类别空间很大时,特征则越稀疏。并且根据不同的特征组合来看,两两组合的特征是十分必要的。原因如下:

通过观察大量的样本数据可以发现,某些特征经过关联之后,与label之间的相关性就会提高。例如,“USA”与“Thanksgiving”、“China”与“Chinese New Year”这样的关联特征,对用户的点击有着正向的影响。换句话说,来自“China”的用户很可能会在“Chinese New Year”有大量的浏览、购买行为,而在“Thanksgiving”却不会有特别的消费行为。这种关联特征与label的正向相关性在实际问题中是普遍存在的,如“化妆品”类商品与“女”性,“球类运动配件”的商品与“男”性,“电影票”的商品与“电影”品类偏好等。

如何表示两个特征的组合呢?一种方法是采用多项式模型表示两个特征的组合,x_i为第 i 个特征的取值,x_ix_j表示特征组合,其参数w_{ij}即为学习参数,也是x_ix_j组合的重要程度:
g(x)=w_{0}+\sum_{i} w_{i} x_{i}+\sum_{i} \sum_{j=i+1} w_{i j} x_{i} x_{j}

上式称为 Poly2 模型,参数个数:一次项有 d+1 个,二次项共有 d(d-1)/2 个,而参数与参数之间彼此独立,在稀疏场景下,二次项的训练是很困难的。因为要训练w_{ij},需要有大量的 x_ix_j都非零的样本,样本少则难以估计。

2. FM 模型

Poly2 模型认为参数w_{ij}是彼此独立的,所以必须进行单独训练。但是实践上不同的特征之间进行组合并非完全独立。参考矩阵分解,rating 矩阵可以分解为 user 矩阵和 item 矩阵的矩阵乘,如下图所示:

user 和 item 矩阵的维度分别为 (n, k) 和 (k, m),相比原来的rating矩阵,空间占用得到降低,并且分解后的user矩阵暗含着user偏好,Item矩阵暗含着item的属性,而user矩阵乘上item矩阵就是rating矩阵中用户对item的评分。

参考矩阵分解的过程,FM 模型将二次项参数w_{ij}进行分解:
g(x)=w_{0}+\sum_{i} w_{i} x_{i}+\sum_{i} \sum_{j=i+1}\left\langle v_{i}, v_{j}\right\rangle x_{i} x_{j}

其中v_i是第 i 维特征的隐向量,长度为k,(v_i, v_j)为内积,值为原来的w_{ij}

上式的复杂度可以从 O(kd^2)优化到O(kd)
\begin{aligned} & \sum_{i=1}^{d} \sum_{j=i+1}^{d}\left\langle\mathbf{v}_{i}, \mathbf{v}_{j}\right\rangle x_{i} x_{j} \\ = & \frac{1}{2} \sum_{i=1}^{d} \sum_{j=1}^{d}\left\langle\mathbf{v}_{i}, \mathbf{v}_{j}\right\rangle x_{i} x_{j}-\frac{1}{2} \sum_{i=1}^{d}\left\langle\mathbf{v}_{i}, \mathbf{v}_{j}\right\rangle x_{i} x_{i} \\ = & \frac{1}{2} \sum_{i=1}^{d} \sum_{j=1}^{d} \sum_{f=1}^{k} v_{i, f} v_{j, f} x_{i} x_{j}-\frac{1}{2} \sum_{i=1}^{d} \sum_{f=1}^{k} v_{i, f} v_{i, f} x_{i} x_{i} \\ = & \frac{1}{2} \sum_{f=1}^{k}\left(\left(\sum_{i=1}^{d} v_{i, f} x_{i}\right)\left(\sum_{j=1}^{d} v_{j, f} x_{j}\right)-\sum_{i=1}^{d} v_{i, f}^{2} x_{i}^{2}\right) \\ = & \frac{1}{2} \sum_{f=1}^{k}\left(\left(\sum_{i=1}^{d} v_{i, f} x_{i}\right)^{2}-\sum_{i=1}^{d} v_{i, f}^{2} x_{i}^{2}\right) \end{aligned}

3. FFM 模型

FM模型中,所有特征共享同一个隐空间,每一维特征对应为唯一的隐向量进行特征交互。存在的问题在于:不同特征是按照不同的 filed 划分,比如 “EPSN、NBC” 属于广告商、“Nike、Adidas”属于广告主、“Male、Female”属于性别。描述(EPSN,Nike)和(EPSN,Male)特征组合,FM模型都用同一w_{EPSN},而实际上,ESPN作为广告商,其对广告主和用户性别的潜在影响可能是不同的。

表示的不同点为:
FM:<v_{EPSN}, v_{Nike}> * 1 * 1 + <v_{EPSN}, v_{Male}> * 1 * 1 + <v_{Nike}, v_{Male}> * 1 * 1
FFM:<v_{EPSN,广告主}, v_{Nike, 广告商}> * 1 * 1 + <v_{EPSN,性别}, v_{Male,广告商}> * 1 * 1 + <v_{Nike,性别}, v_{Male,广告主}> * 1 * 1

FFM的数学公式表示为:
g(x)=w_{0}+\sum_{i} w_{i} x_{i}+\sum_{i} \sum_{j>i}\left\langle v_{i, f_{j}}, v_{j, f_{i}}\right\rangle x_{i} x_{j}

其中f_if_j分别代表第 i 个特征和第 j 个特征所属的 field,若field有f个,隐向量的长度为k,则二次项系数共有dfk个,远多于FM模型的dk个。此外,隐向量和field相关,并不能像FM模型一样将二次项化简,计算的复杂度是d^2k。通常情况下,每个隐向量只需要学习特定field的表示,所以有 k_{FFM} << k_{FM}

下面的图来自criteo,很好的表示了 Poly2、FM、FFM 模型的区别:

4. 总结

4.1. LR
g(x)=w_{0}+\sum_{i} w_{i} x_{i}

弊端:无法利用组合特征

4.2. Poly2
g(x)=w_{0}+\sum_{i} w_{i} x_{i}+\sum_{i} \sum_{j=i+1} w_{i j} x_{i} x_{j}

弊端:真实业务场景中组合特征高度稀疏,w_{i j}无法更新。

4.3. FM
g(x)=w_{0}+\sum_{i} w_{i} x_{i}+\sum_{i} \sum_{j=i+1}\left\langle v_{i}, v_{j}\right\rangle x_{i} x_{j}

局限:1. 仅建模二阶特征交互【引入深度神经网络】;2. 所有特征共享同一个latent space

4.4. FFM
g(x)=w_{0}+\sum_{i} w_{i} x_{i}+\sum_{i} \sum_{j>i}\left\langle v_{i, f_{j}}, v_{j, f_{i}}\right\rangle x_{i} x_{j}

不同特征按域划分。

参考文献:

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

推荐阅读更多精彩内容