arXiv:2510.17139cs.CLcs.IR2025-10被引 6

用生成伪文档替代重写查询,让大模型更高效地提升检索效果。

Rethinking On-policy Optimization for Query Augmentation

  • 让大模型生成伪文档来优化检索,而非直接改写查询。
  • 在多种任务上,该方法性能超越纯提示或强化学习方法。
  • 适合追求高效检索增强的开发者与研究者使用。

大型语言模型(LLMs)的发展推动了信息检索中查询增强技术的兴起。现有方法分为两类:一类通过提示生成答案或伪文档作为新查询,依赖模型参数知识;另一类则利用强化学习(RL)微调模型以优化查询重写,直接提升检索指标。两者尚未在一致条件下对比。本文首次系统比较了基于提示和基于强化学习的查询增强方法,涵盖证据查找、即席检索和工具检索等多类任务。关键发现是,在计算资源受限的评估下,无需训练的简单提示方法表现与昂贵的强化学习方法相当,甚至更优,尤其在使用强大模型时。受此启发,我们提出一种新型混合方法——基于策略的伪文档查询扩展(OPQE),使模型学习生成能最大化检索性能的伪文档,融合提示的灵活性与强化学习的目标导向性。实验表明,OPQE优于独立的提示法和强化学习法,证明协同策略效果最佳。代码已开源,便于复现。

原文摘要 · Abstract (English)

Recent advances in large language models (LLMs) have led to a surge of interest in query augmentation for information retrieval (IR). Two main approaches have emerged. The first prompts LLMs to generate answers or pseudo-documents that serve as new queries, relying purely on the model's parametric knowledge or contextual information. The second applies reinforcement learning (RL) to fine-tune LLMs for query rewriting, directly optimizing retrieval metrics. While having respective advantages and limitations, the two approaches have not been compared under consistent experimental conditions. In this work, we present the first systematic comparison of prompting-based and RL-based query augmentation across diverse benchmarks, including evidence-seeking, ad hoc, and tool retrieval. Our key finding is that under a compute-aware comparison setting, simple, training-free query augmentation often performs on par with, or even surpasses, more expensive RL-based counterparts, especially when using powerful LLMs. Motivated by this discovery, we introduce a novel hybrid method, On-policy Pseudo-document Query Expansion (OPQE), in which the LLM policy learns to generate a pseudo-document that maximizes retrieval performance, rather than rewriting the query, thus merging the flexibility and generative structure of prompting with the targeted optimization of RL. We show OPQE outperforms both standalone prompting and RL-based rewriting, demonstrating that a synergistic approach yields the best results. We open source our implementation to facilitate reproducibility.

查询增强大模型强化学习检索优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。