Memory Reward Inflation in Self-Improving LLM Agents
Source
Mohammad Asadolahi, Amir Amini, Samira Talebi, Amirfarhad Farhadi, Azadeh Zamanifar
Author
Mohammad Asadolahi, Amir Amini, Samira Talebi, Amirfarhad Farhadi, Azadeh Zamanifar
Published
Why it matters
Self-graded episodic memory in agents can inflate confidence in bad past actions, causing agents to repeat their own mistakes; LUCID's answer-free de-inflation approach lifts BIRD execution accuracy from 52.4% (no memory) and 54.0% (self-graded memory) to 56.9%, giving a concrete fix for a subtle but consequential failure mode in agent memory design.
This paper identifies the 'Echo Gap,' a failure mode in memory-based self-improving LLM agents where self-graded episode scores inflate confidence in incorrect past actions, causing agents to preferentially reuse their own mistakes.
It formalizes the Error-Independence Assumption as a necessary condition for fixing this, and introduces LUCID, an answer-free de-inflation algorithm that raises BIRD text-to-SQL execution accuracy to 56.9%, beating both a Memento-style self-graded agent (54.0%) and a memory-less baseline (52.4%).
Transcript
arXiv:2608.00017v1 Announce Type: new Abstract: Self-improving LLM agents increasingly learn from experience without updating any weights. Each episode is stored in an external memory, scored, and retrieved for similar future tasks to shape later behavior. Viewed through a reward lens, the stored score is a proxy reward for an implicit, non-parametric policy. Each retrieved episode then becomes a policy-improvement step whose reliability hinges on how that score is produced. In deployment, ground-truth labels are unavailable, so the stored reward is at best an LLM assessment. This substitution creates a failure mode, the *Echo Gap*, across the memory-based self-improving agents and model families studied. Incorrect episodes receive inflated rewards; thus, the agent preferentially reuses the very mistakes it has most confident in. Because the error compounds through memory rather than averaging out and the confirming judge's errors remain correlated with the original self-grading bias, so it cannot identify which memories are overvalued. The missing property is formalized as the *Error-Independence Assumption* (EIA), which we prove is a *necessary* condition for correcting the inflation, not merely a description of a good verifier: a usable signal must track truth *and* decorrelate its error from the memory bias, and the recoverable payoff is a closed-form function of exactly those two quantities. We further show the inflation compounds not only when retrieval ranks by the stored score but also under plain similarity retrieval which is the regime the deployed agent uses. Finally, the answer-free de-inflation algorithm LUCID delivers a consistent end-to-end gain on the BIRD text-to-SQL benchmark. It raises execution accuracy to $56.9\%$, above both a Memento-style self-graded agent ($54.0\%$, a $+2.9$-point mean gain across seeds) and a memory-less agent of identical architecture ($52.4\%$).