提升无推理稀疏检索模型的搜索相关性,接近双塔模型效果且延迟仅是BM25的1.1倍
Towards Competitive Search Relevance For Inference-Free Learned Sparse Retrievers
- 引入词频惩罚机制,强化高信息量词的匹配权重
- 设计异构集成知识蒸馏框架,融合稠密与稀疏模型优势
- 在BEIR上超越现有最优无推理模型3.3分NDCG@10,延迟极低
学习型稀疏检索通过成熟的倒排索引引擎实现高效检索,近年来受到广泛关注。特别是无推理稀疏检索器因其在检索阶段无需在线模型推理,避免了巨大计算开销,具备合理吞吐量和延迟。然而,即使最先进的无推理稀疏模型在搜索相关性上仍远落后于稀疏和稠密双塔模型。为提升其搜索相关性,本文主张应为其设计专用训练方法,而非沿用双塔编码器的方案。提出两种性能提升策略:首先,设计基于IDF感知的匹配函数惩罚项,抑制低IDF词贡献,增强模型对信息量高的词的关注;其次,提出异构集成知识蒸馏框架,结合双塔稠密与稀疏检索器生成预训练阶段的监督信号。该框架分别利用两类模型的优势,提供强上界监督。为处理异构教师输出的反馈差异,采用归一化与聚合策略消除得分量纲差异。在BEIR基准测试中,所提模型较现有最优无推理稀疏模型提升3.3分NDCG@10,搜索相关性接近双塔稀疏模型,客户端延迟仅为BM25的1.1倍。
原文摘要 · Abstract (English)
Learned sparse retrieval, which can efficiently perform retrieval through mature inverted-index engines, has garnered growing attention in recent years. Particularly, the inference-free sparse retrievers are attractive as they eliminate online model inference in the retrieval phase thereby avoids huge computational cost, offering reasonable throughput and latency. However, even the state-of-the-art (SOTA) inference-free sparse models lag far behind in terms of search relevance when compared to both sparse and dense siamese models. Towards competitive search relevance for inference-free sparse retrievers, we argue that they deserve dedicated training methods other than using same ones with siamese encoders. In this paper, we propose two different approaches for performance improvement. First, we propose an IDF-aware penalty for the matching function that suppresses the contribution of low-IDF tokens and increases the model's focus on informative terms. Moreover, we propose a heterogeneous ensemble knowledge distillation framework that combines siamese dense and sparse retrievers to generate supervisory signals during the pre-training phase. The ensemble framework of dense and sparse retriever capitalizes on their strengths respectively, providing a strong upper bound for knowledge distillation. To concur the diverse feedback from heterogeneous supervisors, we normalize and then aggregate the outputs of the teacher models to eliminate score scale differences. On the BEIR benchmark, our model outperforms existing SOTA inference-free sparse model by \textbf{3.3 NDCG@10 score}. It exhibits search relevance comparable to siamese sparse retrievers and client-side latency only \textbf{1.1x that of BM25}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。