WMF-AM: Probing LLM Working Memory via Depth-Parameterized Cumulative State Tracking
Published in arXiv preprint, 2026
We introduce Working Memory Fidelity-Active Manipulation (WMF-AM), a probe of cumulative state tracking that isolates within-pass cumulative load by parameterizing depth K. Testing 20 open-weight models (0.5B–35B) across 13 families, our probe predicts agent performance with r = 0.612 (p < 0.001).
Recommended citation: Hou, D., Jiang, L., Li, D., Li, Z., Lin, F., & Yamada, K. D. (2026). WMF-AM: Probing LLM Working Memory via Depth-Parameterized Cumulative State Tracking. arXiv preprint arXiv:2603.27343.
Download Paper
