Build a DAG of chunk synopses and use MCTS to find relevant facts for long‑context QA

August 28, 20257 min

Overview

Decision SnapshotNeeds Validation

The method shows consistent gains on three public long‑context QA sets and ablations; costs and domain generality still need wider testing.

Citations0

Evidence Strength0.70

Confidence0.80

Risk Signals9

Trust Signals

Findings with numeric evidence: 4/4

Findings with evidence refs: 4/4

Results with explicit delta: 5/5

Reproducibility

Status: Partial assets available

Open source: Partial

At A Glance

Cost impact: 40%

Production readiness: 60%

Novelty: 60%

Authors

Zhirui Chen, Wei Shen, Jiashui Huang, Ling Shao

Links

Abstract / PDF / Data

Why It Matters For Business

JERR improves answer accuracy and long‑range recall on long documents while producing an interpretable graph of facts; build once and reuse graphs to amortize cost.

Who Should Care

Summary TLDR

JERR is an agent-style pipeline that turns long documents into chunk synopses, deduplicates entities, builds a directed acyclic graph (DAG) of facts, then uses Monte Carlo Tree Search (MCTS) to pick the most relevant graph nodes for answering questions. On three long‑context QA benchmarks (QuALITY, MuSiQue, NarrativeQA) JERR improves accuracy and recall metrics versus retrieval and agent baselines. It costs more tokens during graph construction but supports graph reuse to cut per‑query cost.

Problem Statement

Transformer LLMs struggle to reason over very long inputs and to retrieve the small set of facts needed for complex questions. We need a practical, interpretable method that filters redundancy, preserves causal links, and finds relevant facts without retraining the LLM.

Main Contribution

JERR: a three-stage pipeline—synopsis extraction, deduplication + DAG graph construction, and MCTS-based graph search for reasoning.

A two-stage deduplication (Bloom Filter + Trie for exact matches, SimHash for near-duplicates) to compress facts before graph building.

Key Findings

JERR yields the best accuracy on QuALITY multi-choice QA.

Numbers86.39% (JERR) vs 85.02% (GraphRAG) (Table 2)

Practical UseUse JERR when you need higher multi‑choice accuracy on long documents; gains are small but consistent.

Evidence RefTable 2

JERR improves long‑range recall on multi‑hop QA (MuSiQue).

NumbersLR-1 0.455 vs 0.410; LR-2 0.595 vs 0.550 (Table 3)

Practical UsePrefer JERR for multi-hop questions where retrieving distant, connected facts matters.

Evidence RefTable 3

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Accuracy86.39%GraphRAG 85.02%+1.37 ppQuALITY devTable 2 shows JERR 86.39% vs baselinesTable 2
MuSiQue LR-1 / LR-2 / F10.455 / 0.595 / 0.505GraphRAG 0.410 / 0.550 / 0.488+0.045 / +0.045 / +0.017MuSiQueTable 3 LR and F1 comparisonsTable 3

What To Try In 7 Days

Chunk a sample long document, extract synopses, and build a small DAG with Bloom+SimHash dedupe to test graph quality.

Run MCTS to pick top‑k nodes (k=5 default) and compare answers to a simple retrieval baseline.

Measure token and latency cost; test graph reuse to see per‑query savings.

Agent Features

Memory
synopsis summaries (short structured memory)
Planning
Monte Carlo Tree Search (MCTS)
Tool Use
autogen chunkingqwen-plus-128ktext-embedding v3
Frameworks
graph-based agent
Is Agentic

Yes

Architectures
DAG (directed acyclic graph)

Optimization Features

Token Efficiency
synopsis compressiongraph reuse to amortize cost

Reproducibility

Code AvailableNo
Data AvailableYes
Open Source StatusPartial
LicenseUnknown

Risks & Boundaries

Limitations

Validated only on QuALITY, MuSiQue and NarrativeQA; generalization to other domains unknown.

Relies on Qwen API for node extraction; graph quality depends on that pipeline.

When Not To Use

When per‑query token or latency budgets are very tight and you cannot amortize graph construction.

For short documents where full-context LLMs already handle the input efficiently.

Failure Modes

Noisy or incorrect node extraction leads to wrong edges and hallucinated answers.

MCTS may focus on keyword overlap and miss semantically relevant nodes if keywords are sparse.

Core Entities

Models

qwen-plus-128kGPT-4-128k

Metrics

AccuracyROUGE-1ROUGE-2ROUGE-LF1LR-1LR-2

Datasets

QuALITYMuSiQueNarrativeQA

Benchmarks

QuALITYMuSiQueNarrativeQA