DeepFM: An End-to-End Wide & Deep Learning Framework for CTR Prediction

DeepFM: An End-to-End Wide & Deep Learning Framework for CTR Prediction

Huifeng Guo, Ruiming Tang, Yunming Ye, Zhenguo Li, Xiuqiang He, and Zhenhua Dong This work is done when Huifeng Guo worked as intern at Noah’s Ark Research Lab of Huawei. Yunming Ye is the corresponding author. Huifeng Guo and Yunmming Ye are with Shenzhen Key Laboratory of Internet Information Collaboration, Shenzhen Graduate School, Harbin Institute of Technology, Shenzhen, 518055, P.R.China.
E-mail: huifengguo@yeah.net, yeyunming@hit.edu.cn Ruiming Tang, Xiuqiang He and Zhenhua Dong are with Noah’s Ark Research Lab of Huawei, Shenzhen, P.R.China.
E-mail: {tangruiming, hexiuqiang, dongzhenhua}@huawei.com Zhenguo Li is with Noah’s Ark Research Lab of Huawei, Hong Kong, P.R.China.
E-mail: li.zhenguo@huawei.comThis work has been submitted to the IEEE for possible publication. Copyright may be transferred without notice, after which this version may no longer be accessible.
Abstract

Learning sophisticated feature interactions behind user behaviors is critical in maximizing CTR for recommender systems. Despite great progress, existing methods have a strong bias towards low- or high-order interactions, or rely on 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 framework, 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 raw feature input to both its “wide” and “deep” components, with no need of feature engineering besides raw features. DeepFM, as a general learning framework, can incorporate various network architectures in its deep component. In this paper, we study two instances of DeepFM where its “deep” component is DNN and PNN respectively, for which we denote as DeepFM-D and DeepFM-P. Comprehensive experiments are conducted to demonstrate the effectiveness of DeepFM-D and DeepFM-P over the existing models for CTR prediction, on both benchmark data and commercial data. We conduct online A/B test in Huawei App Market, which reveals that DeepFM-D leads to more than 10% improvement of click-through rate in the production environment, compared to a well-engineered LR model. We also covered related practice in deploying our framework in Huawei App Market.

Recommender System, Deep Learning, Factorization Machines, CTR Prediction.

I Introduction

With the rapid development of the Internet and mobile devices, our daily activities connect closely to online services, such as online shopping, online news and videos, online social networks, and many more. Recommender systems are a powerful information filter for guiding users to find the items of interest in a gigantic and rapidly expanding pool of candidates. As elaborated in [1], if an IR system’s response to each query is a ranking of documents in decreasing order of probability of relevance, the overall effectiveness of the system to its user will be maximized. With this principle, the prediction of click-through rate (CTR) is crucial for recommender systems, where the task is to estimate the probability a user will click on a recommended item. In many recommender systems, the goal is to maximize the number of clicks, and the items returned to a user are ranked by the estimated CTR; while in other application scenarios such as online advertising it is also important to improve revenue, so the ranking strategy can be adjusted accordingly, such as by CTRbid with “bid” being the benefit the system receives once the item is clicked. In either case, the key is in estimating CTR precisely.

Fig. 1: Wide & deep architecture of the DeepFM framework (left). The wide and deep component share the same input raw feature vector, which enables DeepFM to learn low- and high-order feature interactions simultaneously from the input raw features. The wide component of DeepFM is an FM layer, which we refer to as FM Component in the paper. The Deep Component of DeepFM can be any neural network. In this paper, we will study two instances of DeepFM, namely DeepFM-D(top-right) and DeepFM-P (lower-right), the deep components of which are DNN and PNN respectively.

It is important for CTR prediction to learn implicit feature interactions behind user click behaviors. By our study in a mainstream apps market, we found that people often download apps for food delivery at meal-time, suggesting that the (order-2) interaction between app category and time-stamp can be used as a signal for CTR prediction. As a second observation, male teenagers like shooting games and RPG games, which means that the (order-3) interaction of app category, user gender and age is another signal for CTR prediction. In general, such interactions of features behind user click behaviors can be highly sophisticated, where both low- and high-order feature interactions should play important roles. According to the insights of the Wide & Deep model [2] from Google, considering low- and high-order feature interactions simultaneously brings additional improvement over the cases of considering either alone.

The key challenge is in effectively modeling feature interactions. Some feature interactions can be easily understood and engineered by experts (like the instances above). However, most other feature interactions are hidden in data and difficult to identify a priori (for instance, the classic association rule “diaper and beer” is mined from a large amount of data, instead of being discovered by experts), which can only be captured automatically by machine learning. Even for easy-to-understand interactions, it seems unlikely for experts to model them exhaustively, especially when the number of raw features is huge.

Despite their simplicity, generalized linear models, such as FTRL [3], have shown decent performance in practice. However, a (generalized) linear model lacks the ability to learn feature interactions, and a common practice is to manually include pairwise feature interactions in designing its feature vector. Such a method is hard to generalize to model high-order feature interactions or those never or rarely appear in the training data [4]. Factorization Machines (FM) [4] model pairwise feature interactions as inner product of latent vectors between features and show very promising results. While in principle FM can model high-order feature interactions, in practice usually only order-2 feature interactions are considered due to high complexity.

As a powerful approach to learning feature representations, deep neural networks have shown the potential to learn sophisticated feature interactions automatically. Some ideas extend CNN and RNN for CTR prediction [5, 6], but CNN-based models are biased to the interactions between neighboring features while RNN-based models are more suitable for click data with sequential dependency. [7] devises a deep joint learning network of a CNN and a DNN to consider both image feature and basic feature of display advertisers. [8] studies feature representations and proposes Factorization-machine supported Neural Network (FNN). This model pre-trains FM before applying DNN, thus limited by the capability of FM. Feature interaction is studied in [9], by introducing a product layer between embedding layer and fully-connected layers, and proposing the Product-based Neural Network (PNN). As noted in [2], PNN and FNN, like other deep models, capture little low-order feature interactions, which are also essential for CTR prediction. To model both low- and high-order feature interactions, [2] proposes an hybrid network structure (Wide & Deep) that combines a linear (“wide”) model and a deep model. In this hybrid model, two different inputs are required for the “wide” part and “deep” part, respectively, and the input of “wide” part still relies on feature engineering from domain experts.

One can see that existing models are biased to low- or high-order feature interactions, or rely on feature engineering. In this paper, we develop a learning framework that is able to learn feature interactions of all orders in an end-to-end manner, without any feature engineering besides raw features. Our main contributions are summarized as follows:

  • We propose a new neural network framework DeepFM (left part of Figure 1) that integrates the architectures of FM and deep neural networks in a way that models low-order feature interactions like FM and models high-order feature interactions like deep neural networks111In all figures of this paper, a Normal Connection in black refers to a connection with weight to be learned; a Weight-1 Connection, red arrow, is a connection with weight 1 by default; Embedding, blue dashed arrow, means a latent vector to be learned; Addition means adding all input together; Product, including Inner- and Outer-Product, means the output of this unit is the product of two input vector; Sigmoid Function is used as the output function in CTR prediction; Activation Functions, such as relu and tanh, are used for non-linearly transforming the signal;The yellow and blue circles in the sparse features layer represent one and zero in one-hot encoding of the input, respectively.. Unlike the wide & deep model [2], DeepFM can be trained end-to-end without any feature engineering. There is no implicit requirement on the network structure of the “deep” part of DeepFM, so various deep architectures are plausible.

  • DeepFM can be trained efficiently because its “wide” part and “deep” part share the same input and also the embedding vector. In contrast, the input vector to Google’s Wide & Deep model [2] can be of huge size as it includes manually designed pairwise feature interactions in its wide part, which also greatly increases its complexity. Moreover, no expertise feature engineering is needed in our DeepFM framework, while it is required in [2], as the performance of linear models relies heavily on feature engineering. We study two instances of DeepFM in detail, namely DeepFM-D and DeepFM-P (see the right part of Figure 1), where the “deep” part of the DeepFM framework is DNN and PNN, respectively.

  • We evaluate DeepFM-D and DeepFM-P on both benchmark data and commercial data, which shows consistent improvement over existing models for CTR prediction.

  • We conduct online A/B test in Huawei App Market, which reveals that DeepFM-D lead to more than 10% improvement of click-through rate in the production environment, compared to a well-engineered LR model. We also covered our practice in deploying our framework, such as multi-GPU data parallelism and asynchronous data reading. Extensive experiments are conducted to show the effectiveness of our proposed techniques.

