HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Product-based Neural Networks for User Response Prediction

Yanru Qu; Han Cai; Kan Ren; Weinan Zhang; Yong Yu; Ying Wen; Jun Wang

Product-based Neural Networks for User Response Prediction

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.

Code Repositories

wangweitong/DL
pytorch
Mentioned in GitHub
xue-pai/FuxiCTR
pytorch
Mentioned in GitHub
wangweitong/recommend_system
pytorch
Mentioned in GitHub
JianzhouZhan/Awesome-RecSystem-Models
pytorch
Mentioned in GitHub
shenweichen/DeepCTR
tf
Mentioned in GitHub
UlionTse/mlgb
pytorch
Mentioned in GitHub
tangxyw/RecAlgorithm
tf
Mentioned in GitHub
shenweichen/DeepCTR-Torch
pytorch
Mentioned in GitHub
Atomu2014/product-nets
Official
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
click-through-rate-prediction-on-amazonPNN
AUC: 0.8679
click-through-rate-prediction-on-bing-newsPNN
AUC: 0.8321
Log Loss: 0.2775
click-through-rate-prediction-on-companyOPNN
AUC: 0.8658
Log Loss: 0.02641
click-through-rate-prediction-on-companyPNN*
AUC: 0.8672
Log Loss: 0.02636
click-through-rate-prediction-on-companyIPNN
AUC: 0.8664
Log Loss: 0.02637
click-through-rate-prediction-on-criteoIPNN
AUC: 0.7972
Log Loss: 0.45323
click-through-rate-prediction-on-criteoPNN*
AUC: 0.7987
Log Loss: 0.45214
click-through-rate-prediction-on-criteoOPNN
AUC: 0.7982
Log Loss: 0.45256
click-through-rate-prediction-on-dianpingPNN
AUC: 0.8445
Log Loss: 0.3424
click-through-rate-prediction-on-ipinyouPNN*
AUC: 0.7661
click-through-rate-prediction-on-ipinyouIPNN
AUC: 0.7914
click-through-rate-prediction-on-ipinyouOPNN
AUC: 0.8174
click-through-rate-prediction-on-movielensPNN
AUC: 0.7321

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp