Search papers, labs, and topics across Lattice.
This study addresses the challenge of determining when to stop multi-round retrieval-augmented generation (RAG) by employing a structured sufficiency-and-gap judgment approach. By training a Qwen3.5-2B judge on 3,009 states derived from HotpotQA questions, the authors optimize the retrieval process within a fixed Search-R1 framework, achieving a significant reduction in retrieval calls by 3.70%. While this method effectively decreases the number of retrieval operations, it results in a slight decline in Official Exact Match accuracy, highlighting the trade-off between efficiency and precision in RAG systems.
A structured judgment approach can cut retrieval calls by 77 while only slightly impacting answer accuracy in multi-round RAG systems.
Multi-round retrieval-augmented generation (RAG) must decide when to stop searching as evidence accumulates. Because the deployed policy is determined by the first STOP on each trajectory, this is a sequential selection problem rather than an independent state-classification task. We adapt S2G-RAG's structured sufficiency-and-gap judgment to a frozen Search-R1 pipeline and train a Qwen3.5-2B judge on 3,009 states from 900 disjoint HotpotQA questions. Search-R1's reasoner, retriever, corpus, prompt, and search budget remain unchanged, while the judge checkpoint and stopping threshold are selected on grouped validation and frozen before confirmatory evaluation. On the confirmatory test set, the resulting policy reduces retrieval calls by 77 (3.70\%) relative to Native Search-R1, while Official Exact Match decreases by 0.625 percentage points. Thus, the trained S2G-style structured judge reduces retrieval while broadly preserving answer accuracy. The result does not imply unchanged or improved accuracy, safe stopping, or lower total inference cost.