The rest of the paper is organized as follows. Section II gives an overview of some related works on recommender systems. Section III presents our DeepFM framework, as well as two instances DeepFM-D and DeepFM-P of our framework. In Section IV, We extensively evaluate the effectiveness and efficiency of DeepFM-D and DeepFM-P on benchmark datasets and commercial dataset, and we conduct online A/B test in Huawei App Market to investigate if DeepFM is well suited in a production environment. Finally, we conclude our work in Section V.

Ii Related Work

In this section, we review three important categories of models in recommender systems, namely collaborative filtering models, linear models and deep learning models.

Ii-a Collaborative Filtering in Recommender Systems

The collaborative filtering (CF)-based models are well studied for recommender systems from the last decade. The basic assumption is that users with similar past behaviors will like the same kind of items, and the items attracting similar users will share similar ratings from a user. Collaborative filtering models consist of memory-based and model-based methods. The memory-based methods directly define the similarities between pairs of users and pairs of items, which are used to calculate the ratings of unknown user-item pairs [10]. Although it is easy to implement and explain, there are several limitations for memory-based methods. For instance, the similarity values are unreliable when the data is sparse and the common items are few. As a complementary part, model-based methods define a model to fit the known user-item interactions and predict the rating of unknown user-item pairs using the learned model. The most widely used model-based model is matrix factorization (MF) [11, 12]. Based on the low rank assumption and the observed user-item interactions, MF models characterize both items and users by vectors in the same space and predict an unknown rating of a user-item pair relies on the item and user vectors.

Different from CF-based models, the content-based models rely on user portrait or product information [13]. However, both CF-based and content-based models have limitations. While the former does not explicitly incorporate the users’ and items’ feature information, the latter does not necessarily consider the information in preference similarity across individuals. Therefore hybrid methods have gained popularity in recent years. In order to incorporate user-item interaction information and auxiliary information together, such as text, temporal information, location information and so on, several hybrid methods [14, 15, 16] are proposed. By learning side information and user-item interaction simultaneously, hybrid methods are able to alleviate cold-start problem and give the better recommendation.

Above models are not applied in CTR prediction very often, for reasons such as poor scalability, unsatisfactory performance on sparse data.

Ii-B Linear Models in Recommender Systems

Because of the robustness and efficiency, generalized Logistic Regression (LR) models, such as FTRL [3], are widely used in CTR prediction. To learn feature interactions, a common practice is to manually include pairwise feature interactions in its feature vector. The Poly-2 [17] model is proposed to model all order-2 feature interactions to avoid feature engineering. Factorization Machines (FM) [4] adopts a factorization technique to improve the ability of learning feature interactions when data is very sparse. Recently, to model interaction features from different fields, the authors of FFM [18] introduces field information into FM model. LR, Poly-2 and FM variants are widely used in CTR prediction in industry. In addition, a few other models are also proposed for CTR prediction, such as LR+GBDT model [19], tensor based model [20], and bayesian model [21].

Ii-C Deep Learning in Recommender Systems

Due to the powerful ability of feature learning, deep learning models have achieved great success in various areas, such as computer vision [22], natural language processing [23], audio recognition [24] and gaming [25]. In order to take advantage of its feature learning ability to enhance recommender systems, several deep learning models are proposed for recommendation (e.g., [26, 27, 28, 29, 30, 6, 31, 9, 8, 5]). These works can be divided into CTR-based and rating-based models.

CTR-based models [9, 8, 5, 6] are already mentioned in Section I and some of them will be discussed again in Section III, therefore we omit them here.

There are two kinds of rating-based deep models, CF-based and hybrid models. The CF-based models, such as [27, 32, 33, 29], are proposed to improve Collaborative Filtering via deep learning. For instance, [32] and [27] complete the rating matrix by auto encoder [34] and restricted boltzmann machine, respectively. Unlike CF-based models, the hybrid-based models [35, 28, 31] use deep learning to learn features of various domains. Specifically, [30] proposes a recurrent recommender system, which is able to capture temporal information and predict future behavioral trajectories. [31] utilizes both review information and user-item interactions. For the purpose of learning better features, [35] designs a novel end-to-end model to learn features from audio content and user-item interactions simultaneously to make personalized recommendations. In order to ease the cold start problem when recommending new and unpopular songs, [28] adds deep convolution neural network in the latent factor framework to learn audio features better.

Several models are proposed in industry. Google develops a two-stage deep learning framework for YouTube video recommendation [26]. In order to learn the relationship between image features and other features, Alibaba proposes an end-to-end deep model [7], which incorporates convolutional neural network for learning image features and multi-layer perception for other features, for CTR prediction. Readers interested in deep learning models in recommender systems can refer to a comprehensive survey work [36].

Iii Our Approach

Suppose the training data consists of instances , where is an -fields data record usually involving a pair of user and item, and is the associated label indicating user click behaviors ( means the user clicked the item, and otherwise). may include categorical fields (e.g., gender, location) and continuous fields (e.g., age). Each categorical field is represented as a one-hot vector, and each continuous field is represented as the value itself, or a one-hot vector after discretization. Thus, each instance is converted to where is a -dimensional vector, with being the vector representation of the field of . Normally, is high-dimensional and extremely sparse. The task of CTR prediction is to develop a prediction model to estimate the probability of a user clicking a specific item in a given context.

Iii-a DeepFM

To learn both low- and high-order feature interactions, we propose an end-to-end deep learning framework for CTR prediction, namely Factorization-Machine based neural network (DeepFM). As depicted in Figure 1, DeepFM consists of two components, FM Component and Deep Component, that share the same input. For feature , a scalar is used to weigh its order-1 importance, a latent vector is used to measure its impact of interactions with other features. is fed in FM component to model order-2 feature interactions, and fed in deep component to model high-order feature interactions. All parameters, including , , and the network parameters (, below) are trained jointly for the combined prediction model:

(1)

where is the predicted CTR, is the output of FM component, and is the output of deep component. We also present two instances of DeepFM framework in Figure 1, namely DeepFM-D and DeepFM-P, whose deep component is DNN and PNN, respectively. The prediction formulae of DeepFM-D and DeepFM-P update Equation 1 by setting and , respectively. The definitions of , and are introduced in the following sections.

Iii-A1 FM Component of DeepFM

Fig. 2: The architecture of FM.

The FM component is a factorization machines, which is proposed in [4] to learn feature interactions for recommendation. Besides a linear (order-1) interactions among features, FM models pairwise (order-2) feature interactions as inner product of respective feature latent vectors. It can capture order-2 feature interactions much more effectively than previous approaches (such as LR and Poly-2 [17]) especially when the data is sparse. For comparison, we show the prediction models for LR and Poly-2:

(2)
(3)

where and are the prediction of LR model and Poly-2 models, and is a function encoding and into a natural number. LR models linear combination of the features and some order-2 feature interactions that are selected by experts (i.e., and are picked by human). To avoid feature engineering, Poly-2 chooses to model all possible order-2 feature interactions. In these two approaches, each feature interaction is assigned with a parameter, so that the number of parameters in the model is huge. Moreover, the parameter of an interaction of features and can be learned only when feature and feature both appear in a sample.

While in FM, it is measured via the inner product of their latent vectors and . Thanks to this flexible design, FM can train latent vector () whenever (or ) appears in a data record. Therefore, feature interactions, which are never or rarely appeared in the training data, are better learned by FM. As Figure 2 shows, the output of FM is the summation of an Addition unit and a number of Inner Product units:

