English

REMem: Reasoning with Episodic Memory in Language Agent

Artificial Intelligence 2026-03-03 v3

Abstract

Humans excel at remembering concrete experiences along spatiotemporal contexts and performing reasoning across those events, i.e., the capacity for episodic memory. In contrast, memory in language agents remains mainly semantic, and current agents are not yet capable of effectively recollecting and reasoning over interaction histories. We identify and formalize the core challenges of episodic recollection and reasoning from this gap, and observe that existing work often overlooks episodicity, lacks explicit event modeling, or overemphasizes simple retrieval rather than complex reasoning. We present REMem, a two-phase framework for constructing and reasoning with episodic memory: 1) Offline indexing, where REMem converts experiences into a hybrid memory graph that flexibly links time-aware gists and facts. 2) Online inference, where REMem employs an agentic retriever with carefully curated tools for iterative retrieval over the memory graph. Comprehensive evaluation across four episodic memory benchmarks shows that REMem substantially outperforms state-of-the-art memory systems such as Mem0 and HippoRAG 2, showing 3.4% and 13.4% absolute improvements on episodic recollection and reasoning tasks, respectively. Moreover, REMem also demonstrates more robust refusal behavior for unanswerable questions.

Keywords

Cite

@article{arxiv.2602.13530,
  title  = {REMem: Reasoning with Episodic Memory in Language Agent},
  author = {Yiheng Shu and Saisri Padmaja Jonnalagedda and Xiang Gao and Bernal Jiménez Gutiérrez and Weijian Qi and Kamalika Das and Huan Sun and Yu Su},
  journal= {arXiv preprint arXiv:2602.13530},
  year   = {2026}
}

Comments

Accepted by The Fourteenth International Conference on Learning Representations (ICLR 2026) as poster

R2 v1 2026-07-01T10:36:25.010Z