artificial intelligence

Experience Memory Graph: One-Shot Error Correction for Agents

arXiv:2607.13884

summary

The paper introduces Experience Memory Graph (EMG), a method that uses graph matching between failed and successful trajectories to provide one-shot error correction for large language model agents without test-time trial-and-error.

Abstract

Large Language Model (LLM) agents have shown remarkable capabilities in autonomous decision-making by generating sequential trajectories of states, actions, and observations. However, in complex, long-horizon tasks, these agents frequently suffer from compounding errors and struggle to recover from failures. Existing self-correction mechanisms rely on prompt-based reflection, which is inherently brittle, incurs heavy time and API costs due to iterative trial-and-error loops, and produces task-specific memory that may be hard to generalize to new scenarios. To address this, we propose Experience Memory Graph (EMG), a framework that reformulates agent failure recovery as a graph matching problem. At training time, we convert both failed exploration trajectories and successful expert trajectories into directed action decision graphs. By matching these graphs, we extract common subgraphs (successful workflows) and graph edit paths that explicitly indicate how to correct failures (e.g., which actions to add, delete, or relabel under a given observation), and store them in a memory graph with intra-task nodes and cross-task edges. At test time, EMG retrieves relevant insights and guides the agent in a single, loop-free execution. Experiments on ALFWorld and ScienceWorld show that EMG consistently outperforms state-of-the-art reflection baselines in success rate and average reward, while requiring no test-time trial-and-error.

11 pages, 6 figures

Topics & keywords

#large language model agents#error correction#graph matching#task planning#reinforcement learningexperience memory graphgraph edit pathALFWorldScienceWorldsingle-shot correctionprompt-based reflection
Experience Memory Graph: One-Shot Error Correction for Agents · wovepaper