← Latest papers
💬 NLP

ArborMem: Navigating Interaction States with Memory Forests

The paper introduces ArborMem, a framework that models long-running conversations as navigable forests of interaction states to better handle interleaved and resumable tasks, alongside a new diagnostic benchmark called BranchMemEval, demonstrating significant performance improvements over existing baselines in maintaining interaction continuity.

Original authors: Zongwei Lv, Yuemeng Xu, Yilun Yao, Siyi Ding, Xinyu Tan, Yaoming Li, Guangxiang Zhao, Weihong Lin, Lin Sun, Xiangzheng Zhang, Tong Yang

Published 2026-08-19
📖 5 min read🧠 Deep dive

Original authors: Zongwei Lv, Yuemeng Xu, Yilun Yao, Siyi Ding, Xinyu Tan, Yaoming Li, Guangxiang Zhao, Weihong Lin, Lin Sun, Xiangzheng Zhang, Tong Yang

Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer

In the evolving landscape of artificial intelligence, large language models are shifting from tools that answer a single question to companions that remember a life. For these digital assistants to feel truly persistent, they must do more than simply recall facts; they must understand the flow of a conversation that stretches over days or weeks. The core challenge lies in how a machine handles a dialogue where a user might jump between different topics, pause one thread to discuss another, and then return to the first one days later. Traditional systems often struggle here, treating every new message as a fresh start or getting confused by mixing up details from parallel conversations. They might retrieve a fact about a hotel booking when the user is actually asking about a paper revision, simply because the words look similar, failing to grasp that the user has returned to an earlier, distinct line of thought.

Researchers at Peking University and several industry partners have addressed this specific problem with a new framework called ArborMem. Instead of viewing a long conversation as a single, straight line of text, the team designed a system that organizes interactions as a forest of distinct paths. Imagine a conversation not as a river flowing in one direction, but as a collection of separate trails branching off from a central point. Each trail represents a coherent topic or task, such as planning a trip or editing a document. When a user sends a new message, the system first identifies which specific trail the user is walking on again. It then reconstructs the immediate context of that specific path while pulling in useful, reusable facts from other parts of the forest if needed. This approach allows the assistant to maintain continuity without blurring the lines between different subjects.

The team tested this method against existing memory systems using several rigorous benchmarks, including a new diagnostic tool they created specifically to test how well a system handles these branching conversations. In these tests, ArborMem consistently outperformed the strongest previous methods. On established benchmarks for long-term memory, the new system improved accuracy by between 3.36 and 10.31 percentage points. On their own specialized test for branching conversations, it outperformed the next best system by 5.0 points. The advantage was particularly clear when the system was forced to work with a limited amount of information, suggesting that knowing exactly which part of the history to focus on is more important than simply having access to a massive amount of data. Furthermore, the system remained fast, completing its memory searches in less than half a second, even while performing the complex task of locating the correct conversation path.

A key insight driving this success is the separation of "where" a conversation is happening from "what" is being discussed. Previous systems often relied on finding relevant keywords, which works well for simple questions but fails when a user says, "And the revised numbers—did he upload them?" after a long discussion about a conference trip. A keyword-based system might get distracted by the recent talk of travel, whereas ArborMem recognizes that the user is resuming an earlier thread about a paper. By first pinpointing the correct interaction state, the system can restore the local context of that specific branch and then safely add relevant details from elsewhere. This prevents the confusion that arises when two different conversations share similar names or topics but are actually distinct.

The researchers also introduced a new benchmark called BranchMemEval to measure this specific capability. This test creates scenarios where multiple topics are interleaved, requiring the system to distinguish between parallel agendas and resume the correct one after a delay. The results showed that while simply exposing a model to a long history of text is not enough to guarantee reliable reasoning, organizing that history into navigable paths significantly improves performance. The system does not just store information; it actively manages the structure of the dialogue, ensuring that when a user returns to a paused task, the assistant remembers exactly where they left off.

In their experiments, the team found that the system's ability to localize the correct state was the most critical factor. When they removed this feature, the system's accuracy dropped significantly, proving that finding the right context is just as important as retrieving the right facts. The framework also successfully handles updates to information, such as when a user changes a plan or corrects a detail, by tracking these changes within the specific branch where they occurred. This ensures that the assistant does not confuse a new hotel booking with an old, cancelled one. The entire process happens online, meaning the system updates its memory forest in real-time as the conversation progresses, rather than waiting until the end to organize the data.

The findings suggest that for artificial intelligence to become a truly reliable long-term partner, it must move beyond simple retrieval and develop a structural understanding of how human conversations evolve. By treating memory as a forest of interconnected but distinct paths, ArborMem offers a way to preserve the continuity of complex, multi-threaded interactions. The system demonstrates that the key to long-term memory is not just remembering more, but remembering the structure of the relationship between different parts of the conversation. This approach allows the assistant to navigate the twists and turns of a user's life with clarity, ensuring that when a user picks up a thread from days ago, the conversation resumes exactly where it should, without the confusion of mixed-up details or lost context.

Drowning in papers in your field?

Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.

Try Digest →