Can Agent Memory Systems Track Evolving State?
Organizations: University of Illinois Urbana-Champaign
Abstract
As LLM-based agents are deployed for longer and higher-stakes tasks, their memory systems continue to have crucial gaps. While existing memory benchmarks focus largely on recall-shaped tasks, we argue an effective memory system must track the evolving state of the world; as facts, constraints, and decisions are revised over a long interaction, answers must reflect the current state and not a superseded one. We define this capability as state tracking and instantiate it in StateMemBench, a benchmark of 234 multi-session scenarios spanning two conversation-length regimes. Its closed-pool grading scores whether an answer reflects the current state, the superseded state, or fails otherwise, separating state-tracking failures from other errors by construction. Our analysis shows that this task is challenging for existing memory systems, retrieval-augmented baselines, and long-context baselines. We then present StateMem, a state-first memory method that explicitly tracks supersession and relational dependencies, and show it improves current-state accuracy over the strongest same-backbone baseline by 1.8x (0.205 -> 0.363) on DeepSeek-V4-Flash and over the strongest memory system by 1.6x (0.149 -> 0.233) on Qwen-3.5-9B, while remaining competitive with the long-context baselines. Finally, we show the same state approach can be applied as a lightweight single-call wrapper over existing memory systems, lifting current-state accuracy by +32 to +67 points on StateMemBench across six memory and retrieval backends. A length- and cost-matched control attributes +15 to +32 of those points to state structure rather than added context.
Figures & tables
| Benchmark | drift | retr. | comp. | schema | reason | FA | |
|---|---|---|---|---|---|---|---|
| MemoryArena | |||||||
| Shop. | 200 | 63.5 | 22.5 | 2.5 | 0.0 | 9.5 | – |
| Search | 200 | 38.5 | 25.5 | 11.0 | 17.5 | 4.5 | – |
| Travel | 200 | 19.0 | 11.0 | 46.5 | 21.0 | 0.0 | – |
| LongMemEval oracle | |||||||
| LME † | 36 | 44.4 | 0.0 | 0.0 | 25.0 | 2.8 | 27.8 |
| Benchmark | Multi-sess. dialogue | Scale (native unit) | Updates central | Super- session | Verifiable gold | Drift scored | Anti-update controls | Paired horizons |
|---|---|---|---|---|---|---|---|---|
| LoCoMo ( 2024 ) | ✓ | 300 turns | ✗ | ✗ | ✗ | ✗ | ✗ | ✗ |
| LongMemEval ( 2025 ) | ✓ | 40–80 sess. ( ) | (✓) | (✓) | ✗ | ✗ | ✗ | ✗ |
| MemoryAgentBench ( 2025 ) | (✓) | varies | (✓) | (✓) | (✓) | ✗ | ✗ | ✗ |
| MemoryArena ( 2026 ) | (✓) | varies | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
| STATE-Bench ( 2026 ) | ✗ | 450 tasks | ✗ | ✗ | (✓) | ✗ | (✓) | ✗ |
| StateMemBench (ours) | ✓ | 600 turns | ✓ | ✓ | ✓ | ✓ | ✓ | ✓ |
| Condition | Overall | Short | Long | |
|---|---|---|---|---|
| Long-context | Qwen-3.5-9B | 0.149 | 0.137 | 0.167 |
| Qwen-3.6-35B-A3B | 0.097 | 0.063 | 0.146 | |
| GPT-5.4-Nano | 0.277 | 0.311 | 0.227 | |
| DeepSeek-V4-Flash | 0.149 | 0.132 | 0.174 | |
| Qwen-3.5-9B | No memory | 0.006 | 0.000 | 0.015 |
| BM25 | 0.125 | 0.095 | 0.167 |
| LongMemEval | LoCoMo | |||
| Condition | Qwen-9B | DS-V4 | Qwen-9B | DS-V4 |
| Long context | 0.550 | 0.666 | 0.612 | 0.587 |
| No memory | 0.062 | 0.056 | 0.010 | 0.005 |
| Mem0 | 0.566 | 0.594 | 0.481 | 0.462 |
| A-Mem | 0.520 | 0.510 | 0.266 | 0.279 |
| LightMem | 0.064 | 0.056 | 0.027 | 0.020 |
| Qwen-3.5-9B | DeepSeek-V4-Fl. | |||
|---|---|---|---|---|
| Method | I-P | D-R ( ) | I-P | D-R ( ) |
| full context | 263 | 65.5 | 258 | 64.0 |
| Mem0 | 254 | 62.7 | 268 | 59.9 |
| BM25 | 243 | 60.9 | 244 | 59.9 |
| LightMem | 109 | 32.0 | 52 | 14.9 |
| StateMem (ours) | 290 | 64.3 | 286 | 49.1 |
| Qwen-3.5-9B | deepseek-v4-flash | ||||||
|---|---|---|---|---|---|---|---|
| Backend | Alone | +Ctrl | +SMW | Alone | +Ctrl | +SMW | |
| StateMemBench | Mem0 | 25.0 | 28.3 | 56.7 | 28.3 | 56.7 | 71.7 |
| A-Mem | 23.3 | 35.0 | 56.7 | 33.3 | 51.7 | 75.0 | |
| LightMem | 3.3 | 30.0 | 61.7 | 1.7 | 51.7 | 68.3 | |
| MemOS | 5.0 | 25.0 | 56.7 | 1.7 | 50.0 | 68.3 | |
| BM25 | 20.0 | 31.7 | 56.7 | 21.7 | 48.3 | 70.0 | |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Benchmark | agr (%) | PABAK | AC1 | |||
|---|---|---|---|---|---|---|
| MA-shopping | 179 | 91.6 | 0.17 | 0.83 | 0.91 | 0.00 |
| MA-search | 110 | 72.7 | 0.31 | 0.45 | 0.55 | 0.09 |
| LME oracle | 36 | 77.8 | 0.67 | 0.56 | 0.56 | 0.01 |
| LME oracle (cross) | 36 | 69.4 | 0.37 | 0.39 | — | — |
| Arm | correct | rate |
|---|---|---|
| thinking off | 42/50 | 84.0% |
| thinking on | 38/50 | 76.0% |
| pp |
| Source | DOI |
|---|---|
| Biomedical (bioRxiv) | 10.1101/482679 |
| 10.1101/512657 | |
| Computer Science (arXiv) | 10.48550/arXiv.2407.13248 |
| 10.48550/arXiv.2604.10985 | |
| 10.48550/arXiv.2604.11543 | |
| 10.48550/arXiv.2604.21253 |
| By horizon | By category | By domain | ||||||||||
| Condition | Short | Long | Overall | Status | Sequence | Salience | Anti-trap | Compound | Finance | Shopping | Research | |
| Long-context | Qwen-3.5-9B | 0.137 | 0.167 | 0.149 | 0.078 | 0.000 | 0.098 | 0.692 | 0.159 | 0.118 | 0.197 | 0.118 |
| Qwen-3.6-35B-A3B | 0.063 | 0.146 | 0.097 | 0.026 | 0.000 | 0.059 | 0.632 | 0.023 | 0.110 | 0.165 | 0.000 | |
| GPT-5.4-Nano | 0.311 | 0.227 | 0.277 | 0.172 | 0.153 | 0.569 | 0.615 | 0.114 | 0.344 | 0.402 | 0.059 | |
| DeepSeek-V4-Flash | 0.132 | 0.174 | 0.149 | 0.103 | 0.056 | 0.059 | 0.641 | 0.091 | 0.161 | 0.244 | 0.020 | |
| Qwen-3.5-9B | No memory | 0.000 | 0.015 | 0.006 | 0.009 | 0.000 | 0.000 | 0.026 | 0.000 | 0.000 | 0.000 | 0.020 |
| LongMemEval question type | ||||||||
|---|---|---|---|---|---|---|---|---|
| Condition | Overall | SS-user (70) | SS-asst. (56) | SS-pref. (30) | Multi-sess. (133) | Know.-upd. (78) | Temporal (133) | |
| Qwen-3.5-9B | Long context | 0.550 | 0.900 | 0.946 | 0.333 | 0.436 | 0.744 | 0.248 |
| No memory | 0.062 | 0.086 | 0.054 | 0.067 | 0.053 | 0.038 | 0.075 | |
| Mem0 | 0.566 | 0.871 | 0.607 | 0.533 | 0.602 | 0.603 | 0.338 | |
| A-Mem | 0.520 | 0.886 | 0.929 | 0.467 | 0.361 | 0.679 | 0.233 | |
| LightMem | 0.064 | 0.086 | 0.073 | 0.000 | 0.053 | 0.038 | 0.092 | |
| By category | By domain | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Backend | Cond. | Status | Seq. | Sal. | Anti | Comp. | Fin. | Shop. | Res. | |
| Qwen-3.5-9B | Mem0 | alone | 0 | 8 | 0 | 92 | 10 | 27 | 17 | 20 |
| +Ctrl | 17 | 17 | 23 | 92 | 0 | 23 | 33 | 40 | ||
| +SMW | 50 | 58 | 77 | 92 | 40 | 64 | 56 | 70 | ||
| A-Mem | alone | 0 | 0 | 0 | 100 | 10 | 18 | 28 | 20 | |
| +Ctrl | 8 | 25 | 23 | 100 | 10 | 27 | 28 | 50 | ||
| Qwen-3.5-9B | deepseek-v4-flash | ||||||||
| +Ctrl | +SMW | +Ctrl | +SMW | ||||||
| Backend | Fix | Brk | Fix | Brk | Fix | Brk | Fix | Brk | |
| Mem0 | 10 | 11 | 18 | 12 | 15 | 7 | 15 | 6 | |
| A-Mem | 7 | 13 | 17 | 8 | 18 | 8 | 16 | 4 | |
| LightMem | 4 | 0 | 14 | 0 | 17 | 0 | 20 | 1 | |
| MemOS | 4 | 3 | 10 | 0 | 17 | 2 | 20 | 2 | |
| Method | Correct | Drift | Other-opt | Off-pool | Drift-rate |
|---|---|---|---|---|---|
| Long-context (V4-Flash) | 48 | 206 | 4 | 64 | 64.0% |
| Long-context (GPT5.4n) | 89 | 159 | 1 | 73 | 49.4% |
| StateMem (ours) | 117 | 158 | 11 | 36 | 49.1% |
| extraction-only | 56 | 226 | 6 | 34 | 70.2% |
| + supersession | 96 | 184 | 9 | 33 | 57.1% |
| dependency-propagation | 120 | 157 | 9 | 36 | 48.8% |
| Method | Correct | Drift | Other-opt | Off-pool | Drift-rate |
|---|---|---|---|---|---|
| Long-context-9B | 48 | 211 | 4 | 59 | 65.5% |
| Long-context-35B | 31 | 215 | 6 | 68 | 67.2% |
| StateMem (ours) | 75 | 207 | 8 | 32 | 64.3% |
| extraction-only | 50 | 239 | 9 | 24 | 74.2% |
| + supersession | 55 | 230 | 10 | 27 | 71.4% |
| dependency-propagation | 71 | 212 | 13 | 24 | 66.2% |
Agent Memory
Agent memory lets AI systems retain, update, and use facts, experiences, and skills across sessions and long-horizon tasks.
See all 191 papers