(4)

where and ( is given)222We omit a constant offset for simplicity.. The Addition unit () reflects the importance of order-1 features, and the Inner Product units represent the impact of order-2 feature interactions. As presented in Equation 1, the output of FM component is part of the final CTR prediction.

Iii-A2 Deep Component of DeepFM

The deep component is a feed-forward neural network, which is used to learn high-order feature interactions. A data record (a vector) is fed into the neural network. Compared to neural networks with image [22] or audio [24] data as input, which is purely continuous and dense, the input of CTR prediction is quite different, which requires a new network architecture design. Specifically, the raw feature input vector for CTR prediction is usually highly sparse333Only one entry is non-zero for each field vector., super high-dimensional444E.g., in an app store of billion users, the one field vector for user ID is already of billion dimensions., categorical-continuous-mixed, and grouped in fields (e.g., gender, location, age). This suggests an embedding layer to compress the input vector to a low-dimensional, dense real-value vector before further feeding into the first hidden layer, otherwise the network can be overwhelming to train.

Fig. 3: The structure of the embedding layer.

Figure 3 highlights the network structure from the input layer to the embedding layer. We would like to point out two interesting designs of this network structure: 1) while input field vectors can be of different sizes, their embeddings are of the same size (); 2) the latent feature vectors () in FM now serve as network weights which are learned and used to compress the input field vectors to the embedding vectors. In [8], is pre-trained by FM and used as initialization. In this work, rather than using the latent feature vectors of FM to initialize the networks as in [8], we include the FM model as part of our overall learning architecture. As such, we eliminate the need of pre-training by FM and instead jointly train the entire network in an end-to-end manner.

Denote the output of the embedding layer as:

(5)
(6)

where is the embedding of the -th field, is the number of fields, is the parameters between the embedding layer and the input layer of the field (as shown in Figure 3), is the one-hot vector of the field raw input data.

It is worth pointing out that FM component and deep component share the same feature embedding, which brings two important benefits: 1) it learns both low- and high-order feature interactions from raw features; 2) there is no need for expertise feature engineering of the input, as required in Google Wide & Deep model [2].

Note that in the proposed DeepFM framework, there is no implicit requirement on the network structure of the deep component. In this section, we show only a general deep component of DeepFM. In the next sections, we will present in detail the network structure of the deep component of two instances of the DeepFM framework, called DeepFM-D and DeepFM-P.

Iii-A3 Deep Component of DeepFM-D Model

Fig. 4: The architecture of the deep component of DeepFM-D.

The deep component of DeepFM-D is a fully-connected deep neural network (DNN, or equivalently Multilayer Perceptron). The structure of DNN is presented in Figure 4.

In such network structure, output of the embedding layer is fed into the deep neural network, and the forward process is:

(7)

where is the layer depth and is an activation function. , , are the output, model weight, and bias of the -th layer, respectively. After going through several hidden layers, a dense real-value feature vector is generated as,

(8)

where is the number of hidden layers. This feature vector is finally fed into the sigmoid function for CTR prediction, as described in Equation 1.

Iii-A4 Deep Component of DeepFM-P Model

The deep component of DeepFM-P is a product based neural network (PNN) [9]. As presented in [9], there are three variants of PNN models, i.e., IPNN, OPNN, PNN. PNN introduces a product layer between the embedding layer and the first hidden layer (the middle part of Figure 8). The three variants differ from each other in defining different product operations between features as feature interactions. More specifically, IPNN uses inner product, OPNN uses outer product, and PNN uses both inner and outer product. The details of three PNN variants are presented as follows.

In the product layer of PNN model (the middle part of Figure 8), it consists of two parts: linear neurons (right part of the layer) and quadratic neurons (“product symbol” in the left part of the layer). Linear neurons are the concentration of the embedding vectors of all fields, while quadratic neurons are the products of embedding vectors from a pair of fields. The output unit of the first hidden layer is

(9)

In Equation 9, is the output vector of linear neurons in the product layer, which is embedding vectors of different fields themselves. is the output vector of quadratic neurons, which includes the interactions between any two embedding vectors and . are parameters between the product layer and the first hidden layer connecting to linear neurons and quadratic neurons respectively. The three variants of PNN define function differently: IPNN and OPNN define to be the inner product and outer product of two vectors respectively, while PNN considers both inner and outer product. Finally, going through several fully-connected hidden layers (as defined in Equation 7), has a similar output value as (as defined in Equation 8).

Iii-B Practical Issues

Iii-B1 Learning

  • Objective function: In the domain of CTR prediction, the most commonly used objective function is Logloss, which is equivalent to the K-L divergence between two distributions:

    (10)
    (11)

    where is a data instance, is the feature vector and is the label, is the prediction of the instance , is the Logloss of , and is the Logloss of dataset .

  • Overfitting: In machine learning, one important issue is to prevent overfitting. An overfit model has poor performance since it overreacts to the given training data. The authors of [37] state that FM may suffer from overfitting and thus they utilize L2-norm to regularize the objective function. On the other hand, as a complicated model, neural networks are also easy to overfit. The authors of [38] propose a simple strategy to prevent neural networks from overfitting, which is known as dropout. Therefore, in our DeepFM framework, we adopt L2-norm to regularize the FM component and adopt dropout for the deep component.

Iii-B2 Accelerating Strategy

  • Multi-GPU Architecture: In real applications, the amount of training data is so huge that the training process has to take a long time. For the purpose of accelerating this process, we utilize the multi-GPU data parallelism when deploying DeepFM in the production environment (as presented in Figrue 5). At first, we split a batch of data records into Num_GPU pieces (Num_GPU is the number of GPU cards) and feed each piece into different GPU cards simultaneously. Then, the gradient of data records in different pieces is computed by individual GPU card. After that, the gradient is collected and averaged. Finally, the model parameters are updated by the averaged gradient.

    Fig. 5: The Multi-GPU architecture.
    Fig. 6: Performance comparison of multi-GPU data parallelism for DeepFM-D. Note: 1GPU means single GPU; 4GPU means 4 GPUs; 4GPU-A means 4 GPUs with adjusted .

    We evaluate the effectiveness of multi-GPU data parallelism for DeepFM-D model on Company data set, and the test curves of AUC and Logloss related with different settings are shown in Figure 6. Specifically, the batch size of different settings are same, the learning rate of 1GPU, 4GPU and 4GPU-A are 0.0001, 0.0001 and 0.0001, respectively. Compared with 1GPU, the test curves of 4GPU indicate that the training process of 4GPU is slower. It is because the number of updates of 4GPU is only a quarter of 1GPU when adopting same . As a result, it converges slower than 1GPU if we set same . In fact, the variance of the gradient in a mini-batch can be denoted as following,

    where is the gradient of a randomly selected instance. The reason for the second equal sign is that the variance of the gradient related to the randomly selected instances is equal to each other [39]. So the variance of the gradient decreases times when we increase the batch size by times. In other word, the gradient becomes more accurate. Then we can increase to accelerate the training process. Add into the equation of the gradient’s variance:

    Therefore, when using 4 GPU cards, we can increase the value of by times. As a result, the learning curve of 4GPU-A in Figure 6 is similar as that of 1GPU.

    Fig. 7: The comparison of different strategies for loading data.
  • Asynchronous Data Reading: Training a neural network is usually in a mini-batch style, in which a mini-batch of data records (e.g., several thousand) are read and fed into the neural network, and then model parameters are updated in a forward-backward way. There are two possible ways to handle data reading and model updating: sequential and parallel, as shown in Figure 7. In the sequential approach, data reading and model updating are processed interleaved and sequentially, i.e., model updating starts when the current mini-batch of data is read and fed, and the next mini-batch of data will not be read until the model finishes updating with the current mini-batch of data. Obviously, this is not an efficient way. We propose a parallel manner to handle data reading and model updating, namely, asynchronous data reading. A thread is created to read the data records regardless of the model updating, so that the model parameters keep updating without interrupted by reading data.

  • Efficiency Validation Experiment:

    As shown in Table I, we record the speed up rate555In this paper, we define the speed up rate of strategy A over strategy B to be the processing time of strategy B divided by the process time of strategy A. of the 4GPU over 1GPU data parallelism, asynchronous (Asyn) over synchronous (Syn) data reading of DeepFM models on Company and Criteo data sets. We omit the validation result on Criteo-Sequential data set, since Criteo-Sequential and Criteo-Random come from the same original data set with different splitting strategies.

    Company Criteo-Random
    4GPU over 1GPU 2.25 X 2.15 X
    Asyn over Syn 1.12 X 1.19 X
    TABLE I: Speed up rate for DeepFM-D model.

