HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Deep Learning over Multi-field Categorical Data: A Case Study on User Response Prediction

Weinan Zhang; Tianming Du; Jun Wang

Deep Learning over Multi-field Categorical Data: A Case Study on User Response Prediction

Abstract

Predicting user responses, such as click-through rate and conversion rate, are critical in many web applications including web search, personalised recommendation, and online advertising. Different from continuous raw features that we usually found in the image and audio domains, the input features in web space are always of multi-field and are mostly discrete and categorical while their dependencies are little known. Major user response prediction models have to either limit themselves to linear models or require manually building up high-order combination features. The former loses the ability of exploring feature interactions, while the latter results in a heavy computation in the large feature space. To tackle the issue, we propose two novel models using deep neural networks (DNNs) to automatically learn effective patterns from categorical feature interactions and make predictions of users' ad clicks. To get our DNNs efficiently work, we propose to leverage three feature transformation methods, i.e., factorisation machines (FMs), restricted Boltzmann machines (RBMs) and denoising auto-encoders (DAEs). This paper presents the structure of our models and their efficient training algorithms. The large-scale experiments with real-world data demonstrate that our methods work better than major state-of-the-art models.

Code Repositories

ddatta-DAC/Learning
tf
Mentioned in GitHub
shenweichen/DeepCTR
tf
Mentioned in GitHub
UlionTse/mlgb
pytorch
Mentioned in GitHub
shenweichen/DeepCTR-Torch
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
click-through-rate-prediction-on-companyFNN
AUC: 0.8683
Log Loss: 0.02629
click-through-rate-prediction-on-criteoFNN
AUC: 0.7963
Log Loss: 0.45738
click-through-rate-prediction-on-ipinyouFNN
AUC: 0.7619

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