排序模型在工业界已经有了很长时间的历史,从基于策略规则的人工指定特征权值开始,发展到LR线性模型,LR+GBDT半自动特征组合模型,再到FM自动二阶特征组合模型及深度学习模型等不断发展。其中FM系列模型占据比较重要的位置,本篇文章就FM模型进行分析和总结。
1,概述
在机器学习中,预测是一项基本的任务,所谓预测就是估计一个函数,该函数将一个n维的特征向量x映射到一个目标域T:
D={(x(1),y(1)),(x(2),y(2)),...,(x(N),y(N))}
推荐系统中比较重要的因素就是速度,比如容易并行,可解释、可拓展,这就是LR在工业界比较流行的原因之一。我们先回顾下线性模型和逻辑回归模型。
线性回归
y^(x)==w0+w1x1+w2x2+...+wnxnw0+j=1∑nwixi
逻辑回归
核心函数:
hθ(x)=1+e−θx1
对于一个样本:
P(y=1∣x,θ)=P(y=0∣x,θ)=hθ(x)1−hθ(x)
那么进一步,样本x的概率表示为:
P(y∣x;θ)=(hθ(x))y(1−hθ(x))1−y
似然函数:
L(θ)=i=1∏mP(y(i)∣x(i);θ)
接下来,取对数,最大似然,最小loss:
J(θ)=−m1log(L(θ))
求导、梯度下降。。。
2,FM模型
我们说线性模型有个非常大的问题就是没有考虑特征之间的相互关系:
y^(x)==w0+w1x1+w2x2+...+wnxnw0+j=1∑nwixi
事实上,如果特征之间相互关联,不同的特征组合可以提高模型的表达能力,而大多数情况下,特征之间都不是相互独立的。
为了表达这种关联特性,接下来,我们将函数y^改成:
y^(x)=w0+j=1∑nwixi+i=1∑n−1j=i+1∑nwijxixj
这样两个特征之间的关系也考虑了,不过很遗憾,这种直接在xixi前面配上系数wij的方式在稀疏数据面前有很大的缺陷。
在数据稀疏性普遍存在的实际应用场景中,二次项参数的训练是很困难的。其原因是:每个参数 wij的训练需要大量xi和xj非零样本,由于样本数据本来就比较稀疏,满xi和 xj都非零的样本将会非常少。训练样本的不足,很容易导致参数 wij不准确,最终将严重影响模型的性能。一句话,Wij只有在观察样本中有xi和xj的情况下才能优化得到,对于观察样本中未出现过的特征分量不能进行相应参数的评估.
对于推荐系统来讲,高度稀疏的数据场景太普遍了。为了克服这个困难我们引入辅助向量:
Vi=(vi1,vi2,...,vik)T
将w改为:
w^ij=ViTVj=l=1∑kvilvjl
y^(x)=w0+j=1∑nwixi+i=1∑n−1j=i+1∑n(ViTVj)xixj
目前这样的表达式的复杂度是O(n^2),但是我们可以对其进行分解,使得复杂度变为O(n)
![FM系列模型1——FM(Factorization Machine)算法 FM系列模型1——FM(Factorization Machine)算法](/default/index/img?u=aHR0cHM6Ly9waWFuc2hlbi5jb20vaW1hZ2VzLzMwOC8xYjlmYWU0ZGQyOTU4ZDg2N2UxOTZmMDI5MzY5ZWU5NC5wbmc=)
二次项的参数数量减少为 kn个,远少于多项式模型的参数数量。另外,参数因子化使得 xhxi 的参数和 xixj 的参数不再是相互独立的,因此我们可以在样本稀疏的情况下相对合理地估计FM的二次项参数。具体来说, xhxi 和 $ x_ix_j$ 的系数分别为 vh,vi和 vi,vj ,它们之间有共同项 vi 。也就是说,所有包含“ xi 的非零组合特征”(存在某个 j=i ,使得 xixj=0 )的样本都可以用来学习隐向量vi,这很大程度上避免了数据稀疏性造成的影响。而在多项式模型中, whi 和 wij是相互独立的。
到这里,我们就可以通过设置loss函数,梯度下降等方法进行优化了。
![FM系列模型1——FM(Factorization Machine)算法 FM系列模型1——FM(Factorization Machine)算法](/default/index/img?u=aHR0cHM6Ly9waWFuc2hlbi5jb20vaW1hZ2VzLzg4MC9lNmEyZWIyNzY0MmU3Zjc0ZTM2NjFjOWExOTM5NDk2OC5wbmc=)
参考资料:
https://www.cnblogs.com/pinard/p/6370127.html#!comments
https://zhuanlan.zhihu.com/p/37963267
https://blog.****.net/pql925/article/details/79021464
https://www.jianshu.com/p/27679d6f3949