From Passive Retrieval to Active Memory Navigation: Learning to Use Memory as a Structured Action Space
Summary
NapMem is a novel framework that transforms how personalized conversational agents interact with long-term user memory, shifting from passive retrieval to active memory navigation. It conceptualizes memory as a structured action space, organizing user history into a "linked multi-granularity memory pyramid." This pyramid includes raw conversations, typed memory records, topic tracks, and user profiles, all interconnected by provenance relations and exposed via memory tools. Agents are trained using memory-tool reinforcement learning to dynamically select memory granularity based on query and intermediate evidence. Experiments on PersonaMem-v2, LongMemEval, and LoCoMo benchmarks demonstrate NapMem's competitive performance on diverse memory-intensive tasks, while also preserving general reasoning and tool-use abilities on non-memory tasks. Further analyses explored storage, inference cost, tool-use behavior, and ablations over navigation, memory granularity, and RL training, suggesting benefits from coupling structured storage with a learned policy for appropriate memory use.
Key takeaway
For AI Scientists developing personalized conversational agents, consider adopting an active memory navigation approach. Your current passive retrieval systems may limit agent performance on complex, memory-intensive tasks. Implementing a multi-granularity memory pyramid with a learned policy, as demonstrated by NapMem, can significantly enhance an agent's ability to utilize long-term user history effectively, improving personalization and reasoning across diverse interactions.
Key insights
NapMem enables conversational agents to actively navigate structured, multi-granularity memory using a learned policy.
Principles
- Memory as a structured action space.
- Multi-granularity memory improves utility.
- Learned policy for memory selection.
Method
NapMem organizes user history into a linked multi-granularity memory pyramid and trains agents with memory-tool reinforcement learning to select appropriate memory granularity.
In practice
- Implement memory as a pyramid structure.
- Train agents with RL for memory tool use.
- Evaluate on memory-intensive benchmarks.
Topics
- Conversational AI
- Long-term Memory
- Reinforcement Learning
- Memory Navigation
- Multi-granularity Memory
- Personalized Agents
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.