Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Huifeng Guo, Ruiming Tang, Yunming Ye, Zhenguo Li, Xiuqiang He• 2017

Related benchmarks

TaskDatasetResultRank
CTR PredictionCriteo
AUC0.8143
282
Click-Through Rate PredictionAvazu (test)
AUC0.7931
191
CTR PredictionAvazu
AUC78.96
144
CTR PredictionCriteo (test)
AUC0.8142
141
CTR PredictionFrappe
AUC0.9842
83
CTR PredictionMovieLens
AUC96.87
55
Click-Through Rate PredictionKKBOX
AUC85.31
48
Click-Through Rate PredictionCriteo (test)
AUC0.8056
47
Click-Through Rate PredictionML 1M
AUC0.9051
46
CTR PredictionFrappe (test)
AUC0.9835
38
Showing 10 of 76 rows
...

Other info

Follow for update