SOTAVerified

xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems

2018-03-14Code Available1· sign in to hype

Jianxun Lian, Xiaohuan Zhou, Fuzheng Zhang, Zhongxia Chen, Xing Xie, Guangzhong Sun

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

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 https://github.com/Leavingseason/xDeepFM.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
Bing NewsxDeepFMAUC0.84Unverified
Bing NewsDNNAUC0.03Unverified
CriteoxDeepFMAUC0.81Unverified
DianpingxDeepFMAUC0.86Unverified
DianpingDNNAUC0.83Unverified
KKBoxxDeepFMAUC0.85Unverified

Reproductions