Weighted Memory Tree: Remembering What Matters for Long-Horizon LLM Agents
On the map: LLM agents and tool use → Agent memory systems
Abstract (arXiv)
Large language model (LLM) agents have demonstrated the ability to solve multi-step tasks requiring planning, tool use, and external information access, yet growing execution histories increase inference cost and expose reasoning to outdated, irrelevant, or misleading information, potentially degrading reasoning quality. Existing memory approaches organize or compress execution histories but provide limited mechanisms for deciding which memories remain active. We introduce the, a hierarchical memory system that organizes execution into tasks, subtasks, and actions while assigning each memory a dynamic retention score. Event-based updates and selection-based decay revise these scores, allowing WMT to preserve useful information, fold completed trajectories, suppress low-utility content, and retain access to folded context. We evaluate WMT on GAIA-Text using Qwen3-8B, Gemma 4 E4B, and Llama-3.1-8B, with ablations and memory-poisoning experiments. Relative to linear memory, WMT improves accuracy by an average of 9.97 percentage points while reducing prompt-token usage by 32.8%. Memory-poisoning experiments show that WMT limits the persistence and propagation of unreliable information. Our results suggest that effective long-horizon agent memory depends less on storing more information than on deciding which information should remain active.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Agent memory systems, one of 16 directions in LLM agents and tool use. Builds on 22 of its 31 references that are in the map. First posted on arXiv 21 Aug 2026.
Builds on
22 of 31 references.
Closest ancestors · 2
Same direction, newest first
- 2025Task Memory Engine (TME): Enhancing State Awareness for Multi-Step LLM Agent Tasks11 Apr
- 2024A Survey on the Memory Mechanism of Large Language Model based Agents21 Apr
Other references · 20
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2025Qwen3 Technical Reportcited by 8,820
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,182
- 2023Generative Agents: Interactive Simulacra of Human Behaviorcited by 3,568
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2023MemGPT: Towards LLMs as Operating Systemscited by 1,389
- 2023GAIA: a benchmark for General AI Assistantscited by 1,128
- 2024AgentPoison: Red-teaming LLM Agents via Poisoning Memory or Knowledge Basescited by 429
- 2026Gemma 4 Technical Reportcited by 259
- 2025Memory Injection Attacks on LLM Agents via Query-Only Interactioncited by 144
- 2024Found in the Middle: Calibrating Positional Attention Bias Improves Long Context Utilizationcited by 101
8 more
- 2025MemoryGraft: Persistent Compromise of LLM Agents via Poisoned Experience Retrievalcited by 72
- 2025From Prompt Injections to Protocol Exploits: Threats in LLM-Powered AI Agents Workflowscited by 58
- 2025Evaluating LLM-based Agents for Multi-Turn Conversations: A Surveycited by 57
- 2024Distance between Relevant Information Pieces Causes Bias in Long-Context LLMscited by 16
- 2025Bias Amplification in RAG: Poisoning Knowledge Retrieval to Steer LLMscited by 8
- 2026INFA-Guard: Mitigating Malicious Propagation via Infection-Aware Safeguarding in LLM-Based Multi-Agent Systemscited by 6
- 2026OpenSec: Measuring Incident Response Agent Calibration Under Adversarial Evidencecited by 1
- 2026Open, Reliable, and Collective: A Community-Driven Framework for Tool-Using AI Agentscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026AkasicMEM: Governed Enterprise Memory for Agentscited by 0
- 2026AkasicMEM: Governed Enterprise Memory for Agents22 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2025Evaluating Memory in LLM Agents via Incremental Multi-Turn Interactions
- 2025Hierarchical Memory for High-Efficiency Long-Term Reasoning in LLM Agents
- 2026ContextWeaver: Selective and Dependency-Structured Memory Construction for LLM Agents
- 2026EvoMemBench: Benchmarking Agent Memory from a Self-Evolving Perspective
- 2026MemGym: a Long-Horizon Memory Environment for LLM Agents
- 2025Rethinking Memory in LLM based Agents: Representations, Operations, and Emerging Topics
- 2026AdMem: Advanced Memory for Task-solving Agents
- 2026Evaluating Memory Structure in LLM Agents