In the realm of search systems, multi-stage cascade architecture is a prevalent method, typically consisting of sequential modules such as matching, pre-ranking, and ranking. It is generally acknowledged that the model used in the pre-ranking stage must strike a balance between efficacy and efficiency. Thus, the most commonly employed architecture is the representation-focused vector product based model. However, this architecture lacks effective interaction between the query and document, resulting in a reduction in the effectiveness of the search system. To address this issue, we present a novel pre-ranking framework called RankDFM. Our framework leverages DeepFM as the backbone and employs a pairwise training paradigm to learn the ranking of videos under a query. The capability of RankDFM to cross features provides significant improvement in offline and online A/B testing performance. Furthermore, we introduce a learnable feature selection scheme to optimize the model and reduce the time required for online inference, equivalent to a tree model. Currently, RankDFM has been deployed in the search system of a shortvideo App, providing daily services to hundreds of millions users.
翻译:摘要:在搜索系统领域,多阶段级联架构是一种常见方法,通常由匹配、预排序和排序等顺序模块组成。普遍认为,预排序阶段使用的模型必须在效能与效率之间取得平衡。因此,最常用的架构是基于表征驱动的向量积模型。然而,该架构缺乏查询与文档之间的有效交互,导致搜索系统的效果降低。为解决这一问题,我们提出了一种名为RankDFM的新型预排序框架。该框架以DeepFM为主干网络,采用成对训练范式来学习查询下的视频排序。RankDFM的交叉特征能力在离线和在线A/B测试中均带来显著性能提升。此外,我们引入了一种可学习特征选择方案以优化模型,并减少在线推理所需的时间,使其与树模型相当。目前,RankDFM已在某短视频应用的搜索系统中部署,为数亿用户提供日均服务。