Unleashing the Potential of Sparse Attention on Long-term Behaviors for CTR Prediction

TL;DR

SparseCTR is a novel model that efficiently captures long-term user behaviors using personalized sequence segmentation and a three-branch sparse self-attention mechanism, significantly improving CTR prediction in recommender systems.

Contribution

The paper introduces SparseCTR, a new approach that effectively models long-term user behaviors with personalized segmentation and a three-branch attention mechanism, addressing computational challenges.

Findings

Outperforms state-of-the-art methods in efficiency and accuracy

Maintains performance improvements across three orders of magnitude in FLOPs

Increases CTR by 1.72% and CPM by 1.41% in online A/B testing

Abstract

In recent years, the success of large language models (LLMs) has driven the exploration of scaling laws in recommender systems. However, models that demonstrate scaling laws are actually challenging to deploy in industrial settings for modeling long sequences of user behaviors, due to the high computational complexity of the standard self-attention mechanism. Despite various sparse self-attention mechanisms proposed in other fields, they are not fully suited for recommendation scenarios. This is because user behaviors exhibit personalization and temporal characteristics: different users have distinct behavior patterns, and these patterns change over time, with data from these users differing significantly from data in other fields in terms of distribution. To address these challenges, we propose SparseCTR, an efficient and effective model specifically designed for long-term behaviors of users. To be precise, we first segment behavior sequences into chunks in a personalized manner to avoid separating continuous behaviors and enable parallel processing of sequences. Based on these chunks, we propose a three-branch sparse self-attention mechanism to jointly identify users' global interests, interest transitions, and short-term interests. Furthermore, we design a composite relative temporal encoding via learnable, head-specific bias coefficients, better capturing sequential and periodic relationships among user behaviors. Extensive experimental results show that SparseCTR not only improves efficiency but also outperforms state-of-the-art methods. More importantly, it exhibits an obvious scaling law phenomenon, maintaining performance improvements across three orders of magnitude in FLOPs. In online A/B testing, SparseCTR increased CTR by 1.72\% and CPM by 1.41\%. Our source code is available at https://github.com/laiweijiang/SparseCTR.