DeepFM: A Factorization-Machine based Neural Network for CTR Prediction

Huifeng Guo, Ruiming Tang, Yunming Ye, Zhenguo Li, Xiuqiang He
3/13/2017
cs.IRcs.CL

Official code: reczoo/FuxiCTR (1,467 stars, Apache-2.0, PyTorch, last commit 2025-06-16).

Abstract

Learning sophisticated feature interactions behind user behaviors is critical in maximizing CTR for recommender systems. Despite great progress, existing methods seem to have a strong bias towards low- or high-order interactions, or require expertise feature engineering. In this paper, we show that it is possible to derive an end-to-end learning model that emphasizes both low- and high-order feature interactions. The proposed model, DeepFM, combines the power of factorization machines for recommendation and deep learning for feature learning in a new neural network architecture. Compared to the latest Wide \& Deep model from Google, DeepFM has a shared input to its "wide" and "deep" parts, with no need of feature engineering besides raw features. Comprehensive experiments are conducted to demonstrate the effectiveness and efficiency of DeepFM over the existing models for CTR prediction, on both benchmark data and commercial data.

View on arXivView PDF

Code Implementations(1)

reczoo/FuxiCTROfficial100%
PyTorchApache-2.0

A configurable, tunable, and reproducible library for CTR prediction https://fuxictr.github.io

1,467236Shell, PythonSep 30, 20211 years agoApache-2.0
ctrctr-predictioncvrpytorchrecommender-systems

Similar Papers

Cite this paper

@article{guo2017deepfm,
  title  = {DeepFM: A Factorization-Machine based Neural Network for CTR Prediction},
  author = {Huifeng Guo and Ruiming Tang and Yunming Ye and Zhenguo Li and Xiuqiang He},
  year   = {2017},
  eprint = {1703.04247},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1703.04247}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1703.04247.svg)](https://codeofpaper.com/paper/1703.04247)

Discussion