AMA-Bench: Evaluating Long-Horizon Memory for Agentic Applications

TL;DR

This paper introduces AMA-Bench, a new evaluation benchmark for long-horizon memory in LLM-based agents, highlighting current limitations and proposing a causality-aware memory system that significantly improves performance.

Contribution

The paper presents AMA-Bench for evaluating long-term memory in agentic applications and introduces AMA-Agent, a causality-based memory system that outperforms existing methods.

Findings

Existing memory systems lack causality and objective info.

AMA-Agent achieves 57.22% accuracy on AMA-Bench.

AMA-Agent surpasses baselines by 11.16%.

Abstract

Large Language Models (LLMs) are deployed as autonomous agents in increasingly complex applications, where enabling long-horizon memory is critical for achieving strong performance. However, a significant gap exists between practical applications and current evaluation standards for agent memory: existing benchmarks primarily focus on dialogue-centric, human-agent interactions. In reality, agent memory consists of a continuous stream of agent-environment interactions that are primarily composed of machine-generated representations. To bridge this gap, we introduce AMA-Bench (Agent Memory with Any length), which evaluates long-horizon memory for LLMs in real agentic applications. It features two key components: (1) a set of real-world agentic trajectories across representative agentic applications, paired with expert-curated QA, and (2) a set of synthetic agentic trajectories that scale to arbitrary horizons, paired with rule-based QA. Our comprehensive study shows that existing memory systems underperform on AMA-Bench primarily because they lack causality and objective information and are constrained by the lossy nature of similarity-based retrieval employed by many memory systems. To address these limitations, we propose AMA-Agent, an effective memory system featuring a causality graph and tool-augmented retrieval. Our results demonstrate that AMA-Agent achieves 57.22% average accuracy on AMA-Bench, surpassing the strongest memory system baselines by 11.16%.