Iii-C Relationship with Other Neural Networks

Inspired by the enormous success of deep learning in various applications, several deep models for CTR prediction are developed recently. This section compares the proposed DeepFM-D and DeepFM-P models with existing deep models for CTR prediction.

Fig. 8: The architectures of existing deep models for CTR prediction: FNN, PNN, Wide & Deep Model.

Iii-C1 Fnn

As Figure 8 (left) shows, FNN is an FM-initialized feed-forward neural network [8]. The FM pre-training strategy results in two limitations: 1) the embedding parameters might be over affected by FM; 2) the efficiency is reduced by the overhead introduced by the pre-training stage. In addition, FNN captures only high-order feature interactions. In contrast, DeepFM-D and DeepFM-P models need no pre-training and learn both high- and low-order feature interactions in an end-to-end manner.

There is a detailed issue about embedding vectors among different models that needs to be metioned. As shown in Figure 1 of [8], each embedding vector of a field includes both latent vector of this field and an additional neuron representing the weights of the feature values in this field. In other words, if the dimension of latent vectors in FM is , then the latent vectors are of size in FNN. In our experiments, in order to keep the same representation ability, PNN and Wide & Deep use the same size of embedding vectors as FNN. On the other side, DeepFM models have an FM component to model the weights of individual feature values, therefore there is no need to include an additional neuron in the embedding vector as FNN does. Due to this reason, the size of embedding vectors in DeepFM models is smaller than that in FNN by one.

Iii-C2 Pnn

For the purpose of capturing high-order feature interactions, PNN imposes a product layer between the embedding layer and the first hidden layer [9]. According to different types of product operations, there are three variants: IPNN, OPNN, and PNN, where IPNN is based on inner product of vectors, OPNN is based on outer product, and PNN is based on both inner and outer products. To make the computation more efficient, the authors proposed the approximated computations of both inner and outer products: 1) the inner product is approximately computed by eliminating some neurons; 2) the outer product is approximately computed by compressing -dimensional feature vectors to one -dimensional vector. However, we find that the outer product is less reliable than the inner product, since the approximated computation of outer product loses much information that makes the result unstable. Although inner product is more reliable, it still suffers from high computational complexity, because the output of the product layer is connected to all neurons of the first hidden layer. Like FNN, all PNNs ignore low-order feature interactions.

Iii-C3 Wide & Deep

Wide & Deep (Figure 8 (right)) is proposed by Google to model low- and high-order feature interactions simultaneously. As shown in [2], there is a need for expertise feature engineering on the input to the “wide” part (for instance, cross-product of users’ install apps and impression apps in app recommendation). In contrast, DeepFM-D and DeepFM-P need no such expertise knowledge to handle the input by learning directly from the input raw features.

A straightforward extension to this model is replacing LR by FM (we also evaluate this extension in Section IV). This extension is similar to DeepFM-D, but DeepFM framework shares the feature embedding between the FM component and deep component. The sharing strategy of feature embedding influences (in back-propagate manner) the feature representation by both low- and high-order feature interactions, which models the representation more precisely.

Iii-C4 Summarizations

To summarize, the relationship between DeepFM framework and the other deep models in four aspects is presented in Table II. DeepFM is the only framework that requires no pre-training and no feature engineering, and captures both low- and high-order feature interactions.

No High-order Low-order No Feature
Pre-training Features Features Engineering
FNN
PNN
Wide & Deep
DeepFM-D (P)
TABLE II: Comparison of deep models for CTR prediction.

Iv Experiments

In this section, we conduct both offline and online experiments to evaluate our proposed DeepFM framework.

In the offline experiments, we compare two instances of our proposed DeepFM framework (namely, DeepFM-D and DeepFM-P) with the other state-of-the-art models empirically. The evaluation result indicates that DeepFM-D and DeepFM-P are more effective than any other state-of-the-art model. The efficiency tests of DeepFM-D, DeepFM-P and the other baseline models are also performed.

In the online experiments, we conduct a consecutive seven days’ A/B test to evaluate the performance of DeepFM framework. In these DeepFM models, DeepFM-D has a relative better efficiency and performance. Therefore, we adopt DeepFM-D as our model to compare with a well-engineered LR model, which is one of the most popular CTR prediction model in industry. In addition, to understand the result of A/B test better, we compare the recommendation lists generated by LR and DeepFM-D through the online simulation experiment.

Iv-a Setup of Offline Experiments

Iv-A1 Data sets

We evaluate the effectiveness and efficiency of our DeepFM-D and DeepFM-P models on the following three data sets.

  • Criteo Data set: Criteo data set666http://labs.criteo.com/downloads/2014-kaggle-display-advertising-challenge-dataset/ includes 45 million users’ click records. There are 13 continuous features and 26 categorical ones. We split the data set in two different ways: randomly and sequentially, resulting in Criteo-Random and Criteo-Sequential. To get Criteo-Random data set, the original Criteo data set is randomly split into two parts as: 9/10 is for training, while the rest 1/10 is for testing. To get Criteo-Sequential data set, the original data set is split sequentially as: the first 6/7 is for training, while the rest 1/7 is for testing, as the original data set consists of data instances of 7 consecutive days. In Criteo-Sequential data set, information is not leaked, however data is significantly biased between training set and test set, as training set contains only six days’ records instead of one week’s records. On the contrary, in Criteo-Random data set, information may be leaked but it is not significant biased between training set and test set.

  • Company Data set: Company data set is a commercial industrial data set. We collect 8 consecutive days of users’ click records from the game center of the Company App Store: the first 7 days’ records for training, and the next 1 day’s records for testing. There are around 1 billion records in the whole collected dataset. In this dataset, there are app features (e.g., identification, category, and etc), user features (e.g., user’s downloaded apps, and etc), and context features (e.g., operation time, and etc).

Iv-A2 Evaluation Metrics

We use two evaluation metrics in our experiments: AUC (Area Under ROC Curve) and Logloss (Logistic loss).

AUC and Logloss are two of the most commonly used evaluation metrics for binary-class classification problem. For such machine learning models of binary-class classification, the prediction is a probability value that the given data record belongs to a certain class. AUC and Logloss are more suitable than precision and recall, it is because that when computing precision and recall, a user-defined threshold is needed to convert a probability value to a class label and the choice of the threshold value affects the accuracy and recall significantly. However, AUC and Logloss avoid such user-defined threshold values.

AUC is equal to the probability that a classifier will rank a randomly chosen positive instance higher than a randomly chosen negative one (assuming positive ranks higher than negative) [40]. Logloss (or Cross Entropy) measures the distance between two distributions, one of which is predicted by the model while the other is given by the labels of the data instances. Note that Logloss is the objective function of our proposed model at the same time. The formula of Logloss is presented in Equation 11.

Iv-A3 Model Comparison

