arXiv:2507.07634cs.CL2025-07被引 1

用强化学习精简检索步骤,让小模型更高效答多跳问题

FrugalRAG: Less is More in RL Finetuning for Multi-Hop Question Answering

  • 分两阶段训练:先全量搜索,再根据题目难度动态减少检索步数
  • 仅用约1000样本即达顶尖效果,检索成本减半,准确率不降
  • 适合追求高效推理的小模型应用,尤其在资源受限场景

基于最终答案奖励的强化学习(RL)推动了小语言模型(SLMs)在数学与代码等推理任务上的进展。然而,将相同方法应用于多跳问答(multi-hop QA)等检索增强生成(RAG)任务时,增益有限,常落后于监督微调或提示法基线。我们提出FrugalRAG,一种两阶段微调框架,通过测试时缩放策略,自适应减少检索步数以平衡准确率与效率。首先,使用监督微调训练模型生成广泛子查询的全探索策略;随后,采用强化学习根据问题难度动态剪枝搜索深度,直接奖励既正确又节省资源的策略。相比以往方法需10倍数据,本方法仅用约1000个样本即实现竞争力表现。在HotPotQA及其他多跳问答基准上,FrugalRAG达到领先的效率-准确率权衡,检索成本几乎减半。此外,在挑战性BrowseCompPlus基准上,其零样本泛化能力超越基于SLM及其他基线。结果表明,强化学习可用于减少而非增加推理步数,是实现可扩展、高效RAG的有效方案。

原文摘要 · Abstract (English)

Reinforcement learning (RL) based on the final answer's reward has driven recent progress in small language models (SLMs) on reasoning-heavy tasks such as math and code. However, applying the same techniques to retrieval-augmented generation (RAG) benchmarks like multi-hop QA has yielded limited gains, often trailing supervised or prompting-only baselines. Instead, we argue that a viable path for RL in multi-hop QA is to use test-time scaling judiciously to optimize both final answer accuracy and efficiency in reaching that answer. We propose FrugalRAG, a two-stage finetuning framework that adaptively reduces the number of retrieval steps based on a question's difficulty. First, we train an SLM with supervised finetuning on a full-exploration policy that generates broad sub-queries. Then, we apply RL to adaptively prune search depth based on question difficulty, directly rewarding policies that balance correctness with frugality. Unlike prior approaches requiring 10x more data, our method achieves competitive performance with only approximately 1,000 examples. On HotPotQA and other multi-hop QA benchmarks, FrugalRAG attains state-of-the-art efficiency-accuracy tradeoffs, cutting retrieval cost nearly in half. Moreover, on the challenging BrowseCompPlus benchmark, it generalizes zero-shot and surpasses SLM-based and other baselines. These results demonstrate the use of RL not to increase reasoning steps, but to reduce them, as an effective solution for scalable and efficient RAG.

强化学习多跳问答检索增强小模型

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