GRAM: Empowering Agent with Actively Managed Graph-Structured Memory via Reinforcement Learning

Xiangqi Chen, Zihao Yi, Haohao Luo, Ying Shen


Abstract
Large Language Model (LLM) agents face a fundamental bottleneck called context flooding in multi-turn information retrieval: as evidence accumulates across interaction turns, their context windows become saturated with redundant, conflicting, and outdated facts. Existing research either compresses interaction history into rolling summaries, sacrificing relational precision, or relies on retrieval-augmented pipelines that lack mechanisms for resolving knowledge conflicts and maintaining a globally consistent belief state. In this work, we introduce GRAM, a reinforcement learning framework designed to train an agent to actively manage and utilize a dynamic graph-structured memory. Rather than passively accumulating retrieved text, the agent learns to incrementally construct and revise a coherent knowledge graph that serves as its persistent belief state, resolving contradictions and preserving logical relationships in memory. Furthermore, We employ Group Relative Policy Optimization to train small language models (under 4B) to master these complex memory-governance behaviors. Experimental results across multiple mainstream question-answering benchmarks demonstrate the superiority of the GRAM framework over existing agentic memory baselines and conventional RAG systems.
Anthology ID:
2026.findings-emnlp.1183
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
25004–25026
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1183/
DOI:
Bibkey:
Cite (ACL):
Xiangqi Chen, Zihao Yi, Haohao Luo, and Ying Shen. 2026. GRAM: Empowering Agent with Actively Managed Graph-Structured Memory via Reinforcement Learning. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 25004–25026, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
GRAM: Empowering Agent with Actively Managed Graph-Structured Memory via Reinforcement Learning (Chen et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1183.pdf