JA EN
LearnAgents
·★ MEMBER·PAPER·8 min read

Paper Walkthrough: Metis — A 'Memory Foundation Model' That Moves Agent Memory Inside the Model

Agent memory today is mostly bolted on from the outside via RAG. This paper proposes memory foundation models — models whose forward pass natively stores, forgets, and updates information — and builds Metis, the first prototype. A from-scratch walkthrough of how it works, how well it works, and where it breaks.

ModalitytextTaskagents

Metis: Memory Foundation Model

Primary source — what this article is built on

undefined2026-07-29undefined2026-08-13same month

Metis: Memory Foundation ModelZeyu Zhang, Ziliang Guo, Yihang Sun et al. · 2026-07-29 · v2arXiv:2607.26760Paper page·PDF
undefined

Recent advances in AI agents have increasingly internalized native capabilities into their underlying foundation models, giving rise to multimodal foundation models and large reasoning models. However, agent memory is still primarily implemented through external modules, leaving the native memory capability largely unexplored. In this paper, we take a first step toward this direction by introducing memory foundation models, which empower foundation models with native memory capabilities. We formalize native memory from two perspectives: a persistent and dynamically evolving memory state within the backbone, and native memory procedures that autonomously store and utilize information through model computation. We show that native memory offers advantages in architecture, end-to-end optimization, and efficiency. Based on this formulation, we propose Metis, the first prototype of memory foundation models. Metis introduces a new architecture that equips a foundation model with a native memory state, allowing historical information to be compressed into the model and accessed through memory attention. We construct large-scale memory-specific training data and introduce multiple optimization objectives to acquire these native memory procedures through mid-training. The online memory maintenance of Metis is gradient-free, and the memory update requires only a forward pass. At inference time, all learned model weights remain frozen, while the native memory states are autonomously transformed through standard forward computation. Through extensive experiments, we show that Metis exhibits native memory capabilities and further provide a detailed analysis of its strengths, limitations, and behaviors. To facilitate future research on memory foundation models, we release our project and model checkpoints.


AI memory is still a sticky-note system

Imagine a brilliant colleague whose memory is wiped after every meeting. That is, roughly, today's LLM. The standard workaround is to keep transcripts of past interactions in an external store and paste the relevant bits back into the prompt — Retrieval-Augmented Generation (RAG). The model doesn't remember anything; it rereads a stack of sticky notes before every reply.

Human memory doesn't work like that. Experience soaks into the wiring itself, and remembering isn't a separate step from thinking. The paper "Metis: Memory Foundation Model" (MemTensor, Renmin University of China, National University of Singapore, and others; published July 2026) is an attempt to give LLMs that kind of soaked-in memory. The authors call the concept a memory foundation model and build the first prototype, Metis.

Three limits of bolt-on memory

The paper opens by listing three problems with external memory (§1).

  1. Mismatched objectives: the memory module's job is "assemble a useful context," while the backbone's job is "continue the text it was given." The two are designed separately, so the memory may not surface what the model actually needs, and the model may not use what it gets optimally.
  2. No end-to-end training: gradients cannot flow through discrete operations like "retrieve" and "concatenate," so you can't tune memory behavior directly from task performance.
  3. Latency: retrieval, reranking, and stitching run on every request, adding inference delay.

Against this, the paper defines native memory by two components (§2):

The definition of "memory" here is strict: anything the model knew before the interaction started counts as knowledge, not memory. Only information acquired online, during the interaction, qualifies (§2.2). The paper's deeper framing is that memory is fundamentally a prediction problem — predicting how information received now will be used later (§1). And if it's a prediction problem, it should be learnable from data, like everything else.

The primitive underneath all the retrieval is one you already know from Transformers: the dot product. Worth refreshing the intuition before we open the machinery.

FIG 1The more two vectors point the same way, the larger their dot product. Metis decides which memories to read out by the dot product between the current query and the keys written into its memory state

Inside Metis: two blocks share the work

Metis adds a Metis block to each Transformer layer, made of two parts (§3.2).

In other words: the rules for how to remember are baked in by training; only the contents of memory move at inference time.

Writing: select, compress, accumulate

When a step of the conversation finishes, Metis writes it to memory in three moves (§3.3).

First, the hidden states of each layer are scored by the importance vector, and a temperature-scaled softmax turns the scores into a distribution over "which tokens deserve to be remembered." Metis then keeps only the top tokens whose cumulative probability reaches a threshold ρ\rho — a compression step that distills a long input down to its essentials.

FIG 2The temperature-scaled softmax used in Metis's write-selection. Lower the temperature τ and the distribution sharpens ("remember only a few tokens"); raise it and it flattens ("remember broadly but shallowly")

The selected tokens are projected into memory keys and values and folded into the memory matrix:

What's behind this

§

Members-only from here

371 walkthroughs, 26 textbook chapters, 48 student units and 6 close readings — all included for $4.99/mo, with three new explainers every day. Cancel any time; access runs to the end of the period.

Already a member? Sign in to keep reading

References

  1. Zeyu Zhang, Ziliang Guo, Yihang Sun, Xichong Zhang et al.. (2026-07-29) Metis: Memory Foundation Model. arXiv:2607.26760Paper page·PDF

This article is written from the source paper above. Where they differ, the original is authoritative.

Comments

Sign in to comment