arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.02249cs.IRcs.AIcs.LGcs.SI
收藏

RAT:用于点击率预测的检索增强Transformer

RAT: Retrieval-Augmented Transformer for Click-Through Rate Prediction

Tsinghua Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院);College of Computer Science and Software Engineering, Shenzhen University(深圳大学计算机与软件学院);Research Center of Artificial Intelligence, Peng Cheng Laboratory(鹏城实验室人工智能研究中心)

Yushen Li, Jinpeng Wang, Tao Dai, Jieming Zhu, Jun Yuan, Rui Zhang, Shu-Tao Xia

AI总结:

本文提出检索增强Transformer(RAT),通过检索相似样本构建增强输入,并利用级联注意力同时建模样本内与跨样本特征交互,在保持效率的同时提升CTR预测性能,尤其在长尾场景中表现优越。

摘要:

点击率(CTR)预测是Web应用中的一项基础任务,其中一个关键问题是设计有效的特征交互模型。当前方法主要集中于在单个样本内建模特征交互,而忽略了可作为参考上下文以增强预测的潜在跨样本关系。为弥补这一不足,本文提出了一种检索增强Transformer(RAT),旨在获取样本内和样本间的细粒度特征交互。通过检索相似样本,我们为每个目标样本构建增强输入。然后,我们构建具有级联注意力的Transformer层,以同时捕获样本内和跨样本的特征交互,从而在保持效率的同时促进全面推理以改进CTR预测。在真实世界数据集上的大量实验证实了RAT的有效性,并表明其在长尾场景中具有优势。代码已在\url{https://github.com/YushenLi807/WWW24-RAT}开源。

英文摘要:

Predicting click-through rates (CTR) is a fundamental task for Web applications, where a key issue is to devise effective models for feature interactions. Current methodologies predominantly concentrate on modeling feature interactions within an individual sample, while overlooking the potential cross-sample relationships that can serve as a reference context to enhance the prediction. To make up for such deficiency, this paper develops a Retrieval-Augmented Transformer (RAT), aiming to acquire fine-grained feature interactions within and across samples. By retrieving similar samples, we construct augmented input for each target sample. We then build Transformer layers with cascaded attention to capture both intra- and cross-sample feature interactions, facilitating comprehensive reasoning for improved CTR prediction while retaining efficiency. Extensive experiments on real-world datasets substantiate the effectiveness of RAT and suggest its advantage in long-tail scenarios. The code has been open-sourced at \url{https://github.com/YushenLi807/WWW24-RAT}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