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

面向成本受限数据源检索的渐进式查询扩展

Progressive Query Expansion for Retrieval Over Cost-constrained Data Sources

Muhammad Shihab Rashid, Jannat Ara Meem, Yue Dong, Vagelis Hristidis

AI总结:

本文提出渐进式查询扩展算法ProQE,结合伪相关反馈与大语言模型,在检索过程中迭代扩展查询,以降低按文档收费API场景下的检索成本,并在四个数据集上取得比最先进基线高37%的效果且成本效益最优。

摘要:

查询扩展长期以来被用于提高查询检索器的准确性。早期工作依赖伪相关反馈(PRF)技术,该技术利用第一阶段检索到的文档中提取的词项来扩充查询。然而,这些文档可能存在噪声,从而影响排序的有效性。为避免这一问题,近期研究转而使用大语言模型(LLMs)生成额外内容来扩展查询。这些技术容易产生幻觉,并且主要关注LLM的使用成本。然而,在若干重要的实际场景中,成本可能主要由检索环节主导,因为语料库只能通过按检索文档数量收费的API访问。我们提出将经典PRF技术与LLMs相结合,并创建一种渐进式查询扩展算法ProQE,该算法在检索更多文档的过程中迭代地扩展查询。ProQE同时兼容稀疏检索系统和稠密检索系统。我们在四个检索数据集上的实验结果表明,ProQE比最先进的基线方法性能高出37%,并且最具成本效益。

英文摘要:

Query expansion has been employed for a long time to improve the accuracy of query retrievers. Earlier works relied on pseudo-relevance feedback (PRF) techniques, which augment a query with terms extracted from documents retrieved in a first stage. However, the documents may be noisy hindering the effectiveness of the ranking. To avoid this, recent studies have instead used Large Language Models (LLMs) to generate additional content to expand a query. These techniques are prone to hallucination and also focus on the LLM usage cost. However, the cost may be dominated by the retrieval in several important practical scenarios, where the corpus is only available via APIs which charge a fee per retrieved document. We propose combining classic PRF techniques with LLMs and create a progressive query expansion algorithm ProQE that iteratively expands the query as it retrieves more documents. ProQE is compatible with both sparse and dense retrieval systems. Our experimental results on four retrieval datasets show that ProQE outperforms state-of-the-art baselines by 37% and is the most cost-effective.

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

↑