Enhancing Retrieval-Augmented Generation via Evidence Tree Search
Hao Sun, Hengyi Cai, Yuchen Li, Xuanbo Fan, Xiaochi Wei, Shuaiqiang Wang, Yan Zhang, Dawei Yin
Abstract
Retrieval-Augmented Generation (RAG) is widely used to enhance Large Language Models (LLMs) by grounding responses in external knowledge. However, in real-world applications, retrievers often return lengthy documents with redundant or irrelevant content, confusing downstream readers. While evidence retrieval aims to address this by extracting key information, it faces critical challenges: (1) inability to model synergistic inter-dependencies among evidence sentences, (2) lack of supervision for evaluating multi-sentence evidence quality, and (3) computational inefficiency in navigating exponentially growing search spaces of candidate evidence sets. To tackle these challenges, we propose ETS (Evidence Tree Search), a novel framework that reformulates evidence retrieval as a dynamic tree expansion process. Our approach first constructs an evidence tree where each path represents a candidate evidence set, explicitly modeling inter-sentence dependencies through context-aware node selection. We then leverage Monte Carlo Tree Search (MCTS) to efficiently assess evidence quality and introduce an Early-Terminating Beam Search strategy to efficiently accelerate the model inference. Extensive experiments on five datasets demonstrate that ETS significantly outperforms existing methods across different readers. Our code and datasets will be released to facilitate future research.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 818164d0-4fea-4146-b4f6-e302a4701a26Cited by top-tier papers2
- PruneRAG: Confidence-Guided Query Decomposition Trees for Efficient Retrieval-Augmented GenerationShuguang Jiao, Xinyu Xiao, Yunfan Wei, Shuhan Qi et al.WWW 2026 · 2 citations
- HiGoE: Hierarchical Graph of Evidence to Enhance Retrieval-Augmented Generation for Long-context SummarizationLong Yuan, Kaiwen Tian, Zi Chen, Bolong Zheng et al.ACL 2026
Builds on12
- Retrieval Augmented Language Model Pre-TrainingKelvin Guu, Kenton Lee, Zora Tung, Panupong Pasupat et al.ICML 2020 · 2,937 citations
- Solving Quantitative Reasoning Problems with Language ModelsAitor Lewkowycz, Anders Andreassen, David Dohan, Ethan Dyer et al.NeurIPS 2022 · 2,039 citations
- Self-RAG: Learning to Retrieve, Generate, and Critique through Self-ReflectionAkari Asai, Zeqiu Wu, Yizhong Wang, Avirup Sil et al.ICLR 2024 · 1,798 citations
- Active Retrieval Augmented GenerationZhengbao Jiang, Frank F. Xu, Luyu Gao, Zhiqing Sun et al.EMNLP 2023 · 315 citations
- LongLoRA: Efficient Fine-tuning of Long-Context Large Language ModelsYukang Chen, Shengju Qian, Haotian Tang, Xin Lai et al.ICLR 2024 · 254 citations
Related papers
- RetroLLM: Empowering Large Language Models to Retrieve Fine-grained Evidence within GenerationXiaoxi Li, Jiajie Jin, Yujia Zhou, Yongkang Wu et al.ACL 2025
- CARROT: A Learned Cost-Constrained Retrieval Optimization System for RAGZiting Wang, Haitao Yuan, Wei Dong, Gao Cong et al.ICDE 2026 · 1 citation
- DecoupleSearch: Decouple Planning and Search via Hierarchical Reward ModelingHao Sun, Zile Qiao, Bo Wang, Guoxin Chen et al.EMNLP 2025
- Iterative Multi-Granular RAG with Contextual Hierarchical GraphYanli Hu, Teng Liu, Zhuangyi Zhou, Weixin Zeng et al.AAAI 2026
- HAMMER: An Automatic RAG Tuning System via Hierarchical Memory-Guided Monte Carlo Tree SearchYingli Zhou, Zixuan Wang, Yixiang FangSIGMOD 2026 · 1 citation
