让大模型更精准地找出任务隐含相关的文档。
Retro*: Optimizing LLMs for Reasoning-Intensive Document Retrieval
- 用评分标准引导模型细粒度推理文档与任务的关系。
- 结合多条推理路径提升相关性判断可靠性,测试时可扩展。
- 专为文档检索设计强化学习算法,显著提升性能。
随着大模型代理和RAG的普及,准确识别任务所需但关联隐晦的文档变得愈发重要。现有信息检索技术在细粒度推理方面存在不足。本文提出Retro*,一种面向高复杂度推理的文档检索方法。该方法引入基于评分标准的相关性打分机制,使模型能依据明确定义的标准评估任务与候选文档间的关联,生成细粒度且可解释的得分。Retro*支持测试时扩展,通过整合多条推理轨迹实现更可靠的评分。为优化其推理能力,我们设计了一种专用于该打分机制的强化学习算法,采用双重复合奖励机制充分挖掘每个训练样本的推理路径。实验表明,Retro*在BRIGHT基准上表现优异,达到当前最优水平。
原文摘要 · Abstract (English)
With the growing popularity of LLM agents and RAG, it has become increasingly important to retrieve documents that are essential for solving a task, even when their connection to the task is indirect or implicit. Addressing this problem requires fine-grained reasoning to accurately assess the relevance between the task and each candidate document. This capability, however, poses a significant challenge for existing IR techniques. Despite recent progress in reasoning-enhanced IR, existing approaches still face significant challenges in applicability, scalability, and efficiency. In this work, we propose Retro*, a novel approach for reasoning-intensive document retrieval. Our method introduces a rubric-based relevance scoring mechanism, enabling the model to reason about the relationship between a task and a document based on explicitly defined criteria, whereby producing a fine-grained, interpretable relevance score. Retro* also supports test-time scaling by combining multiple reasoning trajectories via score integration, which produces more reliable relevance estimates. To optimize Retro*'s reasoning capabilities, we introduce a novel reinforcement learning algorithm tailored for its relevance scoring mechanism, which employs two composite rewards to fully exploit the trajectories of each training sample. Our experiments show that Retro* outperforms existing document retrieval methods with notable advantages, leading to state-of-the-art performance on the BRIGHT benchmark.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。