Long-horizon reasoning in AI reveals a major challenge: context windows fill quickly, and typical retrieval systems return more noise than useful information. Researchers at the National University of Singapore have introduced MRAgent, a novel memory framework that breaks away from the traditional retrieve-then-reason approach. Instead, MRAgent employs an active memory reconstruction strategy, dynamically building memory during the reasoning process by accumulating evidence step-by-step.
Traditional retrieval methods rely on static structures and fixed similarity scores, which limits their flexibility and often floods the AI’s context with irrelevant data. MRAgent, inspired by cognitive neuroscience, uses a Cue-Tag-Content system to organize memory as an interactive graph. This approach allows it to explore and prune memory paths intelligently during query processing, significantly reducing token usage and computational cost.
For example, when asked how someone used prize money after winning a video game tournament, MRAgent picks up fine-grained cues from the query, navigates relevant tags, retrieves pertinent memories, and gradually hones in on the answer, discarding irrelevant information along the way.
Benchmarked against similar frameworks like A-Mem, MemoryOS, LangMem, and Mem0 on long-horizon tasks, MRAgent not only outperformed them in accuracy but also drastically cut token usage to 118K per query compared to LangMem’s 3.26 million. Runtime was nearly halved, proving its efficiency for real-world applications.
While setting up the Cue-Tag-Content database requires thoughtful design, MRAgent includes an automated pipeline that distills raw user interactions into the graph without manual tagging. The framework’s code is openly available on GitHub, offering practical tools for developers to build scalable, context-aware AI agents.
