Search papers, labs, and topics across Lattice.
This paper addresses the limitations of current retrieval methods in personalized Large Language Models (LLMs) that rely heavily on semantic similarity, which can overlook important logical memories. To tackle this, the authors introduce IMLogic, a benchmark for evaluating implicit logical memory retrieval, and propose RootMem, a framework that distills user histories into structured root memories and employs an LLM-based router to enhance memory retrieval. The results show that RootMem significantly outperforms existing retrieval methods, improving the accuracy of memory agents in long-dialogue scenarios.
RootMem transforms how personalized LLMs access critical logical memories, outperforming traditional methods by integrating structured decision logic into memory retrieval.
Memory systems are essential for personalized Large Language Models (LLMs). However, existing retrieval methods in these systems primarily rely on semantic similarity, potentially missing logically critical memories with limited semantic overlap. Current benchmarks remain inadequate for evaluating this problem. To address this gap, we construct IMLogic, the first high-quality benchmark targeting implicit logical memory retrieval in long-dialogue scenarios. Motivated by this challenge, we introduce root memory, a structured, decision-preserving representation that distills reusable personalized logic from long-term user histories. We then propose RootMem, a plug-and-play framework that first distills raw histories into structured root memories and then uses an LLM-based router to activate logically relevant ones, complementing semantic retrieval with personalized decision logic. Extensive experiments demonstrate that RootMem significantly outperforms the strongest retrieval baselines and consistently boosts the accuracy of existing memory agents. Our benchmark and codes will be available at https://anonymous.4open.science/r/IMLogic-DBB3.
