Search papers, labs, and topics across Lattice.
This paper introduces PGMem, a heterogeneous persona-memory graph designed to enhance the tracking of user preferences in long-term personalized dialogue agents. By tightly coupling event and persona nodes through typed provenance and evidence edges, PGMem addresses the memory-persona validity gap and improves the retrieval of persona-aware information. Experimental results across three benchmarks demonstrate that PGMem outperforms existing memory systems, showing improved performance as context increases.
PGMem bridges the gap between user events and evolving personas, leading to more accurate and personalized dialogue interactions.
Long-term personalized dialogue agents must track user preferences as their personas evolve. Existing memory systems organize past events well, but store personas as flat profiles detached from the events that justify them. This loose coupling leads to the memory-persona validity gap and the persona-aware retrieval gap. We propose PGMem, a heterogeneous persona-memory graph that connects event and persona nodes through typed provenance and evidence edges, keeping each persona signal traceable to the events that support or revise it. At retrieval time, PGMem expands from query-relevant seeds and ranks signals by evidential validity. Across three benchmarks with small language model backbones, PGMem consistently outperforms summary-based, persona-aware, graph-structured, and agentic memory baselines, and improves performance as the context grows. The source code of PGMem is available at https://github.com/wonjunchoi23/pgmem/
