Saved in:
Bibliographic Details
Main Authors: Huo, Yupeng, Lu, Yaxi, Zhang, Zhong, Chen, Haotian, Lin, Yankai
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2601.08323
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910077414277120
author Huo, Yupeng
Lu, Yaxi
Zhang, Zhong
Chen, Haotian
Lin, Yankai
author_facet Huo, Yupeng
Lu, Yaxi
Zhang, Zhong
Chen, Haotian
Lin, Yankai
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2601_08323
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle AtomMem : Learnable Dynamic Agentic Memory with Atomic Memory Operation
Huo, Yupeng
Lu, Yaxi
Zhang, Zhong
Chen, Haotian
Lin, Yankai
Artificial Intelligence
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.
title AtomMem : Learnable Dynamic Agentic Memory with Atomic Memory Operation
topic Artificial Intelligence
url https://arxiv.org/abs/2601.08323