Reasoning in Trees: Improving Retrieval-Augmented Generation for Multi-Hop Question Answering
Yuling Shi, Maolin Sun, Zijun Liu, Mo Yang, Yixiong Fang, Tianran Sun, Xiaodong Gu
TL;DR
This work tackles multi-hop question answering with retrieval-augmented generation by introducing Reasoning Tree Guided RAG (RT-RAG), a hierarchical framework that forces explicit problem structure before retrieval. It decomposes questions into a reasoning tree using core queries, known and unknown entities, and then traverses the tree bottom-up, refining and rewriting queries while filtering inconsistent evidence. Key contributions include a consensus-based tree selection mechanism, adaptive leaf decomposition to avoid over-decomposition, rejection sampling to suppress hallucinations, and iterative refinement to ensure coherence across hops. Empirically, RT-RAG achieves state-of-the-art performance on three benchmarks, with average improvements of about 7.0% in F1 and 6.0% in EM, validating the approach as robust across model families and datasets.
Abstract
Retrieval-Augmented Generation (RAG) has demonstrated significant effectiveness in enhancing large language models (LLMs) for complex multi-hop question answering (QA). For multi-hop QA tasks, current iterative approaches predominantly rely on LLMs to self-guide and plan multi-step exploration paths during retrieval, leading to substantial challenges in maintaining reasoning coherence across steps from inaccurate query decomposition and error propagation. To address these issues, we introduce Reasoning Tree Guided RAG (RT-RAG), a novel hierarchical framework for complex multi-hop QA. RT-RAG systematically decomposes multi-hop questions into explicit reasoning trees, minimizing inaccurate decomposition through structured entity analysis and consensus-based tree selection that clearly separates core queries, known entities, and unknown entities. Subsequently, a bottom-up traversal strategy employs iterative query rewriting and refinement to collect high-quality evidence, thereby mitigating error propagation. Comprehensive experiments show that RT-RAG substantially outperforms state-of-the-art methods by 7.0% F1 and 6.0% EM, demonstrating the effectiveness of RT-RAG in complex multi-hop QA.
