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

COLD:迈向下一代预排名系统

COLD: Towards the Next Generation of Pre-Ranking System

Alibaba Group(阿里巴巴集团)

Zhe Wang, Liqin Zhao, Biye Jiang, Guorui Zhou, Xiaoqiang Zhu, Kun Gai

摘要:

多级级联结构广泛存在于推荐系统、在线广告等工业系统中,通常由匹配、预排序、排序等顺序模块组成。长期以来,人们认为预排序只是排序模块的简化版本,考虑到待排序的候选集规模较大。因此,努力主要是简化排名模型,以处理在线推理计算能力的爆炸。在本文中,我们重新思考的挑战,预排名系统从算法系统协同设计的观点。本文通过对预排序模型和计算能力的联合优化,设计了一种新的预排序系统,而不是通过限制模型结构来节省计算能力,从而导致模型性能的损失。我们称之为COLD(计算能力成本感知在线和轻量级深度预排名系统)。COLD在三个方面击败了SOTA:(i)在可控计算能力成本的约束下,可以在COLD中应用具有交叉特征的任意深度模型。(ii)通过应用用于推理加速的优化技巧来显式地降低计算能力成本。这进一步为COLD提供了应用更复杂的深度模型以实现更好性能的空间。(iii)COLD模型以在线学习和服务的方式工作,使其具有出色的能力来应对数据分布偏移的挑战。自2019年以来,COLD已部署在阿里巴巴展示广告系统中几乎所有涉及预排名模块的产品中,带来了显著的改进。

英文摘要:

Multi-stage cascade architecture exists widely in many industrial systems such as recommender systems and online advertising, which often consists of sequential modules including matching, pre-ranking, ranking, etc. For a long time, it is believed pre-ranking is just a simplified version of the ranking module, considering the larger size of the candidate set to be ranked. Thus, efforts are made mostly on simplifying ranking model to handle the explosion of computing power for online inference. In this paper, we rethink the challenge of the pre-ranking system from an algorithm-system co-design view. Instead of saving computing power with restriction of model architecture which causes loss of model performance, here we design a new pre-ranking system by joint optimization of both the pre-ranking model and the computing power it costs. We name it COLD (Computing power cost-aware Online and Lightweight Deep pre-ranking system). COLD beats SOTA in three folds: (i) an arbitrary deep model with cross features can be applied in COLD under a constraint of controllable computing power cost. (ii) computing power cost is explicitly reduced by applying optimization tricks for inference acceleration. This further brings space for COLD to apply more complex deep models to reach better performance. (iii) COLD model works in an online learning and severing manner, bringing it excellent ability to handle the challenge of the data distribution shift. Meanwhile, the fully online pre-ranking system of COLD provides us with a flexible infrastructure that supports efficient new model developing and online A/B testing.Since 2019, COLD has been deployed in almost all products involving the pre-ranking module in the display advertising system in Alibaba, bringing significant improvements.

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

↑