Learning to Remember: End-to-End Training of Memory Agents for Long-Context Reasoning

arXiv:2602.18493 · cs.LG, cs.AI · Submitted 2026-02-13 · Read on arXiv

cs.LG, cs.AI

Submitted: 2026-02-13

Updated: 2026-09-01

Comments: Accepted at EMNLP 2026 Main Conference. Camera-ready version

Code: https://github.com/ictnlp/unified-memory-agent

License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/

Terminology

Sources

Related papers