In our experiment, we compare the performance of 12 models, which are divided into four categories: Wide models, Deep models, Wide & Deep models and DeepFM models.

  • Wide models: LR, FM.

  • Deep models: DNN, FNN, PNN. There are three variants of PNN, namely IPNN, OPNN and PNN.

  • Wide & Deep models: The original Wide & Deep model is discussed in Section III-C. For the purpose of eliminating feature engineering effort, we adapt the original Wide & Deep model by replacing LR by FM as the wide part. In order to distinguish these two variants of Wide & Deep, we name them LR & DNN and FM & DNN, respectively.777We do not use the Wide & Deep API released by Google, as the efficiency of that implementation is very low. We implement Wide & Deep by ourselves by simplifying it with shared optimizer for both deep and wide part.

  • DeepFM models: DeepFM-D and DeepFM-P. Our DeepFM-P also has three variants, denoted as DeepFM-IP, DeepFM-OP and DeepFM-P, of which the deep components are the three variants of PNN accordingly.

Iv-A4 Parameter Settings

To achieve the best performance for each individual model on Criteo-Random, Criteo-Sequential and Company data sets, we conducted carefully parameter study on all the data sets. Due to the space limit, we only discuss the parameter study on Company data set, which is presented in Section IV-C. The hyper-parameters of compared deep models on Criteo-Sequential and Criteo-Random data sets are stated in Table III, where the activation function, dropout and structure of hidden layers are given. The optimizer for LR and others are FTRL [3] and Adam [41], respectively. The embedding dimensions of FM and DeepFM models are 10, and others are 11 (discussed in Section III-C). Note that hyper-parameters of baseline models on Criteo-Random data set follow the setting in [9], and we keep the deep components of DeepFM models with the same setting to validate the superiority of our models.

Model Criteo-Random Criteo-Sequential
DNN relu;0.5;400-400-400. relu;0.8;800-800-800-800-800.
FNN relu;0.5;400-400-400. relu;0.9;1100-1000-900-800-700-600-500.
IPNN tanh;0.5;400-400-400. tanh;0.8;800-800-800.
OPNN relu;0.5;400-400-400. relu;0.9;800-800-800-800-800.
PNN relu;0.5;400-400-400. relu;0.8;800-800-800-800-800.
LR & DNN relu;0.5;400-400-400. relu;1.0;1100-1000-900-800-700-600-500.
FM & DNN relu;0.5;400-400-400. tanh;0.7;1000-900-800-700-600.
DeepFM-D relu;0.5;400-400-400. relu;0.9;800-800-800-800-800-800-800.
DeepFM-IP relu;0.5;400-400-400. relu;0.8;800-800-800.
DeepFM-OP relu;0.5;400-400-400. relu;0.9;800-800-800-800-800.
DeepFM-p relu;0.5;400-400-400. relu;0.8;800-800-800-800-800.
TABLE III: Hyper-parameters of deep models on Criteo.

Iv-B Performance of Offline Evaluation

In this section, we evaluate the efficiency and effectiveness of the models listed in Section IV-A3 on the three data sets.

Iv-B1 Efficiency Comparison

The efficiency of deep learning models is important to real-world applications. We compare the efficiency of different models on Company data set by the following formula: , which is the normalized running time by LR model. The results are shown in Figure 9, including the tests on CPU and GPU, where we have the following observations:

  • Pre-training of FNN makes it less efficient, especially on GPU, since the pre-training by FM model is not suitable for accelerating by GPU.

  • IPNN, PNN, DeepFM-IP and DeepFM-P are the least efficient models, on both CPU and GPU. Although their speed up on GPU is higher than the other models, they are still computationally expensive because of the inefficient inner product operations.

  • DNN and OPNN are the most efficient models, on both CPU and GPU. Testing on GPU shows a much more obvious gap between these two models and other models.

  • FNN, FM & DNN, LR & DNN, DeepFM-D and DeepFM-OP have similar efficiency on both CPU and GPU.

In real industry applications, we are equipped by powerful servers with GPUs. The efficiency of DeepFM-D and DeepFM-OP is acceptable for us, since they are only 41% and 14% slower than LR model on GPU.

Fig. 9: Running time comparison between CPU and GPU.

Iv-B2 Effectiveness Comparison

AUC LogLoss
Company Criteo-Random Criteo-Sequential Company Criteo-Random Criteo-Sequential
LR 0.8641 0.7804 0.7777 0.02648 0.46782 0.4794
FM 0.8679 0.7894 0.7843 0.02632 0.46059 0.4739
DNN 0.8650 0.7860 0.7953 0.02643 0.4697 0.4580
FNN 0.8684 0.7959 0.8038 0.02628 0.46350 0.4507
IPNN 0.8662 0.7971 0.7995 0.02639 0.45347 0.4543
OPNN 0.8657 0.7981 0.8002 0.02640 0.45293 0.4536
PNN 0.8663 0.7983 0.8005 0.02638 0.453305 0.4533
LR & DNN 0.8671 0.7858 0.7973 0.02635 0.46596 0.4565
FM & DNN 0.8658 0.7980 0.7985 0.02639 0.45343 0.4551
DeepFM-D 0.8715 0.8016 0.8048 0.02619 0.44985 0.4497
DeepFM-IP 0.8720 0.8019 0.8019 0.02616 0.4496 0.4525
DeepFM-OP 0.8713 0.8008 0.8020 0.02619 0.4510 0.4524
DeepFM-P 0.8716 0.7995 0.8015 0.02619 0.4515 0.4530
TABLE IV: Performance (in terms of AUC and Logloss) of all the compared models.

The performance (in terms of AUC and Logloss) of the compared models on Criteo-Random data set, Criteo-Sequential data set and Company data set is presented in Table IV (the values in the table are averaged by 5 runs, and the variances of AUC and Logloss are in the order of ). The following conclusions are observed:

  • Learning feature interactions improves the performance. LR, the only model that does not consider feature interactions, performs worse than the other models. As the best models, DeepFM models outperform LR by 0.91%, 2.75% and 3.48% in terms of AUC (1.21%, 3.89% and 6.2% in terms of Logloss) on Company, Criteo-Random and Criteo-Sequential data sets respectively.

  • The performance of a DeepFM model is better than the model that keeps only the FM component or keeps only the Deep component. That is to say, the performance of DeepFM-D (DeepFM-IP, DeepFM-OP, DeepFM-P, respectively) is better than both FM and DNN (IPNN, OPNN, PNN, respectively). Table V presents performance improvement of the four DeepFM models over FM component and their deep components on the three data sets.

  • Learning high- and low-order feature interactions simultaneously and properly improves the performance. Among DeepFM models, DeepFM-D and DeepFM-IP perform the best. DeepFM-D and DeepFM-IP outperform the models that learn only low-order feature interactions (namely, LR and FM) or high-order feature interactions (namely, FNN, IPNN, OPNN, PNN). Compared with the best baseline that learns high- or low-order feature interactions alone, DeepFM-D and DeepFM-IP achieve more than 0.41%, 0.45% and 0.12% in terms of AUC (0.46%, 0.82% and 0.22% in terms of Logloss) on Company, Criteo-Random and Criteo-Sequential data sets.

  • Learning high- and low-order feature interactions simultaneously while sharing the same feature embedding for high- and low-order feature interactions learning improves the performance. DeepFM-D outperforms the models that learn high- and low-order feature interactions using separate feature embeddings (namely, LR & DNN and FM & DNN). DeepFM-D achieves more than 0.48%, 0.44% and 0.79% in terms of AUC (0.58%, 0.80% and 1.2% in terms of Logloss) on Company, Criteo-Random and Criteo-Sequential data sets, respectively.

Wide Deep
DeepFM-D DeepFM-IP DeepFM-OP DeepFM-P DeepFM-D DeepFM-IP DeepFM-OP DeepFM-P
Company AUC 0.4% 0.47% 0.39% 0.43% 0.75% 0.67% 0.65% 0.61%
LogLoss 0.49% 0.61% 0.49% 0.49% 0.91% 0.87% 0.80% 0.72%
Criteo-Random AUC 1.54% 1.58% 1.44% 1.28% 1.98% 0.60% 0.34% 0.15%
LogLoss 2.33% 2.39% 2.08% 1.97% 4.22% 0.85% 0.43% 0.40%
Criteo-Sequential AUC 2.61% 2.24% 2.26% 2.19% 1.19% 0.30% 0.22% 0.12%
LogLoss 5.1% 4.52% 4.54% 4.41% 1.81% 0.40% 0.26% 0.07%
TABLE V: The imporvement of DeepFM models over its wide component (namely FM model) and deep component.

