HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

MMBAttn: Max-Mean and Bit-wise Attention for CTR Prediction

Hasan Saribas Cagri Yesil Serdarcan Dilbaz Halit Orenbas

MMBAttn: Max-Mean and Bit-wise Attention for CTR Prediction

Abstract

With the increasing complexity and scale of click-through rate (CTR) prediction tasks in online advertising and recommendation systems, accurately estimating the importance of features has become a critical aspect of developing effective models. In this paper, we propose an attention-based approach that leverages max and mean pooling operations, along with a bit-wise attention mechanism, to enhance feature importance estimation in CTR prediction. Traditionally, pooling operations such as max and mean pooling have been widely used to extract relevant information from features. However, these operations can lead to information loss and hinder the accurate determination of feature importance. To address this challenge, we propose a novel attention architecture that utilizes a bit-based attention structure that emphasizes the relationships between all bits in features, together with maximum and mean pooling. By considering the fine-grained interactions at the bit level, our method aims to capture intricate patterns and dependencies that might be overlooked by traditional pooling operations. To examine the effectiveness of the proposed method, experiments have been conducted on three public datasets. The experiments demonstrated that the proposed method significantly improves the performance of the base models to achieve state-of-the-art results.

Benchmarks

BenchmarkMethodologyMetrics
click-through-rate-prediction-on-avazuFinalMLP + MMBAttn
AUC: 0.7666
click-through-rate-prediction-on-avazuDNN + MMBAttn
AUC: 0.765
click-through-rate-prediction-on-criteoFinalMLP + MMBAttn
AUC: 0.81497
click-through-rate-prediction-on-criteoDNN + MMBAttn
AUC: 0.8143
click-through-rate-prediction-on-frappeFinalMLP + MMBAttn
AUC: 0.9861
click-through-rate-prediction-on-frappeDNN + MMBAttn
AUC: 0.985

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