CTR-Sink: Attention Sink for Language Models in Click-Through Rate Prediction

2508.03668v1 cs.CL 2025-08-09
Авторы:

Zixuan Li, Binzong Geng, Jing Xiong, Yong He, Yuxuan Hu, Jian Chen, Dingwei Chen, Xiyu Chang, Liang Zhang, Linjian Mo, Chengming Li, Chuan Yuan, Zhenan Sun

Резюме на русском

**Резюме** Click-Through Rate (CTR) prediction — основная задача в системах рекомендаций, которая предсказывает вероятность пользовательского клика на основе исторических данных поведения. Одной из основных проблем является структурная несоответствие поведенческих последовательностей текстовым моделям: поведенческие действия, соединенные несемантическими разделителями, не соответствуют естественному языку, вызывая семантическую расколотость. Это приводит к рассеиванию внимания Language Models (LMs) на несмысленных токенах, а не на значимых разделителях между действиями, что снижает точность предсказаний. Мы предлагаем $\textit{CTR-Sink}$, новую модель, вводящую behavior-level attention sinks, которые динамически регулируются с помощью внешних сигналов. Мы внедряем sink tokens между действиями, используя рекомендательные сигналы, такие как временное расстояние, и динамически стимулируем внимание LMs к этим токенам. Мы также сильно усиливаем взаимодействие между sink tokens, чтобы лучше воспринимать связи между действиями. Наши эксперименты на данных из индустрии и двух открытых датасетов (MovieLens, Kuairec) показали, что CTR-Sink эффективно решает проблему semantic fragmentation, значительно повышая точность предсказаний CTR.

Abstract

Click-Through Rate (CTR) prediction, a core task in recommendation systems, estimates user click likelihood using historical behavioral data. Modeling user behavior sequences as text to leverage Language Models (LMs) for this task has gained traction, owing to LMs' strong semantic understanding and contextual modeling capabilities. However, a critical structural gap exists: user behavior sequences consist of discrete actions connected by semantically empty separators, differing fundamentally from the coherent natural language in LM pre-training. This mismatch causes semantic fragmentation, where LM attention scatters across irrelevant tokens instead of focusing on meaningful behavior boundaries and inter-behavior relationships, degrading prediction performance. To address this, we propose $\textit{CTR-Sink}$, a novel framework introducing behavior-level attention sinks tailored for recommendation scenarios. Inspired by attention sink theory, it constructs attention focus sinks and dynamically regulates attention aggregation via external information. Specifically, we insert sink tokens between consecutive behaviors, incorporating recommendation-specific signals such as temporal distance to serve as stable attention sinks. To enhance generality, we design a two-stage training strategy that explicitly guides LM attention toward sink tokens and a attention sink mechanism that amplifies inter-sink dependencies to better capture behavioral correlations. Experiments on one industrial dataset and two open-source datasets (MovieLens, Kuairec), alongside visualization results, validate the method's effectiveness across scenarios.

Ссылки и действия