STAGE-RAG: Continuous Entity State Evolution over Dual-Space Subgraphs for Multi-Hop Reasoning

Xinran Xu, Shaowen Chen, Yu Ma, Fulin Lin, Zhaofen Wu, Hongwei Wang


Abstract
Maintaining logical coherence in multi-hop Retrieval-Augmented Generation (RAG) remains a challenging problem, as systems must integrate fragmented evidence without suffering reasoning drift. Iterative frameworks enhance information coupling through query refinement, yet often depend on lossy compressed representations that obscure subtle semantics. Conversely, global graph-based methods ensure structural connectivity but carry topological noise and high computational costs. To overcome these issues, we propose STAGE-RAG, a STAte-aware Graph Evolution framework that conceptualizes multi-hop reasoning as continuous Bayesian entity-state evolution on a bi-relational chunk subgraph. By tracking entity states composed of identities and dynamic belief scores, our method implicitly guides reasoning via state-weighted propagation. A dual-path retrieval strategy constructs local subgraphs to prune irrelevant nodes while retaining fine-grained logical dependencies. Evaluations on HotpotQA, MuSiQue, 2WikiMultiHopQA and MoreHopQA indicate that STAGE-RAG consistently surpasses state-of-the-art baselines in both accuracy and efficiency.
Anthology ID:
2026.findings-emnlp.228
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:
4792–4812
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.228/
DOI:
Bibkey:
Cite (ACL):
Xinran Xu, Shaowen Chen, Yu Ma, Fulin Lin, Zhaofen Wu, and Hongwei Wang. 2026. STAGE-RAG: Continuous Entity State Evolution over Dual-Space Subgraphs for Multi-Hop Reasoning. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 4792–4812, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
STAGE-RAG: Continuous Entity State Evolution over Dual-Space Subgraphs for Multi-Hop Reasoning (Xu et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.228.pdf