xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems

14 Mar 2018  ยท  Jianxun Lian, Xiaohuan Zhou, Fuzheng Zhang, Zhongxia Chen, Xing Xie, Guangzhong Sun ยท

Combinatorial features are essential for the success of many commercial models. Manually crafting these features usually comes with high cost due to the variety, volume and velocity of raw data in web-scale systems. Factorization based models, which measure interactions in terms of vector product, can learn patterns of combinatorial features automatically and generalize to unseen features as well. With the great success of deep neural networks (DNNs) in various fields, recently researchers have proposed several DNN-based factorization model to learn both low- and high-order feature interactions. Despite the powerful ability of learning an arbitrary function from data, plain DNNs generate feature interactions implicitly and at the bit-wise level. In this paper, we propose a novel Compressed Interaction Network (CIN), which aims to generate feature interactions in an explicit fashion and at the vector-wise level. We show that the CIN share some functionalities with convolutional neural networks (CNNs) and recurrent neural networks (RNNs). We further combine a CIN and a classical DNN into one unified model, and named this new model eXtreme Deep Factorization Machine (xDeepFM). On one hand, the xDeepFM is able to learn certain bounded-degree feature interactions explicitly; on the other hand, it can learn arbitrary low- and high-order feature interactions implicitly. We conduct comprehensive experiments on three real-world datasets. Our results demonstrate that xDeepFM outperforms state-of-the-art models. We have released the source code of xDeepFM at \url{https://github.com/Leavingseason/xDeepFM}.

PDF Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Click-Through Rate Prediction Bing News xDeepFM AUC 0.84 # 1
Log Loss 0.2649 # 1
Click-Through Rate Prediction Bing News DNN AUC 0.03 # 7
Log Loss 0.3382 # 5
Click-Through Rate Prediction Criteo xDeepFM AUC 0.8052 # 26
Log Loss 0.4418 # 11
Click-Through Rate Prediction Dianping xDeepFM AUC 0.8639 # 1
Log Loss 0.3156 # 1
Click-Through Rate Prediction Dianping DNN AUC 0.8318 # 5

Methods


No methods listed for this paper. Add relevant methods here