Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Search Wisely: Mitigating Sub-optimal Agentic Searches By Reducing Uncertainty

About

Agentic Retrieval-Augmented Generation (RAG) systems enhance Large Language Models (LLMs) by enabling dynamic, multi-step reasoning and information retrieval. However, these systems often exhibit sub-optimal search behaviors like over-search (retrieving redundant information) and under-search (failing to retrieve necessary information), which hinder efficiency and reliability. This work formally defines and quantifies these behaviors, revealing their prevalence across multiple QA datasets and agentic RAG systems (e.g., one model could have avoided searching in 27.7% of its search steps). Furthermore, we demonstrate a crucial link between these inefficiencies and the models' uncertainty regarding their own knowledge boundaries, where response accuracy correlates with model's uncertainty in its search decisions. To address this, we propose $\beta$-GRPO, a reinforcement learning-based training method that incorporates confidence threshold to reward high-certainty search decisions. Experiments on seven QA benchmarks show that $\beta$-GRPO enable a 3B model with better agentic RAG ability, outperforming other strong baselines with a 4% higher average exact match score.

Peilin Wu, Mian Zhang, Xinlu Zhang, Xinya Du, Zhiyu Zoey Chen• 2025

Related benchmarks

TaskDatasetResultRank
Multi-hop Question AnsweringMulti-Hop QA (HotpotQA, 2Wiki, Musique, Bamboogle) (test)
HotpotQA Score48.02
60
Single-hop Question AnsweringSingle-Hop Suite (NQ, TriviaQA, PopQA) (test)
NQ Accuracy47.89
16
Multi-hop Question AnsweringBamboogle
CEM (%)54.4
12
Multi-hop Question Answering2Wiki
CEM (%)66
12
Question AnsweringPopQA
CEM (%)61.3
12
Question AnsweringNQ
CEM62.4
12
Question AnsweringTriviaQA
CEM73.9
12
Multi-hop Question AnsweringMuSiQue
CEM (%)22.8
12
Multi-hop Question AnsweringHotpotQA
CEM52.1
12
Showing 9 of 9 rows

Other info

Follow for update