CAGE: Coherence-Aware Graph Encoding for Retrieval-Augmented Generation

Tong Qi, Jingyu Wu, Youbing Yin, Spencer Hong, Daben Liu, Erin Babinsky


Abstract
Traditional Retrieval-Augmented Generation (RAG) systems score each passage independently against the query, assembling context sets that may be individually relevant yet collectively incoherent. We introduce Coherence-Aware Graph Encoding (CAGE), a reranking framework that models "between-chunk coherence" across four dimensions: Intra-Domain Relevance, Noise Resistance, Informational Bonding, and Factual Consistency. Our pipeline transforms retrieved passages into directed heterogeneous entity graphs, amplifies factual anchors via min-out-degree reweighting, encodes structural patterns through a Relational Graph Convolutional Network, and fuses inter-chunk coherence with query relevance for final ranking. Evaluated across four multi-hop benchmarks, CAGE matches or outperforms strong baselines including monoT5 in Recall@5 on bridge-dominated datasets and consistently improves downstream Exact Match, demonstrating that structurally coherent context yields more precise answers even when retrieval recall is comparable or lower.
Anthology ID:
2026.findings-emnlp.1254
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:
26512–26530
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1254/
DOI:
Bibkey:
Cite (ACL):
Tong Qi, Jingyu Wu, Youbing Yin, Spencer Hong, Daben Liu, and Erin Babinsky. 2026. CAGE: Coherence-Aware Graph Encoding for Retrieval-Augmented Generation. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 26512–26530, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
CAGE: Coherence-Aware Graph Encoding for Retrieval-Augmented Generation (Qi et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1254.pdf