Overall, our proposed four DeepFM models perform better than the baseline models in all the cases. In particular, our proposed DeepFM-D888Although DeepFM-IP performs slightly better than DeepFM-D on Company dataset, we will still choose DeepFM-D in our real scenario to avoid the high time complexity of DeepFM-IP. model beats the competitors by more than 0.36% and 0.34% in terms of AUC and Logloss on Company data set. In fact, a small improvement in offline AUC evaluation is likely to lead to a significant increase in online CTR. As reported in [2], compared with LR, Wide & Deep improves AUC by 0.275% (offline) and the improvement of online CTR is 3.9%. Moreover, we also conduct t-test between our proposed DeepFM models and the baseline models on the three data sets. We find that the p-values are all less than , which indicates that our improvement over existing models is significant.

Iv-C Offline Hyper-Parameter Study

We study the impact of different hyper-parameters of different deep models, on Company dataset. The order is: 1) activation functions; 2) dropout rate; 3) number of neurons per layer; 4) number of hidden layers; 5) network shape. It can be clearly observed from the following sections that our proposed DeepFM models are significantly superior, compared with the baseline models, in all the studied cases.

Iv-C1 Activation Function

According to [9], relu and tanh are more suitable for deep models than sigmoid. The detailed discussion of different activation functions is presented in Section III-B. In this paper, we compare the performance of deep models when applying relu and tanh as the activation function. As shown in Figure 10, relu is more appropriate than tanh for all the deep models, except for IPNN, due to the reason stated in Section III-B.

Fig. 10: Performance comparison of activation functions in different models.

Iv-C2 Dropout

Dropout [38] refers to the probability that a neuron is kept in the network. Dropout is a regularization technique to compromise the precision and the complexity of the neural network. We set the dropout to be 1.0, 0.9, 0.8, 0.7, 0.6, 0.5. As shown in Figure 11, all the models are able to reach their own best performance when the dropout is properly set (from 0.6 to 0.9). The result shows that adding reasonable randomness to model can strengthen model’s robustness and generalization.

Fig. 11: Performance comparison of dropout values in different models.

Iv-C3 Number of Neurons per Layer

When other factors remain the same, increasing the number of neurons per layer introduces complexity. When we study the effect of number of neurons per layer on the performance, we set the number of hidden layers to 3 and keep the number of neurons the same for each hidden layer. As observed from Figure 12, increasing the number of neurons does not always bring benefit. For instance, DeepFM-D performs stably when the number of neurons per layer is increased from 400 to 800; even worse, OPNN performs worse when we increase the number of neurons from 400 to 800. This is because an over-complicated model is easy to overfit. In our dataset, 200 or 400 neurons per layer is a good choice.

Fig. 12: Performance comparison of number of neurons.

Iv-C4 Number of Hidden Layers

Varying the number of hidden layers, the number of neurons for each hidden layer is fixed. As presented in Figure 13, increasing the number of hidden layers improves the model performance at the beginning, however, their performance is degraded if the number of hidden layers keeps increasing, because of overfitting.

Fig. 13: Performance comparison of number of layers in different models.

Iv-C5 Network Shape

We test four different network shapes: constant, increasing, decreasing, and diamond. When we change the network shape, we fix the number of hidden layers and the total number of neurons. For instance, when the number of hidden layers is 3 and the total number of neurons is 600, then four different shapes are: constant (200-200-200), increasing (100-200-300), decreasing (300-200-100), and diamond (150-300-150). As we can see from Figure 14, the “constant” network shape is empirically better than the other three options, which is consistent with previous studies [42].

Fig. 14: Performance comparison of network shape in different models.

Iv-D Online Experiments

According to the results of the offline experiments on Criteo-sequential, Criteo-random and Company datasets, the DeepFM models have shown their superior effectiveness over the other existing models in terms of AUC and Logloss. In order to verify the superior of DeepFM in a production environment, we implement and deploy DeepFM-D in the recommendation engine of Huawei App Market, which is one of the most popular Android App Markets in China.

Furthermore, we conduct two kinds of experiments to reveal the discriminative power of DeepFM-D compared to LR model, in CTR prediction task.

  • A/B test: Besides offline evaluation, it is more valuable to verify that whether DeepFM-D is able to perform its superiority as well in the real production environment of Huawei App Market. Therefore we conduct a consecutive seven days’ A/B test, to test the performance of DeepFM-D, against a well-engineered LR model.

  • Online simulation: From a model perspective, compared with LR, DeepFM-D is able to capture high-order feature interactions, resulting in highly personalized recommendation. We aim to verify this statement with an online simulation, by analyzing the difference between recommendation results generated by DeepFM-D and LR.

In the following of this section, we give a brief description of the experiment settings in Section IV-D1, then we present the results of online A/B test and online simulation in Section IV-D2 and Section IV-D3, respectively.

Iv-D1 Setting

In this section, we present the settings of A/B test and online simulation, including the experiment set up and evaluation metrics.

  • A/B test: Considering the project launching schedule, we split the users into 2 groups, one group receives the recommendation by an LR model, which is one of the most popular CTR prediction model in industry; the other one gets the recommendation by DeepFM-D. The update frequencies of DeepFM and LR models are both on the daily basis. The A/B test is conducted on “fun games” scenario in Huawei App Market. There are hundreds of millions of real users in Huawei App Market from whom consent has been acquired. After online A/B test in consecutive seven days, we collect the number of browsing and downloading records for both user groups in each day. We use CTR (Click Through Rate) and CVR (ConVersion Rate) as evaluation metrics:

    (12)
    (13)

    where is the number of download records, is the number of browsing records, and is the number of visited users.

  • Online simulation: Online simulation analyzes the properties of recommendation lists generated by LR and DeepFM-D. In order to study the difference between the recommendation results by LR and DeepFM-D in terms of personalization and diversity, we compare the cases for different types of users. Differentiated by users’ downloading history, we generate types of users, and each type includes users. The user set is denoted as , where represents the user of type and represents all the users of type . A user of type is generated by sampling several apps of type as the user’s downloading history. Then for user set , we use the trained LR model (and DeepFM-D model) to generate recommendation list , where and is for user . Based on the recommendation lists, we adopt personalization@[43], coverage@L and popularity@[44] as the evaluation metrics to investigate the differences between LR and DeepFM-D model.

    • The personalization@L considers the diversity of Top- places in different users’ recommendation lists. We define the inter-list distance between recommendation lists of user and user as

      (14)

      where is the common items in the Top- places of both lists. The inter-group distance between recommendation lists of user group and is defined as the aggregated inter-list distances between the users across and , as

      (15)

      where is cardinality of user group . Finally, the personalization of recommendation lists by a model is defined as aggregated inter-group distances between all pairs of user groups, as

      (16)
    • The coverage@L considers the percentage of recommended apps in Top- places of all the users’ recommendation lists over all the candidate apps.

      (17)

      where is the recommended apps in Top- places in .

    • In addition, we also measure the popularity@L, which is defined as:

      (18)

      where is the number of historical cumulative downloads of the recommended app in , is the number of historical cumulative downloads of the most downloaded app.

Iv-D2 Performance of Online Experiments

In this section, we present the results of online A/B test. Because of commercial concerns, we only report the improvements of DeepFM-D over LR in terms of CTR and CVR, as shown in Figure 15. The x-axis represents different days, and the y-axis is the improvement of DeepFM-D over LR. Note that the blue bar with slash line represents the improvement of CTR, while the red bar with horizonal line represents the improvement of CVR.

