HSEARCH-R1: Learning Compact Value Models for LLM-guided Hierarchical Search via RL

Nilesh Gupta, Nasib Ullah, Jinbin Zhang, Rohit Babbar, Inderjit S Dhillon


Abstract
Reasoning-intensive retrieval often requires evidence whose relevance is not captured by keyword or embedding similarity alone, exposing a core limitation of traditional IR systems. An emerging direction known as LLM-guided hierarchical search addresses this limitation by letting an LLM directly traverse a semantic tree over the corpus. Such methods so far work well only with a frontier LLM in the search loop, which makes inference expensive. We propose HSearch-R1, a distillation recipe for learning compact (4B) LLMs that perform efficient LLM-guided hierarchical search. The recipe distils a frontier teacher’s per-step value estimates from its search trajectories into a much smaller LLM (Qwen3.5-4B) via RL, using a composite reward that matches the teacher’s ranking and score distribution at each step; this outcome-level RL distillation substantially outperforms supervised fine-tuning on teacher outputs. We further propose a retrieval-prior-informed variant of the search that injects a cheap retriever’s shortlist as a prior into the tree and adds robustness when the offline index is stale. The base HSearch-R1-4B matches or closely tracks frontier LLMs on the seven static BRIGHT-StackExchange subsets at roughly 50× lower per-query cost. Adding the retrieval prior brings HSearch-R1 to 52.5 nDCG@10 on StackExchange and 45.6 across all 12 BRIGHT subsets, within 1 nDCG of the strongest published ensembled baselines while remaining substantially cheaper.
Anthology ID:
2026.emnlp-main.2494
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
56623–56642
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2494/
DOI:
Bibkey:
Cite (ACL):
Nilesh Gupta, Nasib Ullah, Jinbin Zhang, Rohit Babbar, and Inderjit S Dhillon. 2026. HSEARCH-R1: Learning Compact Value Models for LLM-guided Hierarchical Search via RL. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 56623–56642, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
HSEARCH-R1: Learning Compact Value Models for LLM-guided Hierarchical Search via RL (Gupta et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2494.pdf