PerMIT: Securing Persistent Memory for LLM Agents via Machine-Checked Integrity Theorems and Scalable Community Trust

Tanqiu Jiang (Stony Brook, intern) with Min Bai, Sadat Shahriar, Wanyu Du and Sandesh Swamy (AWS AI Labs) present PerMIT, an agent memory architecture with two integrity theorems machine-checked in Lean 4 that confine every memory-poisoning attack in its threat model to one auditable confirmation step.
Ask this paper
Trust assumption. Only user-typed inputs and values confirmed at a gate are trusted. Environment-derived values pass through a finite memory interface that still allows agent-driven multi-turn planning and tool choice.
Proved guarantees. The two theorems state that environment-derived memory reaches a sensitive action only if confirmed at the gate, and influences behavior only through what the gate displays. Each attack therefore reduces to a single decision a human or LLM verifier can make.
Attack results. On Trojan Hippo, PerMIT records zero observed attack success with self-verification and lightweight verifiers.
Community memory. Logged confirmations accumulate across users, so well-endorsed items become community-confirmed. On tau2-bench, communities of up to 90 agents build shared knowledge bases with no injected item promoted, and the shared knowledge significantly improves new agents on held-out tasks.
Abstract
Persistent memory is a largely undefended surface for injection and poisoning, and the same risk limits memory sharing among agents. We present PerMIT, a memory architecture with provable integrity guarantees under an explicit trust assumption: only user-typed inputs and values confirmed at a gate are trusted. PerMIT mediates the release and use of untrusted environment-derived values through a finite memory interface, while preserving agent-driven multi-turn planning and tool choice. Over this finite interface, we prove two integrity theorems that are also machine-checked in Lean 4, which together ensure that environment-derived memory reaches a sensitive action only if confirmed at the gate and influences the agent's behavior only through what the gate displays. Every attack in our threat model thereby reduces to a single auditable decision, which a human, an LLM verifier, or both can make in practice. On Trojan Hippo, PerMIT records zero observed attack success with self-verification and lightweight verifiers. Because each confirmation is a logged verdict on exact content, endorsements accumulate across users, and sufficiently endorsed items become community-confirmed with no further verification required. Under explicit assumptions, we prove and corroborate in simulation a co-scaling property: as the community grows, the risks of promoting poison and of wrongly evicting benign knowledge fall, while per-user verification effort does not rise. On $τ^2$-bench, communities of up to 90 autonomous agents use PerMIT to build shared knowledge bases with no direct memory injection promoted, and the resulting knowledge significantly improves new agents on held-out tasks, suggesting a path toward agents that safely self-improve at test time.