FFM
2022-11-11

 

FFM简介

FFM(Field-aware Factorization Machines),对FM模型做了优化。在FM中,特征xi与其它特征交叉时,其使用的隐向量都是vi。但是,考虑到不同特征的含义,如”年龄+周末”与”年龄+收入”,这两种情况下特征”年龄”的隐向量应该有所不同。因此,在FFM模型中,引入了”field”的概念,将类似的特征归入到同一个field,让某一特征与不同特征做交互时,可发挥不同的重要性,提升模型表达能力。

field直观表示如下:

field

FFM定义

FFM的模型定义如下:

(1)y^(x)=w0+i=1nwixi+i=1nj=i+1nvi,fj,vj,fixixjvi,fjjfjjfieldn

 

总结

  1. 优点

    • 引入 field 域的概念,让某一特征与不同特征做交互时,可发挥不同的重要性,提升模型表达能力;
    • 可解释性强,可提供某些特征组合的重要性。
  2. 缺点

    • 复杂度为o(kn2),不适用于特征数较多的场景。

 

参考文档

  1. ffm.pdf (ntu.edu.tw)
  2. 深入FFM原理与实践 - 美团技术团队 (meituan.com)
  3. 2.FFM数学模型保姆级解读哔哩哔哩bilibili