Think-on-Graph 3.0: Efficient and Adaptive LLM Reasoning on Heterogeneous Graphs via Multi-Agent Dual-Evolving Context Retrieval
Problem Statement
GraphRAG systems typically depend on pre-built knowledge graphs that are either manually curated (not scalable) or LLM-extracted (bottlenecked by extractor quality, especially for smaller local models). This limits the practicality of graph-augmented reasoning for organizations that cannot rely on large proprietary LLMs or exhaustive manual KG construction.
Key Novelty
- MACER: a multi-agent mechanism that jointly evolves context during retrieval rather than relying on a static, pre-extracted graph
- Chunk-Triplets-Community heterogeneous graph index unifying raw text, structured triplets, and community-level summaries for multi-granularity evidence
- Dual-Evolution process that adaptively co-refines both the query and the retrieved sub-graph across reasoning iterations
- Query-targeted, on-the-fly graph index construction instead of exhaustive whole-corpus KG extraction, reducing dependence on strong LLM extractors
Evaluation Highlights
- Outperforms compared GraphRAG/ToG baselines on both deep (multi-hop) and broad (wide-coverage) reasoning benchmarks
- Ablation studies isolate and confirm the contribution of individual MACER components
- Demonstrated efficacy with lightweight LLMs, addressing the extraction-quality bottleneck of prior automatic KG-based approaches
Signal Assessment
Methodology
- Initialize a lightweight heterogeneous index over the corpus spanning text chunks, extracted triplets, and community summaries
- Use specialized agents within MACER to incrementally construct and expand a graph index targeted to the specific query rather than the entire corpus
- Apply the Dual-Evolution process to alternately refine the query representation and the retrieved sub-graph over successive reasoning steps
- Feed the iteratively refined sub-graph and query context into the LLM to produce grounded, evidence-based answers
- Evaluate on deep and broad reasoning benchmarks against GraphRAG/ToG baselines, with ablations dissecting MACER's components
System Components
Coordinates multiple agents to iteratively construct, retrieve, and refine reasoning context rather than relying on a fixed pre-built graph.
A hybrid index combining raw text chunks, extracted triplets, and higher-level community summaries to support multi-granularity retrieval.
Adaptively and jointly evolves the query and the retrieved sub-graph in tandem across reasoning iterations for more precise evidence gathering.
Builds a graph index specific to each query on demand, reducing reliance on comprehensive upfront KG extraction and easing use of smaller LLMs.
Results
| Benchmark Type | Baseline Approach | ToG-3 | Delta |
|---|---|---|---|
| Deep (multi-hop) reasoning benchmarks | Prior GraphRAG/ToG baselines | Outperforms baselines | Reported improvement (exact metrics not in abstract) |
| Broad (wide-coverage) reasoning benchmarks | Prior GraphRAG/ToG baselines | Outperforms baselines | Reported improvement (exact metrics not in abstract) |
| Lightweight/local LLM deployment | Degraded due to weak KG extraction | Maintains strong retrieval/reasoning quality | Improved robustness to smaller models |
Key Takeaways
- Practitioners constrained to smaller or local LLMs can achieve strong GraphRAG performance via query-targeted dynamic indexing instead of relying on high-quality full KG extraction
- Combining chunks, triplets, and community summaries in one heterogeneous index can improve retrieval precision and coverage simultaneously
- Iterative, multi-agent co-evolution of query and sub-graph is a reusable pattern applicable to other agentic retrieval/reasoning pipelines
- Open-source release (GitHub) makes it feasible to directly benchmark or integrate ToG-3 into existing RAG stacks
Abstract
Graph-based Retrieval-Augmented Generation (GraphRAG) has become the important paradigm for enhancing Large Language Models (LLMs) with external knowledge. However, existing approaches are constrained by their reliance on high-quality knowledge graphs: manually built ones are not scalable, while automatically extracted ones are limited by the performance of LLM extractors, especially when using smaller, local-deployed models. To address this, we introduce Think-on-Graph 3.0 (ToG-3), a novel framework featuring a Multi-Agent Context Evolution and Retrieval (MACER) mechanism. Its core contribution is the dynamic construction and iterative refinement of a Chunk-Triplets-Community heterogeneous graph index, powered by a Dual-Evolution process that adaptively evolves both the query and the retrieved sub-graph during reasoning. ToG-3 dynamically builds a targeted graph index tailored to the query, enabling precise evidence retrieval and reasoning even with lightweight LLMs. Extensive experiments demonstrate that ToG-3 outperforms compared baselines on both deep and broad reasoning benchmarks, and ablation studies confirm the efficacy of the components of MACER framework. The source code are available in https://github.com/DataArcTech/ToG-3.