解决推荐系统中特征交互的信息丢失与坍塌问题
Addressing Information Loss and Interaction Collapse: A Dual Enhanced Attention Framework for Feature Interaction
- 引入双注意力机制,保留原始交互信息
- 在工业数据集上提升点击率预测效果
- 适合做推荐系统特征交互建模的研究者
Transformer 在点击率预测的特征交互中表现优异,但存在信息丢失和交互坍塌问题。由于依赖内积表示成对关系,原始交互信息被压缩,导致保真度下降;同时,长尾特征分布使得低信息量特征嵌入限制了其他特征的信息表达,引发嵌入矩阵坍塌。为此,提出双增强注意力框架(Dual Enhanced Attention),包含组合ID注意力机制和防坍塌注意力机制:前者直接保留特征交互对以减少信息损失,后者自适应过滤低信息量交互对以避免交互坍塌。在工业数据集上的大量实验验证了该方法的有效性。
原文摘要 · Abstract (English)
The Transformer has proven to be a significant approach in feature interaction for CTR prediction, achieving considerable success in previous works. However, it also presents potential challenges in handling feature interactions. Firstly, Transformers may encounter information loss when capturing feature interactions. By relying on inner products to represent pairwise relationships, they compress raw interaction information, which can result in a degradation of fidelity. Secondly, due to the long-tail features distribution, feature fields with low information-abundance embeddings constrain the information abundance of other fields, leading to collapsed embedding matrices. To tackle these issues, we propose a Dual Attention Framework for Enhanced Feature Interaction, known as Dual Enhanced Attention. This framework integrates two attention mechanisms: the Combo-ID attention mechanism and the collapse-avoiding attention mechanism. The Combo-ID attention mechanism directly retains feature interaction pairs to mitigate information loss, while the collapse-avoiding attention mechanism adaptively filters out low information-abundance interaction pairs to prevent interaction collapse. Extensive experiments conducted on industrial datasets have shown the effectiveness of Dual Enhanced Attention.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。