T-Retriever: Tree-based Hierarchical Retrieval Augmented Generation for Textual Graphs

Authors

  • Chunyu Wei Renmin University of China
  • Huaiyu Qin Renmin University of China
  • Siyuan He Renmin University of China
  • Yunhai Wang Renmin University of China
  • Yueguo Chen Renmin University of China

DOI:

https://doi.org/10.1609/aaai.v40i19.38625

Abstract

Retrieval-Augmented Generation (RAG) has significantly enhanced Large Language Models' ability to access external knowledge, yet current graph-based RAG approaches face two critical limitations in managing hierarchical information: they impose rigid layer-specific compression quotas that damage local graph structures, and they prioritize topological structure while neglecting semantic content. We introduce T-Retriever, a novel framework that reformulates attributed graph retrieval as tree-based retrieval using a semantic and structure-guided encoding tree. Our approach features two key innovations: (1) Adaptive Compression Encoding, which replaces artificial compression quotas with a global optimization strategy that preserves the graph's natural hierarchical organization, and (2) Semantic-Structural Entropy (S²-Entropy), which jointly optimizes for both structural cohesion and semantic consistency when creating hierarchical partitions. Experiments across diverse graph reasoning benchmarks demonstrate that T-Retriever significantly outperforms state-of-the-art RAG methods, providing more coherent and contextually relevant responses to complex queries.

Downloads

Published

2026-03-14

How to Cite

Wei, C., Qin, H., He, S., Wang, Y., & Chen, Y. (2026). T-Retriever: Tree-based Hierarchical Retrieval Augmented Generation for Textual Graphs. Proceedings of the AAAI Conference on Artificial Intelligence, 40(19), 15924–15932. https://doi.org/10.1609/aaai.v40i19.38625

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management III