Skip to content

Column

The Weakness of AI Agents Is Forgetting — AutoMem Turns Memory Management into a Learnable Skill

What to remember, when to recall, how to organize — AutoMem implements this 'metamemory' as a learnable skill for LLM agents. Memory optimization alone boosted performance 2–4x and lifted a 32B open model to compete with top-tier commercial models. Essential reading for enterprises deploying agents on long-horizon work.

5 min read 日本語版 →
Flat illustration of a small AI robot retrieving notes from a large, well-organized memory shelf with drawers while progressing along a long winding task road

Hello. I am Miura from Affectosphere Group.

Try assigning an AI agent a multi-week job, and you will hit the same wall every time: it forgets.

Context windows have limits. Last week’s decisions and lessons from past failures vanish across sessions. RAG and note-keeping help, but the judgment about what to keep is still designed by humans.

A paper published on arXiv in July 2026 (Wu et al., arXiv:2607.01224) proposes making that judgment itself — what to remember, when to recall, how to organize — a skill the agent learns.


Three takeaways

  1. AutoMem implements “metamemory” — a concept from cognitive science covering what to store, when to retrieve, and how to organize — as a learnable skill for LLM agents.
  2. The mechanism is a dual loop: a first loop where a strong LLM reviews full trajectories and automatically revises the memory structure, and a second loop that extracts the agent’s own good memory decisions as training signals.
  3. Across three long-horizon game environments (Crafter, MiniHack, NetHack), optimizing memory management alone improved base-agent performance roughly 2–4x, lifting a 32B open model to a level competitive with Claude Opus 4.5 and Gemini 3.1 Pro Thinking.

① The conceptual flip: memory as a first-class action

The most interesting design choice in AutoMem is how radically it reframes memory.

For conventional agents, writing and rereading notes is a side activity, subordinate to the “real” task. AutoMem inverts this. Writing to, organizing, and consulting a file-system memory are promoted to “memory actions” — fully equal in status to task actions.

At every step, the agent chooses on the same footing: make the next move, or tend to memory. It mirrors how skilled humans behave — the busier things get, the more the expert pauses to organize their notes.

On top of that, the way of remembering is itself trained. That is where the dual loop comes in.


② The dual loop: an external reviewer and an internal distiller

In the first loop, a strong LLM acts as a reviewer. It examines the agent’s complete trajectories and revises the memory structure automatically: “this information should have been kept,” “this organization hurts retrieval.” Think of it as review by an external coach.

The second loop is self-learning. From the agent’s own behavior, memory operations that turned out well — say, writing down at the right moment a fact that later proved decisive — are extracted as training signals and folded back into the policy.

External review plus internal distillation of successes. Running both loops trains the very judgment of what deserves remembering.


③ Memory alone: 2–4x performance, 32B competing with the top tier

Validation used three long-horizon game environments: Crafter, MiniHack, and NetHack. All are standard stress tests for long-term memory, requiring situational awareness across hundreds to thousands of steps.

The results are striking. Without touching the base model, optimizing memory management alone improved agent performance roughly 2–4x.

The cost implication matters even more. A 32B open model equipped with AutoMem reached a level competitive with top commercial models such as Claude Opus 4.5 and Gemini 3.1 Pro Thinking. Instead of competing on raw intelligence, it caught up on memory craft.


A practical proposal: design principles for long-horizon enterprise agents

The target users are enterprises that want AI agents inside multi-week business flows: legal due diligence, large-deal project management, continuous supply chain monitoring, long-term customer follow-up. In all of these, remembering last month’s context is the core of the value.

First, make memory management an independent axis in agent evaluation. Agent selection today fixates on model intelligence and tool integrations, but for long-horizon work, the quality of the memory architecture can swing outcomes at the 2–4x level — that is this paper’s message. Add “can it carry context across a week?” to your PoC test items.

Second, use it to rethink cost structure. Long-horizon work does not automatically require a top-tier model. A memory-optimized mid-size model is now a realistic option — worth having your ML team evaluate as a configuration that cuts cloud spend while preserving business continuity.

Third, measure the KPI as handover accuracy: the share of cross-week tasks where past decisions and constraints were correctly referenced. Track it over time and memory improvements become visible in numbers.


From smarter AI to AI that does not forget

Progress in agents is usually framed as better reasoning and better tool use.

But the longer an agent stays in real operations, the more the unglamorous skill of remembering well starts to dominate. Human organizations are no different — recording and handing over properly matters about as much as brilliance.

Treat memory as a skill to train, not a talent to buy. That is the perspective worth taking home from this paper for any team planning agent deployments.

Cheers!


References

  1. Wu, Shengguang, Zhu, Hao, Zhang, Yuhui, Wang, Xiaohan, & Yeung-Levy, Serena (2026). AutoMem: Automated Learning of Memory as a Cognitive Skill. arXiv preprint.

* This article was written in part with AI assistance and may contain inaccuracies.