Entity-Memory Graph Retrieval Improves Evidence Coverage in Long-Conversation Question Answering
Organizations: Tsinghua University
Abstract
Entity-Memory graph retrieval keeps dialogue turns as verbatim Memory nodes, links repeated mentions through shared Entities, and connects adjacent Memories with directed chronological edges. At query time the retriever moves from Entity gating through semantic fusion and one-hop chronological recovery to dense backfill. The path can keep a neighboring Memory that dense cosine ranking would otherwise omit. A matched dense control shares the Memory and query vectors, context budget, requested answer protocol, and evaluator, isolating graph structure from changes to the reader. On 1,986 questions from ten LoCoMo conversations, graph retrieval raises official evidence recall at top-k 25 from 79.7468% to 84.4842%. The recall advantage is supported from top-k 5 to 50, while no matched cutoff supports an overall final-answer F1 difference. Four paper-eligible requested configurations support empirical robustness across the tested GPT-3.5 and DeepSeek extractors on both outcomes. Embedding robustness is mixed: F1 has no supported contrast, but recall is sensitive to the embedding artifact. The comparison isolates a retrieval-coverage gain from graph structure. It does not establish a final-answer F1 gain, model or embedding equivalence, or cross-dataset generalization.
Figures & tables
| Family | Treatment and control | Fixed factors | Purpose |
|---|---|---|---|
| Structural components | Remove sequence expansion, Entity-score fusion, semantic scoring, or speaker links from complete B | Dataset, Memory text, primary vectors, top-k 25, requested reader protocol, frozen evaluator | Attribute the primary retrieval difference |
| Cutoff and input | Matched A/B at top-k 5, 10, 25, 50; raw-text versus time-annotated input | Paired rows and within-profile artifacts | Test context budget and input-profile dependence |
| Parameters | Entity/Semantic weights 0.10/0.90, 0.30/0.70, 0.50/0.50; sequence scale 0.25, 0.5, 1.0 | Complete-B graph and reader | Bound local sensitivity without claiming universal optima |
| Robustness across configurations | Requested gpt-3.5-turbo or deepseek-v4-flash Entity extraction crossed with TES or Doubao embeddings | Complete B, top-k 25, requested extraction budget 2,500, requested GPT-3.5 reader protocol | Test extractor and embedding substitutions in a paper-eligible matrix with outcome-specific robustness claims |
| Diagnostic only | Two gpt-5-mini extraction cells | Reader and retrieval settings fixed, but runtime extraction budget was 4,000 rather than the frozen 2,500 | Excluded from model-only inference pending rerun |
| Condition | Overall F1 | recall_acc |
|---|---|---|
| A (dense Memory) | 42.0681 | 79.7468 |
| (dense graph control) | 42.0677 | 79.7468 |
| B (complete graph retrieval) | 42.5680 | 84.4842 |
| Contrast | F1 | recall_acc | Conclusion |
|---|---|---|---|
| B B without sequence expansion | Recall supported; F1 not supported | ||
| B B without Entity-score fusion | Recall supported; F1 not supported | ||
| Sequence expansion after Entity gating | Recall supported; F1 not supported | ||
| B Entity-only retrieval | Both outcomes supported | ||
| Raw-text B annotated-input B | Neither outcome supported | ||
| Raw-text B raw-text A | Recall supported; F1 not supported |
| top-k | A F1 | B F1 | F1 | A recall | B recall | recall | Paired recall CI | Cluster recall CI |
|---|---|---|---|---|---|---|---|---|
| 5 | 39.9053 | 39.8264 | 59.3584 | 64.1667 | [3.3450, 6.2808] | [2.8809, 6.6134] | ||
| 10 | 41.7607 | 42.1412 | 68.9145 | 74.4847 | [4.2928, 6.8624] | [4.2365, 6.7522] | ||
| 25 | 42.0681 | 42.5680 | 79.7468 | 84.4842 | [3.6504, 5.8425] | [3.5286, 6.0124] | ||
| 50 | 42.0782 | 41.7832 | 86.7148 | 90.3306 | [2.7754, 4.4735] | [2.4603, 4.6426] |
| Requested extractor | Embedding alias | F1 | recall_acc |
|---|---|---|---|
| GPT-3.5 | TES | 42.5680 | 84.4842 |
| DeepSeek v4 Flash | TES | 42.1087 | 84.6005 |
| GPT-3.5 | DB | 42.1507 | 82.1978 |
| DeepSeek v4 Flash | DB | 42.8158 | 82.5778 |
| Category | B F1 range | Dialog F1 | LightGMEM F1 | B recall range | Dialog recall | Descriptive position |
|---|---|---|---|---|---|---|
| Multi-hop | 37.6–39.9 | 38.7 | 41.9 | 61.5–67.3 | 62.5 | B F1 range overlaps the Dialog anchor |
| Temporal | 40.2–43.1 | 37.2 | 56.4 | 89.2–90.2 | 83.5 | B recall is numerically above the Dialog anchor |
| Open-domain | 17.3–18.6 | 25.0 | 27.2 | 48.4–56.4 | 52.6 | Lowest B answer F1 range among Categories 1–4 |
| Single-hop | 63.8–64.1 | 59.9 | 64.1 | 91.1–92.7 | 87.1 | Highest B answer F1 range |
| Adversarial | 8.3–11.2 | 12.8 | — | 80.2–82.6 | 66.3 | B recall is numerically above Dialog; raw B F1 is lowest under the separate Category-5 branch |
| Provider stage | Cold req. | Input tokens | Output tokens | Cold call time | Warm req. |
|---|---|---|---|---|---|
| Conversation Entity extraction | 5,873 | 3,731,393 | 425,637 | 9,792.76 s | 0 |
| Memory embedding | 591 | 214,229 | 0 | 443.38 s | 0 |
| Question Entity extraction | 1,974 | 1,214,259 | 104,272 | 2,970.83 s | 0 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Dimension | Resolved value |
|---|---|
| Dataset | LoCoMo ten-conversation release; 1,986 QA rows |
| Graph inputs | Session time, dialogue id, speaker, text, image caption |
| Excluded graph inputs | Questions, answers, evidence, categories, judges, predictions |
| Memory nodes | 5,882 |
| Conversation Entity extractor | gpt-3.5-turbo ; temperature 0.3 |
| Memory embedding model | text-embedding-3-small |
| Gate | Requirement | Outcome |
|---|---|---|
| Graph construction | Conversation-only inputs; no QA-derived construction fields | Pass |
| Evaluator integrity | All 16 vendored files match the frozen SHA-256 manifest | Pass |
| Output isolation | One absent or empty directory per resolved formal condition | Pass |
| Query vectors | Complete ordered coverage; zero miss; zero live query embedding | Pass |
| Dense control | A and ordered contexts equal for all 1,986 rows | Pass |
| Prompt budget | Non-data scaffold no longer than 5,000 characters | Pass (2,410) |