mlsanityData to 14 Oct 2026

Agent Memory Distillation: Empowering Small LLM Agents with Hierarchical Teacher Memory

arXiv 2608.07169 ↗First posted 7 Aug 2026cs.AI cs.LGcited by 3

On the map: LLM agents and tool use → Agent memory systems

Abstract (arXiv)

Memory systems have shown promise for improving agent performance, but their potential remains largely unexplored for small language models, which struggle to generate sufficient successful trajectories on their own. We propose Agent Memory Distillation (AMD), a training-free framework that transfers structured knowledge from a large teacher agent to a small student agent through hierarchical memory. AMD constructs three complementary memory types from successful teacher trajectories: Workflow memory encodes task-level strategies, Subtask memory provides concrete behavioral examples at an intermediate granularity, and Function memory captures per-function calling conventions and common pitfalls. Workflow and Subtask memories are injected proactively at the start of each task, while Function memory is retrieved reactively upon tool-calling errors. We evaluate AMD on three tool-use benchmarks using four student models (4B-8B parameters) with GPT-5-mini as the teacher, achieving average accuracy gains of 27.2%p, 11.2%p, and 3.4%p on AppWorld, BFCL V3, and ToolSandbox, while consistently outperforming existing memory-based baselines. Further analysis shows that Subtask memory contributes the largest gains, teacher effectiveness depends on both teacher capability and student compatibility, and 4B-sized students benefit most from AMD.

Key facts. Cited by 3 papers 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 25 of its 35 references that are in the map. First posted on arXiv 7 Aug 2026.

Builds on

25 of 35 references.

Closest ancestors · 5

Same direction, newest first

  1. 2026From Storage to Experience: A Survey on the Evolution of LLM Agent Memory Mechanisms7 May
  2. 2026Memory Transfer Learning: How Memories are Transferred Across Domains in Coding Agents15 Apr
  3. 2025MemEvolve: Meta-Evolution of Agent Memory Systems21 Dec
  4. 2025From Experience to Strategy: Empowering LLM Agents with Trainable Graph Memory11 Nov
  5. 2025Sample-Efficient Online Learning in LM Agents via Hindsight Trajectory Rewriting11 Oct

Other references · 20

  1. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  2. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  3. 2023ExpeL: LLM Agents Are Experiential Learnerscited by 876
  4. 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 737
  5. 2019Improved Knowledge Distillation via Teacher Assistantcited by 573
  6. 2024AppWorld: A Controllable World of Apps and People for Benchmarking Interactive Coding Agentscited by 401
  7. 2024ToolSandbox: A Stateful, Conversational, Interactive Evaluation Benchmark for LLM Tool Use Capabilitiescited by 295
  8. 2023Synapse: Trajectory-as-Exemplar Prompting with Memory for Computer Controlcited by 187
  9. 2025From Human Memory to AI Memory: A Survey on Memory Mechanisms in the Era of LLMscited by 124
  10. 2024Small LLMs Are Weak Tool Learners: A Multi-LLM Agentcited by 93
  11. 2024Speculative Knowledge Distillation: Bridging the Teacher-Student Gap Through Interleaved Samplingcited by 90
  12. 2025Distilling LLM Agent into Small Models with Retrieval and Code Toolscited by 47
8 more
  1. 2025AgentDistill: Training-Free Agent Distillation with Generalizable MCP Boxescited by 30
  2. 2024ReflecTool: Towards Reflection-Aware Tool-Augmented Clinical Agentscited by 20
  3. 2024Is In-Context Learning Sufficient for Instruction Following in LLMs?cited by 19
  4. 2026The Evolution of Tool Use in LLM Agents: From Single-Tool Call to Multi-Tool Orchestrationcited by 14
  5. 2026AgentHER: Hindsight Experience Replay for LLM Agent Trajectory Relabelingcited by 13
  6. 2026Experiential Reflective Learning for Self-Improving LLM Agentscited by 13
  7. 2026Structurally Aligned Subtask-Level Memory for Software Engineering Agentscited by 12
  8. 2025Fine-tuning with RAG for Improving LLM Learning of New Skillscited by 5

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026PSD: Pseudo Self-Distillation of Memory Representation Capabilities for LLM Agents
  2. 2025Intrinsic Memory Agents: Heterogeneous Multi-Agent LLM Systems through Structured Contextual Memory
  3. 2026DuoMem: Towards Capable On-Device Memory Agents via Dual-Space Distillation
  4. 2024HiAgent: Hierarchical Working Memory Management for Solving Long-Horizon Agent Tasks with Large Language Model
  5. 2026MemAgent: Learning to Manage Heterogeneous Memory Providers for LLM Agents
  6. 2026Memory as a Controlled Process: Learned Adaptive Memory Management for LLM Agents
  7. 2026AgentBrew: Lifelong Knowledge Brewing from Strong Teachers to Weak LLM Agents
  8. 2025LEGOMem: Modular Procedural Memory for Multi-agent LLM Systems for Workflow Automation