AtomMem : Learnable Dynamic Agentic Memory with Atomic Memory Operation

View PDF HTML (experimental)

Abstract:Equipping agents with memory is essential for solving real-world long-horizon problems. However, most existing agent memory mechanisms rely on static and hand-crafted workflows. This limits the performance and generalization ability of these memory designs, which highlights the need for a more flexible, learning-based memory framework. In this paper, we propose AtomMem, which reframes memory management as a dynamic decision-making problem. We deconstruct high-level memory processes into fundamental atomic CRUD (Create, Read, Update, Delete) operations, transforming the memory workflow into a learnable decision process. By combining supervised fine-tuning with reinforcement learning, AtomMem learns an autonomous, task-aligned policy to orchestrate memory behaviors tailored to specific task demands. Experimental results across 3 long-context benchmarks demonstrate that the trained AtomMem-8B consistently outperforms prior static-workflow memory methods. Further analysis of training dynamics shows that our learning-based formulation enables the agent to discover structured, task-aligned memory management strategies, highlighting a key advantage over predefined routines.

Subjects:

Artificial Intelligence (cs.AI)

Cite as: arXiv:2601.08323 [cs.AI]

(or arXiv:2601.08323v3 [cs.AI] for this version)

https://doi.org/10.48550/arXiv.2601.08323

arXiv-issued DOI via DataCite

Submission history

From: Yupeng Huo [view email] [v1] Tue, 13 Jan 2026 08:22:28 UTC (834 KB) [v2] Tue, 27 Jan 2026 13:47:50 UTC (834 KB) [v3] Fri, 27 Mar 2026 03:26:42 UTC (575 KB)