TraceMem: Weaving Narrative Memory Schemata from User Conversational Traces

TL;DR

TraceMem introduces a cognitively-inspired framework that constructs structured narrative memory schemata from user conversational traces, improving long-term dialogue coherence and reasoning in large language models.

Contribution

It presents a novel three-stage pipeline for weaving narrative memory from conversational traces, enhancing LLMs' ability to manage long-term, coherent dialogues.

Findings

Achieves state-of-the-art results on LoCoMo benchmark.

Surpasses baselines in multi-hop reasoning tasks.

Constructs coherent narrative threads from user traces.

Abstract

Sustaining long-term interactions remains a bottleneck for Large Language Models (LLMs), as their limited context windows struggle to manage dialogue histories that extend over time. Existing memory systems often treat interactions as disjointed snippets, failing to capture the underlying narrative coherence of the dialogue stream. We propose TraceMem, a cognitively-inspired framework that weaves structured, narrative memory schemata from user conversational traces through a three-stage pipeline: (1) Short-term Memory Processing, which employs a deductive topic segmentation approach to demarcate episode boundaries and extract semantic representation; (2) Synaptic Memory Consolidation, a process that summarizes episodes into episodic memories before distilling them alongside semantics into user-specific traces; and (3) Systems Memory Consolidation, which utilizes two-stage hierarchical clustering to organize these traces into coherent, time-evolving narrative threads under unifying themes. These threads are encapsulated into structured user memory cards, forming narrative memory schemata. For memory utilization, we provide an agentic search mechanism to enhance reasoning process. Evaluation on the LoCoMo benchmark shows that TraceMem achieves state-of-the-art performance with a brain-inspired architecture. Analysis shows that by constructing coherent narratives, it surpasses baselines in multi-hop and temporal reasoning, underscoring its essential role in deep narrative comprehension. Additionally, we provide an open discussion on memory systems, offering our perspectives and future outlook on the field. Our code implementation is available at: https://github.com/YimingShu-teay/TraceMem