小模型通过伪相关反馈也能达到大模型的检索效果
Pseudo Relevance Feedback is Enough to Close the Gap Between Small and Large Dense Retrieval Models
- 用大模型提取文档特征并注入查询表示,无需训练
- 在TREC DL和BEIR上性能接近大模型,且成本更低
- 适合资源有限但需高效检索的场景
将密集检索器扩展到更大语言模型(LLM)主干是提升检索效果的主流策略,但带来高昂成本:需要更高性能硬件(如大显存GPU),并导致索引和查询成本上升(延迟、能耗)。本文提出PromptPRF,一种基于特征的伪相关反馈框架,使小型LLM驱动的密集检索器能达到远大于自身的大型模型的检索效果。PromptPRF利用LLM从排名靠前的文档中提取独立于查询的结构化与非结构化特征(如实体、摘要、思维链关键词、文章段落),这些特征离线生成,并通过提示词注入密集查询表示,实现高效检索而无需额外训练。与依赖在线查询特定生成和稀疏检索的GRF不同,PromptPRF将反馈生成与查询处理解耦,支持完全零样本设置。在TREC DL和BEIR基准上的实验表明,PromptPRF持续提升检索效果,并具备更优的成本效益权衡。消融研究揭示了位置反馈的作用,并分析了特征提取器大小、PRF深度与模型性能之间的关系。结果表明,通过有效设计伪相关反馈,扩大检索器规模并非必要,可在降低推理成本的同时缩小小模型与大模型间的差距。
原文摘要 · Abstract (English)
Scaling dense retrievers to larger large language model (LLM) backbones has been a dominant strategy for improving their retrieval effectiveness. However, this has substantial cost implications: larger backbones require more expensive hardware (e.g. GPUs with more memory) and lead to higher indexing and querying costs (latency, energy consumption). In this paper, we challenge this paradigm by introducing PromptPRF, a feature-based pseudo-relevance feedback (PRF) framework that enables small LLM-based dense retrievers to achieve effectiveness comparable to much larger models. PromptPRF uses LLMs to extract query-independent, structured and unstructured features (e.g., entities, summaries, chain-of-thought keywords, essay) from top-ranked documents. These features are generated offline and integrated into dense query representations via prompting, enabling efficient retrieval without additional training. Unlike prior methods such as GRF, which rely on online, query-specific generation and sparse retrieval, PromptPRF decouples feedback generation from query processing and supports dense retrievers in a fully zero-shot setting. Experiments on TREC DL and BEIR benchmarks demonstrate that PromptPRF consistently improves retrieval effectiveness and offers favourable cost-effectiveness trade-offs. We further present ablation studies to understand the role of positional feedback and analyse the interplay between feature extractor size, PRF depth, and model performance. Our findings demonstrate that with effective PRF design, scaling the retriever is not always necessary, narrowing the gap between small and large models while reducing inference cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。