From Unstructured Recall to Schema-Grounded Memory: Reliable AI Memory via Iterative, Schema-Aware Extraction
Alex Petrov, Alexander Gusak, Denis Mukha, and Dima Korolev

TL;DR
This paper introduces a schema-grounded memory system for AI that improves reliability and accuracy in memory tasks by focusing on structured, verified records rather than unstructured retrieval.
Contribution
It proposes an iterative, schema-aware write process with validation and retries, shifting from inference-based reads to verification-based queries for more reliable AI memory.
Findings
Achieved 90.42% object-level accuracy in structured extraction.
Reached 97.10% F1 in end-to-end memory benchmarks.
Outperformed existing memory systems and models on application-level tasks.
Abstract
Persistent AI memory is often reduced to a retrieval problem: store prior interactions as text, embed them, and ask the model to recover relevant context later. This design is useful for thematic recall, but it is mismatched to the kinds of memory that agents need in production: exact facts, current state, updates and deletions, aggregation, relations, negative queries, and explicit unknowns. These operations require memory to behave less like search and more like a system of record. This paper argues that reliable external AI memory must be schema-grounded. Schemas define what must be remembered, what may be ignored, and which values must never be inferred. We present an iterative, schema-aware write path that decomposes memory ingestion into object detection, field detection, and field-value extraction, with validation gates, local retries, and stateful prompt control. The result…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
