Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Hasan Saribas, Cagri Yesil, Serdarcan Dilbaz, Halit Orenbas• 2023

Related benchmarks

TaskDatasetResultRank
CTR PredictionCriteo
AUC0.815
282
CTR PredictionAvazu
AUC76.66
144
CTR PredictionFrappe
AUC0.9861
83
Showing 3 of 3 rows

Other info

Code

Follow for update