WMF-AM: Probing LLM Working Memory via Depth-Parameterized Cumulative State Tracking

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hou, Dengzhe, Jiang, Lingyu, Li, Deng, Li, Zirui, Lin, Fangzhou, Yamada, Kazunori D
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914527030804480
author Hou, Dengzhe
Jiang, Lingyu
Li, Deng
Li, Zirui
Lin, Fangzhou
Yamada, Kazunori D
author_facet Hou, Dengzhe
Jiang, Lingyu
Li, Deng
Li, Zirui
Lin, Fangzhou
Yamada, Kazunori D
contents Existing large language models (LLMs) evaluations use fixed-difficulty benchmarks that cannot adapt as models improve, and rarely isolate specific cognitive processes. We introduce Working Memory Fidelity-Active Manipulation (WMF-AM), a probe of cumulative state tracking, the ability to maintain and update intermediate results across K sequential operations within a single query, without a scratchpad. Unlike multi-step agent benchmarks that stress task orchestration, WMF-AM isolates within-pass cumulative load by parameterizing depth K. The core probe uses arithmetic accumulation on 28 models from 12 families (0.5B to frontier); a matched non-arithmetic extension (permissions, schedules, inventories) confirms the design generalizes beyond arithmetic. Three construct-isolation ablations confirm that cumulative load, not arithmetic skill or entity tracking, drives difficulty. We release WMF-AM as a lightweight, recalibratable diagnostic for characterizing where models degrade under cumulative load. Code and data can be accessed at https://github.com/dengzhe-hou/WMF-AM
format Preprint
id arxiv_https___arxiv_org_abs_2603_27343
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle WMF-AM: Probing LLM Working Memory via Depth-Parameterized Cumulative State Tracking
Hou, Dengzhe
Jiang, Lingyu
Li, Deng
Li, Zirui
Lin, Fangzhou
Yamada, Kazunori D
Artificial Intelligence
Existing large language models (LLMs) evaluations use fixed-difficulty benchmarks that cannot adapt as models improve, and rarely isolate specific cognitive processes. We introduce Working Memory Fidelity-Active Manipulation (WMF-AM), a probe of cumulative state tracking, the ability to maintain and update intermediate results across K sequential operations within a single query, without a scratchpad. Unlike multi-step agent benchmarks that stress task orchestration, WMF-AM isolates within-pass cumulative load by parameterizing depth K. The core probe uses arithmetic accumulation on 28 models from 12 families (0.5B to frontier); a matched non-arithmetic extension (permissions, schedules, inventories) confirms the design generalizes beyond arithmetic. Three construct-isolation ablations confirm that cumulative load, not arithmetic skill or entity tracking, drives difficulty. We release WMF-AM as a lightweight, recalibratable diagnostic for characterizing where models degrade under cumulative load. Code and data can be accessed at https://github.com/dengzhe-hou/WMF-AM
title WMF-AM: Probing LLM Working Memory via Depth-Parameterized Cumulative State Tracking
topic Artificial Intelligence
url https://arxiv.org/abs/2603.27343