Search papers, labs, and topics across Lattice.
This paper introduces MRAgent, a novel framework that enhances long-term reasoning in LLM agents by integrating an associative memory graph with an active reconstruction mechanism. Unlike traditional static memory retrieval methods, MRAgent dynamically adapts memory access based on intermediate evidence, allowing for more contextually relevant reasoning. Experimental results on the LoCoMo and LongMemEval benchmarks show that MRAgent achieves up to 23% improvement over existing models while reducing both token usage and runtime costs.
Active reconstruction in memory-augmented LLMs can boost reasoning performance by over 20% while cutting down on resource consumption.
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 LongMemEval 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.