arXiv:2503.00223cs.IR2025-03被引 64

用强化学习让大模型自动生成高效检索词,不依赖标注数据。

DeepRetrieval: Hacking Real Search Engines and Retrievers with Large Language Models via Reinforcement Learning

  • 通过强化学习让大模型试错生成查询词,用检索效果当奖励。
  • 在真实搜索引擎上,文献搜索召回率提升至65.07%,超之前最好方法三倍。
  • 仅30亿参数就超越GPT-4o等商用模型,在11个数据集表现更优。

信息检索系统对高效访问大规模文档至关重要。现有方法虽利用大语言模型(LLM)通过查询增强提升性能,但常依赖昂贵的监督学习或知识蒸馏,需大量计算资源与人工标注数据。本文提出DeepRetrieval,一种无需参考查询的强化学习(RL)方法,训练LLM通过试错生成查询词,以检索指标作为奖励信号。该方法在真实搜索引擎上实现65.07%(文献搜索)和63.18%(临床试验搜索)的召回率,显著优于此前最优方法(分别为24.68%和32.11%)。它在证据检索、经典信息检索及SQL数据库搜索中也表现卓越。仅30亿参数即超越行业领先模型如GPT-4o和Claude-3.5-Sonnet,在11/13个数据集上胜出。结果表明,该强化学习范式更高效且有效。代码与数据见:https://github.com/pat-jj/DeepRetrieval。

原文摘要 · Abstract (English)

Information retrieval systems are crucial for enabling effective access to large document collections. Recent approaches have leveraged Large Language Models (LLMs) to enhance retrieval performance through query augmentation, but often rely on expensive supervised learning or distillation techniques that require significant computational resources and hand-labeled data. We introduce DeepRetrieval, a reinforcement learning (RL) approach that trains LLMs for query generation through trial and error without supervised data (reference query). Using retrieval metrics as rewards, our system generates queries that maximize retrieval performance. DeepRetrieval outperforms leading methods on literature search with 65.07% (vs. previous SOTA 24.68%) recall for publication search and 63.18% (vs. previous SOTA 32.11%) recall for trial search using real-world search engines. DeepRetrieval also dominates in evidence-seeking retrieval, classic information retrieval and SQL database search. With only 3B parameters, it outperforms industry-leading models like GPT-4o and Claude-3.5-Sonnet on 11/13 datasets. These results demonstrate that our RL approach offers a more efficient and effective paradigm for information retrieval. Our data and code are available at: https://github.com/pat-jj/DeepRetrieval.

检索增强强化学习大模型应用信息检索

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