Memory is Reconstructed, Not Retrieved: Graph Memory for LLM Agents

Paper · arXiv 2606.06036 · Published June 4, 2026
LLM Memory

Despite recent progress, LLM agents still struggle with reasoning over long interaction histories. While current memory-augmented agents rely on a static “retrieve-then-reason” paradigm, this rigid pipeline design prevents them from dynamically adapting memory access to intermediate evidence discovered during inference. To bridge this gap, we propose MRAgent, a framework that combines an associative memory graph with an active reconstruction mechanism. We represent memory as a Cue–Tag–Content graph, where associative tags serve as semantic bridges connecting fine-grained cues to memory contents. Operating on this structure, our active reconstruction mechanism integrates LLM reasoning directly into memory access, allowing the agent to iteratively explore and prune retrieval paths based on accumulated evidence. This ensures that memory retrieval is dynamically adapted to the reasoning context while avoiding combinatorial explosion caused by unconstrained expansion. Experiments on the LOCOMO benchmark and LONG- MEMEVAL benchmark demonstrate significant improvements over strong baselines (up to 23%), while substantially reducing token and runtime cost, highlighting the effectiveness of active and associative reconstruction for long-horizon memory reasoning.

Introduction. LLMs exhibit a “jagged” cognitive profile (Hendrycks et al., 2025), excelling at math and reasoning, but deficient in tasks requiring long-term memory, such as interactive assistance or decision-support systems acting over extended interactions (Gao et al., 2026). In such long-horizon tasks, LLMs are fundamentally constrained by their limited context windows, which restrict their ability to retain interaction history over time (Hatalis et al., 2023). To mitigate these limitations, prior work equips LLM agents with external memory systems. Early approaches adopt Retrieval-Augmented Generation (RAG) (Lewis et al., 2020), where memory access is realized via similarity-based retrieval over unstructured text or embedding stores. Subsequent work introduces more structured memory representations, including hierarchical stores (Fang et al., 2025; Kang et al., 2025) and knowledge graphs (KGs) (Rasmussen et al., 2025; Xu et al., 2025; Huang et al., 2025), which explicitly encode entities and relations to support more interpretable and relational retrieval.

Discussion / Conclusion. We proposed MRAgent, a reconstructive memory agent that formulates memory access as an active, multi-step reconstruction process over a structured memory graph. A key design choice of MRAgent is to shift the modeling of complex relational dependencies to the retrieval stage, enabling the agent to resolve more complex queries with fewer computational cost through targeted, state-dependent exploration. As a result, our current implementation adopts a relatively simple memory construction strategy, without introducing additional complexity in memory updating or forgetting mechanisms. This design choice also exposes several limitations that suggest directions for future work. First, because relational reasoning is deferred to retrieval, the cost of reconstruction grows with the depth of exploration, and queries that require many traversal steps incur higher latency than single-shot retrieval.