The histograms shows that the performance of DeepFM-D is consistently better than LR, through the whole A/B testing period. Specifically, the improvements of DeepFM-D over LR are at least 10% in terms of CTR and CVR, except for the CVR on day-7 (which is still very close to 10%). In addition, the highest improvement of CTR reaches about 24% on the day-4 and the maximum of CVR improvement is about 25% on day-1. The online A/B test results reveal that DeepFM-D leads to a higher CTR and CVR over LR in a recommendation engine of industry scale.

Fig. 15: The online CTR and CVR improvement of DeepFM-D over LR.

Iv-D3 The Property of Online Recommendations

To better understand the results of the online experiments, we conduct online simulation experiments and make a comparison of personalization@L, coverage@L and popularity@L () between the recommendation lists by LR and DeepFM-D. The results are presented in Figure 16, Figure 17 and Figure 18, respectively. In these three figures, the blue bar with slash line and the red bar with horizonal line represent the measurement of recommendation lists by LR and DeepFM-D, respectively.

As shown in Figure 16 and Figure 17, the personalization@L and coverage@L of recommendation lists generated by DeepFM-D are much larger than that of LR. The personalization@L is the aggregated inter-group distance between recommendation lists of different user groups, therefore a low personalization@L means Top- places in the recommendation lists of different users are similar. The coverage@L has similar semantics as the personalization@L. When coverage@L is low, the Top- places in the recommendation lists concentrate in a small range of apps. The results of personalization@L and coverage@L demonstrate that the Top- places in recommendation lists of DeepFM-D are more diverse than that of LR.

Fig. 16: The comparison of personalization of Top-L recommendation lists between DeepFM-D and LR.
Fig. 17: The comparison of coverage of Top-L recommendation lists between DeepFM-D and LR.

Figure 18 presents the comparison of popularity@L of recommendation lists generated by LR and DeepFM-D. There are two indices in this experiment, the average and the variance of the historical cumulative downloads of the apps contained in the Top- places in recommendation lists. Specifically, the black line on the top of each bar is the variance of the 600 recommendation lists (100 users per type 6 types). Compared with DeepFM-D, LR generates the recommendation lists with higher average popularity and lower variance. That is to say, LR model trends to recommend the popular apps in top positions and is more likely to ignore the specific interest of different users. In contrast, due to the superior ability on capturing feature interactions, DeepFM-D is able to capture specific interests of different users better.

Fig. 18: The comparison of average degree and standard variance of Top-L recommendation lists between DeepFM-D and LR.

V Conclusions

In this paper, we proposed DeepFM, an end-to-end wide & deep learning framework for CTR prediction, to overcome the shortcomings of the state-of-the-art models. DeepFM trains a deep component and an FM component jointly. It gains performance improvement from these advantages: 1) it does not need any pre-training; 2) it learns both high- and low-order feature interactions; 3) it introduces a sharing strategy of feature embedding to avoid feature engineering. We studied two instances of DeepFM framework, namely DeepFM-D and DeepFM-P, of which the deep component are DNN and PNN, respectively. The offline experiments on three real-world data sets demonstrate that 1) our proposed DeepFM-D and DeepFM-P outperform the state-of-the-art models in terms of AUC and Logloss on all the three datasets; 3) As one of the best performed model, DeepFM-D has comparable efficiency as LR model on GPU, which is acceptable in industrial applications.

To verify the superiority of DeepFM framework in production environment, we deployed DeepFM-D in the recommendation engine of Huawei App Market. We also covered related practice in deploying our framework, such as multi-GPU architecture and asynchronous data reading. Compared with a well-engineered LR model, which is one of the most popular CTR prediction models, DeepFM-D achieves more than 10% improvement of CTR in online A/B test.

