中文

Fusion-in-T5:统一文档排序信号以改进信息检索

信息检索 2024-05-07 v2

摘要

搜索系统中常见的文档排序流水线为级联系统,涉及多个排序层以逐步整合不同信息。本文提出一种新颖的重排序器 Fusion-in-T5(FiT5),其通过基于模板的输入与全局注意力,将文本匹配信息、排序特征及全局文档信息整合进单一统一模型。在段落排序基准 MS MARCO 与 TREC DL 上的实验表明,作为单一模型的 FiT5 相较复杂级联流水线显著提升了排序性能。分析发现,通过注意力融合,FiT5 逐步关注相关文档与排序特征,联合利用多种形式的排序信息,并改进了对细微差异的辨别。我们的代码开源于 https://github.com/OpenMatch/FiT5。

关键词

引用

@article{arxiv.2305.14685,
  title  = {Fusion-in-T5: Unifying Document Ranking Signals for Improved Information Retrieval},
  author = {Shi Yu and Chenghao Fan and Chenyan Xiong and David Jin and Zhiyuan Liu and Zhenghao Liu},
  journal= {arXiv preprint arXiv:2305.14685},
  year   = {2024}
}

备注

COLING 2024