arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.09340cs.IR
收藏

基于多预训练策略的Web搜索包围排序模型

Ensemble Ranking Model with Multiple Pretraining Strategies for Web Search

Xiaojie Sun, Lulu Yu, Yiting Wang, Keping Bi, Jiafeng Guo

摘要:

一个有效的排序模型通常需要大量的训练数据来学习文档和查询之间的相关性。用户点击经常被用作训练数据,因为它们可以指示相关性并且收集起来很便宜,但是它们包含大量的偏差和噪声。在减轻模拟用户点击中的各种类型的偏差以基于多个特征训练有效的学习排名模型方面已经有了一些工作。然而,如何在具有真实点击数据的大规模预训练模型上有效地使用这些方法是未知的。为了减轻现实世界中的数据偏差,我们在预训练阶段引入了基于统计学的特征,改进了排名目标,添加了随机否定,并校准了倾向计算。然后,我们微调几个预训练的模型,并训练一个集成模型,以在微调阶段将来自各种预训练模型的所有预测与人类注释数据聚合在一起。我们的方法在WSDM Cup 2023的“网络搜索预训练”任务中获得第三名,比第四名的团队好22.6%。

英文摘要:

An effective ranking model usually requires a large amount of training data to learn the relevance between documents and queries. User clicks are often used as training data since they can indicate relevance and are cheap to collect, but they contain substantial bias and noise. There has been some work on mitigating various types of bias in simulated user clicks to train effective learning-to-rank models based on multiple features. However, how to effectively use such methods on large-scale pre-trained models with real-world click data is unknown. To alleviate the data bias in the real world, we incorporate heuristic-based features, refine the ranking objective, add random negatives, and calibrate the propensity calculation in the pre-training stage. Then we fine-tune several pre-trained models and train an ensemble model to aggregate all the predictions from various pre-trained models with human-annotation data in the fine-tuning stage. Our approaches won 3rd place in the "Pre-training for Web Search" task in WSDM Cup 2023 and are 22.6% better than the 4th-ranked team.

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

↑