arXiv:2510.10009cs.CLcs.AI2025-10被引 5

用强化学习训练大模型自动扩展查询,提升多跳问答准确率。

Beyond the limitation of a single query: Train your LLM for query expansion with Reinforcement Learning

  • 通过强化学习让大模型在每轮搜索中生成多个查询变体并行检索。
  • 在7个问答基准上平均提升4.4%准确率,多跳推理任务增益显著。
  • 小模型(3B)配合压缩器也能达到顶尖性能,适合资源受限场景。

推理增强型搜索代理(如 Search-R1)通过迭代推理、搜索与生成来回答问题。然而,受限于推理与搜索能力,其在多跳问答基准上的表现仍不理想。为应对复杂或复合型查询,本文提出基于强化学习的 LLM 搜索代理,具备原生查询扩展能力。每轮搜索中,代理会生成多个查询变体并行检索,以覆盖更相关的信息。由于后训练数据和计算资源有限,搜索代理需同时掌握查询生成、文档理解与答案生成等多重任务,难度较大。为此,我们引入预训练压缩器模型,辅助代理理解检索到的文档,使其专注查询生成以提高召回率。实验表明,即使仅使用 3B 规模的小型 LLM,结合压缩器后,ExpandSearch 在七个问答基准上平均提升 4.4% 准确率,尤其在需要多样证据聚合的多跳推理任务中表现突出。

原文摘要 · Abstract (English)

Reasoning-augmented search agents, such as Search-R1, are trained to reason, search, and generate the final answer iteratively. Nevertheless, due to their limited capabilities in reasoning and search, their performance on multi-hop QA benchmarks remains far from satisfactory. To handle complex or compound queries, we train an LLM-based search agent with the native capability of query expansion through reinforcement learning. In each turn, our search agent proposes several query variants, which are searched simultaneously to cover more relevant information. Meanwhile, given limited post-training data and computing resources, it is very challenging for a search agent to master multiple tasks, including query generation, retrieved information understanding, and answer generation. Therefore, we propose incorporating a pre-trained squeezer model that helps the search agent understand the retrieved documents, allowing the search agent to focus on query generation for high retrieval recall. With the assistance of the squeezer model, we discover that even a small-scale 3B LLM can demonstrate a strong capability of query expansion and achieve state-of-the-art accuracy on the multi-hop QA benchmarks. To be specific, our experiments across seven question-answering benchmarks demonstrate that our method, named ExpandSearch, achieves an average improvement of 4.4% compared to state-of-the-art baselines, with strong gains on multi-hop reasoning tasks requiring diverse evidence aggregation.

大模型搜索增强强化学习多跳问答

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