References

  • [1] C. L. A. Clarke, M. Kolla, G. V. Cormack, O. Vechtomova, A. Ashkan, S. Büttcher, and I. MacKinnon, “Novelty and diversity in information retrieval evaluation,” in ACM SIGIR, 2008, pp. 659–666.
  • [2] H. Cheng, L. Koc, J. Harmsen, T. Shaked, T. Chandra, H. Aradhye, G. Anderson, G. Corrado, W. Chai, M. Ispir, R. Anil, Z. Haque, L. Hong, V. Jain, X. Liu, and H. Shah, “Wide & deep learning for recommender systems,” CoRR, vol. abs/1606.07792, 2016.
  • [3] H. B. McMahan, G. Holt, D. Sculley, M. Young, D. Ebner, J. Grady, L. Nie, T. Phillips, E. Davydov, D. Golovin, S. Chikkerur, D. Liu, M. Wattenberg, A. M. Hrafnkelsson, T. Boulos, and J. Kubica, “Ad click prediction: a view from the trenches,” in ACM SIGKDD, 2013.
  • [4] S. Rendle, “Factorization machines,” in ICDM.
  • [5] Q. Liu, F. Yu, S. Wu, and L. Wang, “A convolutional click prediction model,” in CIKM.
  • [6] Y. Zhang, H. Dai, C. Xu, J. Feng, T. Wang, J. Bian, B. Wang, and T. Liu, “Sequential click prediction for sponsored search with recurrent neural networks,” in AAAI.
  • [7] J. Chen, B. Sun, H. Li, H. Lu, and X. Hua, “Deep CTR prediction in display advertising,” in ACM MM.
  • [8] W. Zhang, T. Du, and J. Wang, “Deep learning over multi-field categorical data - - A case study on user response prediction,” in ECIR.
  • [9] Y. Qu, H. Cai, K. Ren, W. Zhang, Y. Yu, Y. Wen, and J. Wang, “Product-based neural networks for user response prediction,” CoRR, vol. abs/1611.00144, 2016.
  • [10] B. M. Sarwar, G. Karypis, J. A. Konstan, and J. Riedl, “Item-based collaborative filtering recommendation algorithms,” in WWW, pp. 285–295.
  • [11] Y. Koren, R. M. Bell, and C. Volinsky, “Matrix factorization techniques for recommender systems,” IEEE Computer, vol. 42, no. 8, pp. 30–37, 2009.
  • [12] Y. Koren, “Factorization meets the neighborhood: a multifaceted collaborative filtering model,” in ACM SIGKDD.   ACM, pp. 426–434.
  • [13] L. Si and R. Jin, “Flexible mixture model for collaborative filtering,” in ICML, 2003, pp. 704–711.
  • [14] C. Wang and D. M. Blei, “Collaborative topic modeling for recommending scientific articles,” in ACM SIGKDD, 2011, pp. 448–456.
  • [15] T. Chen, W. Zhang, Q. Lu, K. Chen, Z. Zheng, and Y. Yu, “Svdfeature: a toolkit for feature-based collaborative filtering,” JMLR, vol. 13, pp. 3619–3622, 2012.
  • [16] X. Li, G. Cong, X. Li, T. N. Pham, and S. Krishnaswamy, “Rank-geofm: A ranking based geographical factorization method for point of interest recommendation,” in ACM SIGIR, 2015, pp. 433–442.
  • [17] Y. Chang, C. Hsieh, K. Chang, M. Ringgaard, and C. Lin, “Training and testing low-degree polynomial data mappings via linear svm,” JMLR, vol. 11, pp. 1471–1490, 2010.
  • [18] Y. Juan, Y. Zhuang, W. Chin, and C. Lin, “Field-aware factorization machines for CTR prediction,” in RecSys, 2016, pp. 43–50.
  • [19] X. He, J. Pan, O. Jin, T. Xu, B. Liu, T. Xu, Y. Shi, A. Atallah, R. Herbrich, S. Bowers, and J. Q. Candela, “Practical lessons from predicting clicks on ads at facebook,” in ADKDD, 2014, pp. 5:1–5:9.
  • [20] S. Rendle and L. Schmidt-Thieme, “Pairwise interaction tensor factorization for personalized tag recommendation,” in WSDM, 2010, pp. 81–90.
  • [21] T. Graepel, J. Q. Candela, T. Borchert, and R. Herbrich, “Web-scale bayesian click-through rate prediction for sponsored search advertising in microsoft’s bing search engine,” in ICML, 2010, pp. 13–20.
  • [22] K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” in CVPR, 2016, pp. 770–778.
  • [23] T. Mikolov, I. Sutskever, K. Chen, G. S. Corrado, and J. Dean, “Distributed representations of words and phrases and their compositionality,” in NIPS, 2013, pp. 3111–3119.
  • [24] A. Graves, A. Mohamed, and G. E. Hinton, “Speech recognition with deep recurrent neural networks,” in IEEE ICASSP, 2013, pp. 6645–6649.
  • [25] D. Silver, A. Huang, C. J. Maddison, A. Guez, L. Sifre, G. van den Driessche, J. Schrittwieser, I. Antonoglou, V. Panneershelvam, M. Lanctot, S. Dieleman, D. Grewe, J. Nham, N. Kalchbrenner, I. Sutskever, T. P. Lillicrap, M. Leach, K. Kavukcuoglu, T. Graepel, and D. Hassabis, “Mastering the game of go with deep neural networks and tree search,” Nature, vol. 529, no. 7587, pp. 484–489, 2016.
  • [26] P. Covington, J. Adams, and E. Sargin, “Deep neural networks for youtube recommendations,” in RecSys, 2016, pp. 191–198.
  • [27] R. Salakhutdinov, A. Mnih, and G. E. Hinton, “Restricted boltzmann machines for collaborative filtering,” in ICML, 2007, pp. 791–798.
  • [28] A. van den Oord, S. Dieleman, and B. Schrauwen, “Deep content-based music recommendation,” in NIPS, 2013, pp. 2643–2651.
  • [29] Y. Wu, C. DuBois, A. X. Zheng, and M. Ester, “Collaborative denoising auto-encoders for top-n recommender systems,” in ACM WSDM, 2016, pp. 153–162.
  • [30] C. Wu, A. Ahmed, A. Beutel, A. J. Smola, and H. Jing, “Recurrent recommender networks,” in WSDM, 2017, pp. 495–503.
  • [31] L. Zheng, V. Noroozi, and P. S. Yu, “Joint deep modeling of users and items using reviews for recommendation,” in WSDM, 2017, pp. 425–434.
  • [32] S. Sedhain, A. K. Menon, S. Sanner, and L. Xie, “Autorec: Autoencoders meet collaborative filtering,” in WWW, 2015, pp. 111–112.
  • [33] H. Wang, N. Wang, and D. Yeung, “Collaborative deep learning for recommender systems,” in ACM SIGKDD, 2015, pp. 1235–1244.
  • [34] P. Vincent, H. Larochelle, I. Lajoie, Y. Bengio, and P. Manzagol, “Stacked denoising autoencoders: Learning useful representations in a deep network with a local denoising criterion,” JMLR, vol. 11, pp. 3371–3408, 2010.
  • [35] X. Wang and Y. Wang, “Improving content-based and hybrid music recommendation using deep learning,” in ACM MM, 2014, pp. 627–636.
  • [36] S. Zhang, L. Yao, and A. Sun, “Deep learning based recommender system: A survey and new perspectives,” CoRR, vol. abs/1707.07435, 2017.
  • [37] S. Rendle, Z. Gantner, C. Freudenthaler, and L. Schmidt-Thieme, “Fast context-aware recommendations with factorization machines,” in ACM SIGIR, 2011, pp. 635–644.
  • [38] N. Srivastava, G. E. Hinton, A. Krizhevsky, I. Sutskever, and R. Salakhutdinov, “Dropout: a simple way to prevent neural networks from overfitting,” JMLR, vol. 15, no. 1, pp. 1929–1958, 2014.
  • [39] N. S. Keskar, D. Mudigere, J. Nocedal, M. Smelyanskiy, and P. T. P. Tang, “On large-batch training for deep learning: Generalization gap and sharp minima,” CoRR, vol. abs/1609.04836, 2016.
  • [40] T. Fawcett, “An introduction to roc analysis,” Pattern Recogn. Lett., vol. 27, no. 8, pp. 861–874, Jun. 2006.
  • [41] I. J. Goodfellow, Y. Bengio, and A. C. Courville, Deep Learning, ser. Adaptive computation and machine learning.   MIT Press, 2016.
  • [42] H. Larochelle, Y. Bengio, J. Louradour, and P. Lamblin, “Exploring strategies for training deep neural networks,” JMLR, vol. 10, pp. 1–40, 2009.
  • [43] T. Zhou, Z. Kuscsik, J.-G. Liu, M. Medo, J. R. Wakeling, and Y.-C. Zhang, “Solving the apparent diversity-accuracy dilemma of recommender systems,” Proceedings of the National Academy of Sciences, vol. 107, no. 10, pp. 4511–4515, 2010.
  • [44] D. Jannach and M. Ludewig, “When recurrent neural networks meet the neighborhood for session-based recommendation,” in Proceedings of the Eleventh ACM Conference on Recommender Systems, RecSys 2017, Como, Italy, August 27-31, 2017, 2017, pp. 306–310.

Huifeng Guo received the B.S. and M.S. degrees in Computer Science from Lanzhou University and Harbin Institute of Technology in China in 2012 and 2014, respectively. He is currently working toward the Ph.D. degree in the department of Computer Science, Shenzhen Graduate School, Harbin Institute of Technology. His research interests are in the areas of machine learning and recommendation, including graph mining and deep learning.

Ruiming Tang is currently a Senior Researcher in Huawei Noah’s Ark Lab. He received the B.S. degrees from the Department of Computer Science at Northeastern University China in 2009, and the Ph.D. degree from the Department of Computer Science at National University of Singapore, in 2014. He joined Huawei Noah’s Ark Lab since 2014. His research interests include machine learning and artificial intelligence, particularly, deep learning and recommender systems.

Yunming Ye received the Ph.D. degree in Computer Science from Shanghai Jiao Tong University. He is now a professor in the Shenzhen Graduate School, Harbin Institute of Technology. His research interests include data mining, text mining, and ensemble learning algorithms.

Zhenguo Li is currently the director of AI Theory Lab and Principal Researcher in Huawei Noah’s Ark Lab. He received the B.S. and M.S. degrees from the Department of Mathematics at Peking University, in 2002 and 2005, respectively, and the Ph.D. degree from the Department of Information Engineering at the Chinese University of Hong Kong, in 2008. He was an associate research scientist in the Department of Electrical Engineering at Columbia University. His research interests include machine learning and artificial intelligence.

Xiuqiang He is currently the director of Recommendation & Search Lab and Principal Researcher in Huawei Noah’s Ark Lab. He received the B.S. and M.S. degrees from the Department of Computer Science at Xi’an Jiaotong University, in 2003 and 2006, respectively, and the Ph.D. degree from the Department of Computer Science at the Hong Kong University of Science and Technology, in 2010. His research interests include machine learning algorithms in the area of recommendation and search.

Zhenhua Dong is a Senior Researcher of Huawei Noah’s ark lab, he received the BEng degree from Tianjin University in 2006 and the PhD degree in computer science and technology from Nankai University, China, in 2012. He was a research assistant at GroupLens lab in the University of Minnesota during 2010-2011. His research interests include recommender systems, mobile computing.

Comments 0
Request Comment
You are adding the first comment!
How to quickly get a good reply:
  • Give credit where it’s due by listing out the positive aspects of a paper before getting into which changes should be made.
  • Be specific in your critique, and provide supporting evidence with appropriate references to substantiate general statements.
  • Your comment should inspire ideas to flow and help the author improves the paper.

The better we are at sharing our knowledge with each other, the faster we move forward.
""
The feedback must be of minimum 40 characters and the title a minimum of 5 characters
   
Add comment
Cancel
Loading ...
230626
This is a comment super asjknd jkasnjk adsnkj
Upvote
Downvote
""
The feedback must be of minumum 40 characters
The feedback must be of minumum 40 characters
Submit
Cancel

You are asking your first question!
How to quickly get a good answer:
  • Keep your question short and to the point
  • Check for grammar or spelling errors.
  • Phrase it like a question
Test
Test description