Search papers, labs, and topics across Lattice.
GAAMA is introduced as a novel graph-augmented associative memory system for AI agents, designed to maintain coherent, personalized behavior across multiple sessions. It constructs a hierarchical knowledge graph from conversations using LLMs to extract atomic facts, topic-level concepts, and higher-order reflections, connected by typed edges. Retrieval combines cosine similarity with edge-type-aware Personalized PageRank, achieving a 78.9% mean reward on the LoCoMo-10 benchmark, outperforming several baselines.
By blending semantic similarity with graph-based traversal, GAAMA unlocks more effective long-term memory retrieval for conversational agents than standard RAG or memory compression techniques.
AI agents that interact with users across multiple sessions require persistent long-term memory to maintain coherent, personalized behavior. Current approaches either rely on flat retrieval-augmented generation (RAG), which loses structural relationships between memories, or use memory compression and vector retrieval that cannot capture the associative structure of multi-session conversations. There are few graph based techniques proposed in the literature, however they still suffer from hub dominated retrieval and poor hierarchical reasoning over evolving memory. We propose GAAMA, a graph-augmented associative memory system that constructs a concept-mediated hierarchical knowledge graph through a three-step pipeline: (1)~verbatim episode preservation from raw conversations, (2)~LLM-based extraction of atomic facts and topic-level concept nodes, and (3)~synthesis of higher-order reflections. The resulting graph uses four node types (episode, fact, reflection, concept) connected by five structural edge types, with concept nodes providing cross-cutting traversal paths that complement semantic similarity. Retrieval combines cosine-similarity-based $k$-nearest neighbor search with edge-type-aware Personalized PageRank (PPR) through an additive scoring function. On the LoCoMo-10 benchmark (1,540 questions across 10 multi-session conversations), GAAMA achieves 78.9\% mean reward, outperforming a tuned RAG baseline (75.0\%), HippoRAG (69.9\%), A-Mem (47.2\%), and Nemori (52.1\%). Ablation analysis shows that augmenting graph-traversal-based ranking (Personalized PageRank) with semantic search consistently improves over pure semantic search on graph nodes (+1.0 percentage point overall).
