GRASP: GRanularity-Aware Search Policy for Agentic RAG

Varun Gandhi, Jaewook Lee, Shantanu Todmal, Franck Dernoncourt, Ryan A. Rossi, Andrew Lan, Zichao Wang


Abstract
Agentic retrieval-augmented generation (RAG) extends RAG by allowing language models to iteratively reason, generate search queries, retrieve evidence, and predict answers. However, it remains challenging for models to decide when to retrieve, whether to use lexical matching or semantic similarity, and how to control context granularity to prevent irrelevant tokens from interfering with agent reasoning. In this paper, we introduce GRASP, a reinforcement learning framework for training agents to adaptively coordinate complementary retrieval tools during multi-step reasoning. GRASP provides the agent with semantic search, keyword search, and paragraph-reading actions, enabling it to retrieve sentence-level evidence and expand context only when needed. We train the policy with a reward that accounts for answer accuracy, grounded reading, complementary search, and turn efficiency. Experiments on multi-hop reasoning benchmarks across 3B and 7B models show that GRASP improves retrieval and question answering performance compared with single-step retrieval and multi-step agentic RAG baselines. Qualitative and ablation analyses show that the learned policy develops interpretable skimming and scanning behavior: it uses semantic search for broad exploration, paragraph reading for local verification, and keyword search for entity-specific evidence. These results suggest that learning to coordinate retrieval signals and context granularity is critical for agent’s reasoning.
Anthology ID:
2026.findings-emnlp.1226
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
25944–25975
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1226/
DOI:
Bibkey:
Cite (ACL):
Varun Gandhi, Jaewook Lee, Shantanu Todmal, Franck Dernoncourt, Ryan A. Rossi, Andrew Lan, and Zichao Wang. 2026. GRASP: GRanularity-Aware Search Policy for Agentic RAG. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 25944–25975, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
GRASP: GRanularity-Aware Search Policy for Agentic RAG (Gandhi et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1226.pdf