Search papers, labs, and topics across Lattice.
This paper introduces ReTree, a self-correcting tree-structured memory mechanism designed to enhance the performance of large language model (LLM)-based search agents by managing context growth and preserving evidence integrity. By modeling search as an evidence tree, ReTree effectively maintains a bounded reasoning context and allows for the dynamic replacement of outdated information while ensuring traceability to original sources. Experimental results demonstrate that ReTree significantly improves answer accuracy by up to 25.6 percentage points compared to existing methods, while also reducing the average maximum per-step reasoning context by 1.27 to 1.51 times.
ReTree not only boosts answer accuracy by up to 25.6 percentage points but also streamlines context management for long-horizon search agents.
Large language model (LLM)-based search agents answer questions through multi-step interactions with external environments. However, providing complete execution trajectories to the LLM causes unbounded context growth and introduces noise. Existing compression methods reduce context at the cost of important details and often replace erroneous facts without repairing downstream reasoning derived from them. To address this problem, we propose ReTree, a self-correcting tree-structured memory mechanism for search agents. ReTree constructs a bounded per-step reasoning context while preserving source-linked evidence. It models search as an evidence tree whose nodes store bounded summaries, evidence, and revision histories. When newly retrieved evidence contradicts an earlier claim, ReTree traces back to the node where the claim was introduced, replaces outdated evidence, regenerates summaries, prunes affected branches, and resumes search. Source-grounded evidence provenance supports reliable conflict localization and keeps final claims traceable to retrieved passages. Experiments on four public question-answering and search benchmarks show that ReTree consistently outperforms Full-Trajectory ReAct, improving answer accuracy by up to 25.6 percentage points (pp); the average maximum per-step reasoning context of Full-Trajectory ReAct is $1.27$--$1.51\times$ that of ReTree. These results establish ReTree as an effective self-correcting memory abstraction for long-horizon search.
