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.
Ссылки и действия
Дополнительные ресурсы: