论文标题
Inttower:预先级别系统的下一代两位型号
IntTower: the Next Generation of Two-Tower Model for Pre-Ranking System
论文作者
论文摘要
精确地在几毫秒内得分大量候选者对于工业预制系统至关重要。现有的预级系统主要采用\ textbf {两键}模型,因为``用户 - 项目解耦架构''范式能够平衡\ textIt {效率}和\ textit {效率}。但是,高效率的成本是忽略了用户和物品塔之间的潜在信息相互作用,从而阻碍了预测准确性。在本文中,我们表明可以设计一个两塔模型,该模型强调信息交互和推理效率。提出的模型,Inttower(\ textit {相互作用增强的两位塔})由Light-SE,Fe-Block和CIR模块组成。具体而言,轻巧的Light-SE模块用于确定不同特征的重要性,并在每个塔中获得精致的特征表示。 Fe-Block模块执行细粒度和早期特征交互,以明确捕获用户和项目塔之间的交互信号,并且CIR模块利用了对比的相互作用正规化,以进一步增强交互。三个公共数据集的实验结果表明,与排名模型相比,Inttower的表现高于SOTA预先级别模型,甚至达到可比的性能。此外,我们进一步验证了Inttower对大规模广告预制系统的有效性。 Inttower代码可公开可用\ footNote {https://github.com/archersama/inttower}
Scoring a large number of candidates precisely in several milliseconds is vital for industrial pre-ranking systems. Existing pre-ranking systems primarily adopt the \textbf{two-tower} model since the ``user-item decoupling architecture'' paradigm is able to balance the \textit{efficiency} and \textit{effectiveness}. However, the cost of high efficiency is the neglect of the potential information interaction between user and item towers, hindering the prediction accuracy critically. In this paper, we show it is possible to design a two-tower model that emphasizes both information interactions and inference efficiency. The proposed model, IntTower (short for \textit{Interaction enhanced Two-Tower}), consists of Light-SE, FE-Block and CIR modules. Specifically, lightweight Light-SE module is used to identify the importance of different features and obtain refined feature representations in each tower. FE-Block module performs fine-grained and early feature interactions to capture the interactive signals between user and item towers explicitly and CIR module leverages a contrastive interaction regularization to further enhance the interactions implicitly. Experimental results on three public datasets show that IntTower outperforms the SOTA pre-ranking models significantly and even achieves comparable performance in comparison with the ranking models. Moreover, we further verify the effectiveness of IntTower on a large-scale advertisement pre-ranking system. The code of IntTower is publicly available\footnote{https://github.com/archersama/IntTower}