中文

FiBiNET:结合特征重要性与双线性特征交互的点击率预测模型

机器学习 2019-11-13 v1 人工智能 机器学习

摘要

广告与信息流排序对Facebook、新浪微博等许多互联网公司至关重要。在众多现实世界的广告与信息流排序系统中,点击率(CTR)预测起着核心作用。该领域已有许多提出的模型,如逻辑回归、基于树的模型、基于因子分解机的模型以及基于深度学习的CTR模型。然而,许多当前工作以简单方式计算特征交互(如Hadamard积与内积),且较少关注特征的重要性。本文提出一种名为FiBiNET(即Feature Importance and Bilinear feature Interaction NETwork的缩写)的新模型,以动态学习特征重要性与细粒度特征交互。一方面,FiBiNET可通过Squeeze-Excitation网络(SENET)机制动态学习特征重要性;另一方面,它能通过双线性函数有效学习特征交互。我们在两个真实世界数据集上进行了大量实验,表明我们的浅层模型优于其他浅层模型如因子分解机(FM)与域感知因子分解机(FFM)。为进一步提升性能,我们将一个经典深度神经网络(DNN)组件与浅层模型结合成为深层模型。深层FiBiNET持续优于其他最先进的深层模型如DeepFM与极端深度因子分解机(XdeepFM)。

关键词

引用

@article{arxiv.1905.09433,
  title  = {FiBiNET: Combining Feature Importance and Bilinear feature Interaction for Click-Through Rate Prediction},
  author = {Tongwen Huang and Zhiqi Zhang and Junlin Zhang},
  journal= {arXiv preprint arXiv:1905.09433},
  year   = {2019}
}

备注

8 pages,5 figures