AIAny
Icon for item

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

Transfers hierarchical, structured memory from a large teacher agent into small LLM agents to improve tool-use success. Constructs Workflow, Subtask and Function memories with proactive/reactive injection; training-free and validated on multiple tool-use benchmarks.

Introduction

Most small LLM agents fail to produce enough successful trajectories on their own to learn robust tool-use behaviors. The core insight of this work is that a large, capable teacher agent encodes reusable procedural knowledge in its successful interactions, and that those patterns can be packaged as hierarchical memories and injected into smaller students without additional training.

Key Findings
  • Three complementary memory types: Workflow memory (task-level strategies), Subtask memory (intermediate-granularity behavioral examples), and Function memory (per-function calling conventions and common pitfalls). This split yields targeted guidance at different abstraction levels so students get both strategy and concrete examples.
  • Training-free distillation with proactive (Workflow/Subtask at task start) and reactive (Function retrieved on tool-call errors) injection. This design reduces compute and engineering overhead compared to trajectory imitation or full-policy distillation while remaining broadly applicable.
  • Empirical gains: across AppWorld, BFCL V3 and ToolSandbox, average accuracy improvements reported are +27.2, +11.2 and +3.4 percentage points respectively when using GPT-5-mini as teacher with 4B–8B student models. Subtask memory provides the largest single contribution to improvements.
  • Analysis shows teacher efficacy depends on both teacher capability and student compatibility; smaller (4B) students see the largest relative benefit, while mismatched teacher-student pairs reduce gains.
Who it's for and tradeoffs

Great fit if you need to boost tool-use or multi-step agent performance for resource-constrained LLMs without retraining—especially in settings with a strong teacher agent available and standardized tool APIs. Look elsewhere if you require end-to-end policy learning from scratch, continuous online adaptation of student weights, or if your use case cannot be expressed as reusable subtask/function patterns. The approach trades off absolute optimality for low-cost, modular transferability and depends on the quality and compatibility of teacher trajectories.

Information

  • Websitearxiv.org
  • AuthorsTaeil Kim, Kangsan Kim, Sung Ju Hwang
  • Published date2026/08/07

More Items

Alternates a Planner (issues sub-queries) and a Synthesizer (integrates retrieved evidence into a persistent summary) to tackle long-horizon deep-search; introduces Role‑Decoupled Policy Optimization (RDPO) for role-specific RL credit assignment and shows strong results (IterSynth-8B reaches 50.7% on five benchmarks).

A mobile-planning agent paper that develops a closed-loop AI-for-AI lifecycle to generate data, train a planner model, and co-evolve a runtime Harness for multi-app task execution. Demonstrates top performance on MobilePA-Bench with improved tool use, memory, skill coordination and low estimated per-task output cost.

Edits an LLM agent's internal task state and action continuations to remove unsupported assumptions and noisy plans, improving downstream decisions instead of reconstructing high-entropy tool outputs. Combines Action Judge, State Revision, and EditAct and shows consistent gains across benchmarks and agent backbones.