SOTAVerified

Product-based Neural Networks for User Response Prediction

2016-11-01Code Available0· sign in to hype

Yanru Qu, Han Cai, Kan Ren, Wei-Nan Zhang, Yong Yu, Ying Wen, Jun Wang

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

Predicting user responses, such as clicks and conversions, is of great importance and has found its usage in many Web applications including recommender systems, web search and online advertising. The data in those applications is mostly categorical and contains multiple fields; a typical representation is to transform it into a high-dimensional sparse binary feature representation via one-hot encoding. Facing with the extreme sparsity, traditional models may limit their capacity of mining shallow patterns from the data, i.e. low-order feature combinations. Deep models like deep neural networks, on the other hand, cannot be directly applied for the high-dimensional input because of the huge feature space. In this paper, we propose a Product-based Neural Networks (PNN) with an embedding layer to learn a distributed representation of the categorical data, a product layer to capture interactive patterns between inter-field categories, and further fully connected layers to explore high-order feature interactions. Our experimental results on two large-scale real-world ad click datasets demonstrate that PNNs consistently outperform the state-of-the-art models on various metrics.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
AmazonPNNAUC0.87Unverified
Bing NewsPNNAUC0.83Unverified
Company*PNN*AUC0.87Unverified
Company*OPNNAUC0.87Unverified
Company*IPNNAUC0.87Unverified
CriteoPNN*AUC0.8Unverified
CriteoIPNNAUC0.8Unverified
CriteoOPNNAUC0.8Unverified
DianpingPNNAUC0.84Unverified
iPinYouIPNNAUC0.79Unverified
iPinYouPNN*AUC0.77Unverified
iPinYouOPNNAUC0.82Unverified
MovieLens 20MPNNAUC0.73Unverified

Reproductions