4 个月前

DeepFM:一种基于因子分解机的神经网络用于点击率预测

DeepFM:一种基于因子分解机的神经网络用于点击率预测

摘要

学习用户行为背后的复杂特征交互对于最大化推荐系统的点击率(CTR)至关重要。尽管已取得显著进展,现有方法似乎对低阶或高阶交互存在强烈偏见,或者需要专家进行特征工程。在本文中,我们展示了可以构建一个端到端的学习模型,该模型同时强调低阶和高阶特征交互。所提出的模型 DeepFM 结合了因子分解机在推荐中的优势和深度学习在特征学习中的能力,形成了一种新的神经网络架构。与谷歌最新的 Wide & Deep 模型相比,DeepFM 的“宽”部分和“深”部分共享相同的输入,除了原始特征外无需额外的特征工程。通过广泛的实验验证,证明了 DeepFM 在基准数据集和商业数据集上对现有 CTR 预测模型的有效性和高效性。

基准测试

基准方法指标
click-through-rate-prediction-on-amazonDeepFM
AUC: 0.8683
click-through-rate-prediction-on-bing-newsDeepFM
AUC: 0.8376
Log Loss: 0.2671
click-through-rate-prediction-on-companyDeepFM
AUC: 0.8715
Log Loss: 0.02618
click-through-rate-prediction-on-criteoDeepFM
AUC: 0.8007
Log Loss: 0.45083
click-through-rate-prediction-on-dianpingDeepFM
AUC: 0.8481
Log Loss: 0.3333
click-through-rate-prediction-on-kkboxDeepFM
AUC: 0.8531
click-through-rate-prediction-on-movielensDeepFM
AUC: 0.7324

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DeepFM:一种基于因子分解机的神经网络用于点击率预测 | 论文 | HyperAI超神经