Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MemSearch-o1: Empowering Large Language Models with Reasoning-Aligned Memory Growth in Agentic Search

About

Recent advances in large language models (LLMs) have scaled the potential for reasoning and agentic search, wherein models autonomously plan, retrieve, and reason over external knowledge to answer complex queries. However, the iterative think-search loop accumulates long system memories, leading to memory dilution problem. In addition, existing memory management methods struggle to capture fine-grained semantic relations between queries and documents and often lose substantial information. Therefore, we propose MemSearch-o1, an agentic search framework built on reasoning-aligned memory growth and retracing. MemSearch-o1 dynamically grows fine-grained memory fragments from memory seed tokens from the queries, then retraces and deeply refines the memory via a contribution function, and finally reorganizes a globally connected memory path. This shifts memory management from stream-like concatenation to structured, token-level growth with path-based reasoning. Experiments on eight benchmark datasets show that MemSearch-o1 substantially mitigates memory dilution, and more effectively activates the reasoning potential of diverse LLMs, establishing a solid foundation for memory-aware agentic intelligence.

Sheng Zhang, Junyi Li, Yingyi Zhang, Pengyue Jia, Yichao Wang, Xiaowei Qian, Wenlin Zhang, Maolin Wang, Yong Liu, Xiangyu Zhao• 2026

Related benchmarks

TaskDatasetResultRank
Long-context Question AnsweringLongBench v2
Overall Accuracy42.31
33
Multi-document Question AnsweringMultiDocQA
HotpotQA Accuracy67.78
14
Single-document Question AnsweringSingledoc QA
DuReader Score32.06
14
Question AnsweringLongBookQA-zh (test)
F139.44
5
Question AnsweringLongBookQA en
F1 Score25.04
5
Showing 5 of 5 rows

Other info

Follow for update