计算机研究与发展2023,Vol.60Issue(1) :85-94.DOI:10.7544/issn1000-1239.202110853

基于多任务学习的位置倾向性得分预测算法

Prediction of the Positional Propensity Scores Based on Multi Task Learning

曹泽麟 徐君 董振华 文继荣
计算机研究与发展2023,Vol.60Issue(1) :85-94.DOI:10.7544/issn1000-1239.202110853

基于多任务学习的位置倾向性得分预测算法

Prediction of the Positional Propensity Scores Based on Multi Task Learning

曹泽麟 1徐君 2董振华 3文继荣2
扫码查看

作者信息

  • 1. 中国人民大学高瓴人工智能学院 北京 100872
  • 2. 中国人民大学高瓴人工智能学院 北京 100872;大数据管理与分析方法研究北京市重点实验室(中国人民大学)北京 100071
  • 3. 华为技术有限公司诺亚方舟实验室 广东深圳 518129
  • 折叠

摘要

用户搜索时产生的点击数据分布,在不同的搜索场景下存在较大差异.现有算法如融合上下文的位置模型(contextual position based model,CPBM)往往只通过单个模型预测多种场景下的位置倾向性得分,不可避免地降低了模型在不同场景下的预测准确性,影响去除位置偏置的效果.基于上述问题提出一种基于多任务学习的多门专家混合位置倾向性得分预测模型(multi-gate contextual position based model,MCPBM),在CPBM模型的基础上加入信息筛选结构,解决了多场景数据联合训练时预测准确性不佳的问题.同时,为了缓解不同任务收敛速度不一致的问题,提出了指数加权平均权重动态调整算法,在加速模型训练的同时提升了模型整体预测性能.实验结果表明提出的MCPBM模型在多场景数据联合训练时,预测准确性优于传统的CPBM;在使用MCPBM模型去除位置偏置后,基于生成的无偏数据训练得到的排序模型,在AvgRank排序指标上有1%-5%的提升.

关键词

位置偏置/无偏排序/逆概率加权/多任务学习/倾向性得分预测

引用本文复制引用

基金项目

国家重点研发计划(2019YFE0198200)

国家自然科学基金(61872338)

国家自然科学基金(61832017)

北京高校卓越青年科学家计划(BJJWZYJH012019100020098)

出版年

2023
计算机研究与发展
中国科学院计算技术研究所 中国计算机学会

计算机研究与发展

CSTPCDCSCD北大核心
影响因子:2.649
ISSN:1000-1239
被引量1
参考文献量1
段落导航相关论文