国内刊号:37-1389/N
国际刊号:1671-9352
发布日期:
作者:赵峰叙,王健,林原,林鸿飞
单位:1. 大连理工大学计算机科学与技术学院,辽宁 大连 116024;2. 大连理工大学公共管理学院,辽宁 大连 116024
关键词:息检索,排序学习,概率分布,
基金:国家自然科学基金资助项目(61976036)
现有的排序学习模型依赖于模型输出的评分来表示文档间的偏序关系。考虑到这种将评分看作单一数值的限制,提出一种概率分布排序学习模型优化方法,引入排序分数的不确定性,以概率分布的形式对排序分数进行平滑,进而将排序分数大小的比较变成对分数偏序关系的概率估计。在此基础上,将该方法应用于排序学习模型RankNet、LambdaRank以及LambdaMART,更合理地拟合模型概率与目标概率之间的差距,从而对排序学习模型进行优化, 并在多个大规模真实数据集上进行实验。结果表明, 经过优化后的模型性能相比于优化前具有显著提高,验证本文方法的有效性。
来源:2024年第7期
《山东大学学报(理学版)》期刊编辑部