REVIEW 3 major objections 6 minor 37 references
High reward does not prove an agent learned the task's state
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-02 07:13 UTC pith:MPMKXRAK
load-bearing objection A genuinely useful measurement paper: it separates reward success from latent-state learning with an exact oracle and a held-out structural predictor, and it deserves a serious referee despite a load-bearing linear-probe assumption. the 3 major comments →
When Does Reward Teach State? A Hidden-Automaton Instrument and the Group-Language Boundary
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The paper's central claim is that reward success and latent-state learning are distinct, separately measurable quantities, and that their coupling is governed by three axes: optimizer strength, task structure (above all permutation/group-language structure), and observation informativeness. Using a hidden deterministic finite automaton (DFA) testbed where the agent observes a symbol stream, occasionally chooses the next symbol, and gets a single terminal reward for acceptance, the experimenter can compute the optimal return and the exact latent state at every step — yielding an oracle-normalized reward score and a direct linear probe of the agent's representation. The paper shows that under
What carries the argument
The hidden-automaton instrument: express the task as a hidden deterministic finite automaton (DFA) with partial control and sparse terminal reward, so the exact optimal return and the latent state at every timestep are computable by the experimenter. This yields an oracle-normalized reward score and a linear probe of the policies' pre-head features against the true state, with an off-policy supervised capacity check as a control. The structural predictor is the permutation (group-language) property — every symbol induces a bijection on the reachable states, so no input ever merges states — which a credit-assignment analysis (bounded by 'state-influence' and geometric decay at a state-blind p
Load-bearing premise
The perception gap is defined as 'latent state not linearly recoverable, though representable' via a linear probe on pre-head features plus a saturated off-policy supervised capacity check; if the agent stored the state in non-linear features that the linear probe misses, the perception-vs-planning distinction could be an artifact of probe class rather than a property of the agent.
What would settle it
Run the full instrument on a single permutation automaton with saturated supervised capacity where a sufficiently strong and budgeted PPO+GAE agent achieves near-optimal oracle-normalized reward, and check whether a non-linear probe (e.g., a small MLP on the same features) recovers the latent state with high accuracy while the paper's linear probe stays near chance. If so, the 'perception gap' is a probe-class artifact. Alternatively, find one non-permutation automaton that the paper's structural predictor flags as hard but where the agent learns the state perfectly with saturated capacity, wh
If this is right
- High reward is not evidence that the latent state was learned; state-aware evaluation must include an internal probe, not just behavioral reward.
- A perception gap and a planning gap demand opposite fixes: make the latent state observable or latchable to close the first, and change the policy or reward to exploit the second.
- Any finite-state workflow can be audited white-box by recovering its automaton and applying the same oracle and probe; the paper demonstrates this on real checksum logic and process logs.
- The permutation/group-language property is a cheap pre-training red flag for state-blindness, but its one-directionality means non-group automata still need individual auditing.
- Observation informativeness has a sharp threshold: a label-free auxiliary recovers state as soon as observations carry any state signal (β≈0.1), suggesting small observation leaks can bootstrap representation learning.
Where Pith is reading between the lines
- If the linear-probe operationalization is right, the perception-gap concept could be ported to other settings where a known hidden variable exists, such as model-based planning or world-model evaluation in supervised sequence models, giving a measurement of 'state blindness' beyond RL.
- The sharp β≈0.1 threshold hints that even negligible state leakage in observations can ignite representation learning, which is a testable design principle for environments where state is otherwise hidden.
- The failure of the converse — non-permutation automata are heterogeneous — suggests there is no single structural predictor for all perception gaps; a broader theory would need to account for reward-shortcut stability beyond group structure.
- The LLM-agent port suggests the same credit-starvation mechanism may limit large language models on accumulation-heavy tasks; the authors' open corner is a stronger optimizer, which is a concrete next experiment.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces a white-box instrument for measuring whether an RL agent that earns reward has also learned the task's latent state. Tasks are hidden deterministic finite automata (DFAs) with partial control and sparse terminal reward; since the automaton is known, the experimenter can compute an exact oracle for normalized return and can probe the exact latent state on every step without showing it to the agent. The central claim is that reward success and latent-state learning are distinct, separately measurable quantities, and that their coupling is governed by three axes: optimizer strength, task structure (in particular permutation/group-language structure), and observation informativeness. The paper reports a pre-registered A2C-vs-PPO control, a held-out validation of the structural predictor on 153 fresh automata (precision 0.86 for permutation structure flagging perception gaps), and a distinction between perception gaps (latent state not linearly recoverable) and planning gaps (state recoverable but not exploited), concluding that reward alone cannot localize the failure.
Significance. If valid, this is a significant contribution: it makes goal-misgeneralization and reward-shortcut worries measurable rather than purely behavioral, provides an exact oracle and exact probe target for any FSM-approximable task, and identifies a pre-training-computable warning signal (permutation/group-language structure) with a mechanistic credit-assignment explanation. The paper's strengths are substantial: the §5.2 pre-registered 2×2 de-confound, the episode-split probe protocol with selectivity controls, the saturated supervised-capacity checks, the 153-automaton held-out evaluation with Wilson CI and threshold robustness, and the explicit demotion of the non-transferring leverage threshold. The manuscript is unusually transparent about its own corrections and limitations.
major comments (3)
- [§4, §5.5, Fig. 2] The perception gap is operationalized as 'state not linearly recoverable, though representable': a task is labeled a perception gap when an on-policy linear probe recovers <0.5 balanced accuracy while an off-policy supervised capacity check saturates. The capacity check only shows that the architecture can be trained, under supervision, to linearly decode state; it does not rule out a nonlinearly encoded but present state in the RL-trained policy. A nonlinear probe with matched capacity could reclassify some of the 103 permutation automata, and the headline held-out precision 0.86 is computed against exactly this linear-probe label. Fig. 2's caption goes further and calls perception gaps 'state absent,' which is stronger than 'not linearly recoverable.' I recommend running nonlinear probes (e.g., a matched MLP probe) with the same episode split and selectivity control, and re-scoring the
- [§4, §5.4, §6] The planning-gap construct is defined as 'state recoverable but not exploited,' yet the interventional probe described in §4 as distinguishing a representation the policy consults from one merely present is reported only for 'the seven policies where an auxiliary installed the state' (§6). It is not applied to the planning-gap cases such as ends_with_1000 (§5.4), where a high probe gap accompanies low reward. Without the interventional check, a high linear-probe reading plus low reward could also be consistent with a state representation that is partially used but insufficient for control, or with a probe recovering state from features that do not influence action selection. The perception/planning distinction would be on firmer ground if the interventional probe were run on naturally arising planning-gap policies.
- [§5.5, Appendix D, Table 3] The held-out validation uses a stratified sample designed to populate both predictor classes, and the text notes that the predicted-hard class is only about 1% of uniform random diagnostic DFAs. Consequently the precision 0.86 is a conditional positive-predictive value within the constructed sample, not an estimate for a natural deployment distribution. The paper is transparent about the sampling, but the practical claim that permutation structure is a 'high-precision warning signal' would benefit from an explicit statement of how precision varies with the natural prevalence of permutation automata and with the within-class ratio of low- versus high-leverage permutation automata; otherwise a reader may over-generalize the 0.86 figure.
minor comments (6)
- [Fig. 2 caption] "Perception gap (state absent)" conflicts with the formal definition in §4/§5.5 as "not linearly recoverable." Please align the caption wording with the operational definition.
- [Table 2 and §5.4 text] The text says the recovered fraction shrinks "from ≈1 at k=2,3," but Table 2 reports PPO/ceiling = 0.83 at k=2 and 0.99 at k=3. Please reconcile the textual summary with the table.
- [Appendix A] The pre-registration is described as fixed before running, but no timestamp, versioned document, or independent registration identifier is given. Adding a date and/or hash would strengthen the claim.
- [§5.4 (Spearman correlations)] The correlations between behavioral signals and probe gap (Spearman −0.29, −0.25) are reported without confidence intervals or exact sample size. Given the small curated suite, bootstrap CIs would help calibrate the "nearly uncorrelated" claim.
- [§5.5, Appendix D] The 120-automaton extension uses 3 seeds per automaton. Please report seed-level label stability (e.g., how many automaton labels would change if labels were computed from individual seeds versus the mean), since the Wilson CI treats automata as independent units.
- [General wording] The phrase "pre-training structural warning signal" may be misread as referring to pretrained models; consider "pre-RL" or "pre-training-time" structural warning signal.
Circularity Check
No significant circularity: the oracle, probe labels, and structural predictor are independent measurements.
full rationale
The paper's derivation chain is self-contained and its central predictive claim is not circular. The hidden DFA and exact oracle are defined before any RL and are used only as measurement instruments; the latent state is known exactly from the automaton, so the probe labels are read from the agent's activations, not from the predictor. The structural predictor (permutation/group-language structure) is computed from the transition function δ before any training and is validated on 153 fresh automata against probe-derived labels, with capacity controls (off-policy supervised probes saturate) and explicit checks for control-relevant state quotients. The in-sample leverage threshold was fitted to 15 curated tasks, but the paper transparently reports that it does not transfer and demotes it, so it is not presented as a held-out prediction. The Appendix B credit-assignment result is a conditional mathematical derivation under an explicit state-blind assumption, not an equation that reduces the empirical finding to its own input. No load-bearing self-citation or author-imported uniqueness theorem appears; the algebraic references (Schützenberger, Krohn–Rhodes) are external classical results used for classification, not to force the empirical conclusion. The linear-probe operationalization concern raised by the skeptic is a legitimate construct-validity limitation, but it is not a circularity: the paper defines 'perception gap' as 'state not linearly recoverable, though representable' and is careful to use that definition in the main claims, even if Figure 2's caption uses the shorthand 'state absent.' That is an overstatement of the operational definition, not a reduction of the prediction to its inputs. Overall, no step exhibits a fitted parameter being renamed as a prediction, no equation equals itself by construction, and no conclusion is forced by a self-citation chain.
Axiom & Free-Parameter Ledger
free parameters (3)
- p_control (partial-control probability) =
0.5
- perception-gap probe threshold =
probe recovery < 0.5
- in-sample control-leverage threshold (L <= 0.50 hard) =
0.50/0.519 boundary
axioms (5)
- domain assumption The latent DFA state is a deterministic function of the fully observed symbol prefix.
- standard math Known DFA and control distribution make optimal acceptance probability computable by backward induction.
- standard math Policy-gradient score-function identity with an action-independent baseline.
- domain assumption At a state-blind policy, the random-symbol operator on a permutation automaton is doubly stochastic with Dobrushin contraction lambda < 1, giving geometric credit decay.
- domain assumption A saturated off-policy supervised probe means the architecture can represent the state, so low on-policy linear recovery is a learning gap rather than a capacity limit.
read the original abstract
Does a reinforcement-learning agent that earns high reward represent its task's latent state, or only a reward-correlated shortcut? Usually this is unanswerable: the "true state" is undefined. We make it exactly answerable with a white-box instrument: express the task as a hidden deterministic finite automaton (DFA), let the agent observe a symbol stream and intermittently choose the next symbol under partial control, with a sparse terminal reward for acceptance. Knowing the automaton gives two things for free: the optimal return (so reward becomes an interpretable normalized score) and the exact latent state at every step (so we can probe the agent's representation without ever showing it). Reward success and latent-state learning become separately measured quantities whose coupling is governed by three axes. Optimizer strength: under weak on-policy RL the agent can lock onto a reward-shortcut, earning reward with the state probe at chance; whether it escapes is set by optimizer strength, budget, and task structure -- a pre-registered control isolates the optimizer (PPO+GAE), which escapes fastest but only partially and with high seed variance. Task structure: permutation (group-language) structure is a warning sign computable from the transition function before any training, and held out on 153 capacity-controlled fresh automata it flags perception gaps at precision 0.86 (89 of 103), in one direction only. Observation informativeness: a label-free auxiliary is vacuous when observations carry no state and recovers it sharply once they carry any (a threshold, saturating near beta=0.1). The payoff is a distinction reward-only evaluation cannot make: a perception gap (state not linearly recoverable, though representable) versus a planning gap (state recoverable but not exploited). Reward alone cannot localize which; high reward alone is not evidence the state was learned.
Figures
Reference graph
Works this paper leans on
-
[1]
L. Langosco, J. Koch, L. Sharkey, J. Pfau, L. Orseau, D. Krueger. Goal Misgeneralization in Deep RL. ICML, 2022. arXiv:2105.14111
Pith/arXiv arXiv 2022
-
[2]
R. Shah et al. Goal Misgeneralization: Why Correct Specifications Aren't Enough for Correct Goals. 2022. arXiv:2210.01790
Pith/arXiv arXiv 2022
-
[3]
B. Liu, J. T. Ash, S. Goel, A. Krishnamurthy, C. Zhang. Transformers Learn Shortcuts to Automata. ICLR, 2023. arXiv:2210.10749
Pith/arXiv arXiv 2023
-
[4]
M. P. Sch\"utzenberger. On Finite Monoids Having Only Trivial Subgroups. Information and Control, 1965
1965
-
[5]
Krohn, J
K. Krohn, J. Rhodes. Algebraic Theory of Machines, I. Transactions of the American Mathematical Society, 116:450--464, 1965
1965
-
[6]
L. Strobl et al. What Formal Languages Can Transformers Express? A Survey. TACL, 12:543--561, 2024. arXiv:2311.00208
Pith/arXiv arXiv 2024
-
[7]
Y. Chang, Y. Bisk. Language Models Need Inductive Biases to Count Inductively. 2024. arXiv:2405.20131
Pith/arXiv arXiv 2024
-
[8]
X. Huang et al. A Formal Framework for Understanding Length Generalization in Transformers. ICLR, 2025. arXiv:2410.02140
Pith/arXiv arXiv 2025
- [9]
-
[10]
K. Li, A. K. Hopkins, D. Bau, F. Vi\'egas, H. Pfister, M. Wattenberg. Emergent World Representations. ICLR, 2023. arXiv:2210.13382
Pith/arXiv arXiv 2023
-
[11]
N. Nanda, A. Lee, M. Wattenberg. Emergent Linear Representations in World Models of Self-Supervised Sequence Models. 2023. arXiv:2309.00941
Pith/arXiv arXiv 2023
-
[12]
J. Hewitt, P. Liang. Designing and Interpreting Probes with Control Tasks. EMNLP, 2019. arXiv:1909.03368
Pith/arXiv arXiv 2019
-
[13]
G. Alain, Y. Bengio. Understanding Intermediate Layers Using Linear Classifier Probes. 2016. arXiv:1610.01644
Pith/arXiv arXiv 2016
-
[14]
C. Gelada, S. Kumar, J. Buckman, O. Nachum, M. Bellemare. DeepMDP. ICML, 2019. arXiv:1906.02736
Pith/arXiv arXiv 2019
-
[15]
A. Zhang, R. McAllister, R. Calandra, Y. Gal, S. Levine. Learning Invariant Representations for RL without Reconstruction (DBC). ICLR, 2021. arXiv:2006.10742
Pith/arXiv arXiv 2021
-
[16]
T. Ni, B. Eysenbach, R. Salakhutdinov. Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPs. ICML, 2022. arXiv:2110.05038
Pith/arXiv arXiv 2022
-
[17]
M. Jaderberg et al. Reinforcement Learning with Unsupervised Auxiliary Tasks (UNREAL). ICLR, 2017. arXiv:1611.05397
Pith/arXiv arXiv 2017
-
[18]
Toro Icarte et al
R. Toro Icarte et al. Using Reward Machines for High-Level Task Specification and Decomposition in RL. ICML, 2018
2018
-
[19]
D. Angluin. Learning Regular Sets from Queries and Counterexamples. Information and Computation, 1987
1987
-
[20]
Oncina, P
J. Oncina, P. Garc\'ia. Inferring Regular Languages in Polynomial Updated Time. In Pattern Recognition and Image Analysis, World Scientific, 1992
1992
-
[21]
G. Del\'etang, A. Ruoss, J. Grau-Moya, et al. Neural Networks and the Chomsky Hierarchy. ICLR, 2023. arXiv:2207.02098
Pith/arXiv arXiv 2023
-
[22]
S. Bhattamishra, K. Ahuja, N. Goyal. On the Ability and Limitations of Transformers to Recognize Formal Languages. EMNLP, 2020. arXiv:2009.11264
Pith/arXiv arXiv 2020
-
[23]
G. Weiss, Y. Goldberg, E. Yahav. On the Practical Computational Power of Finite Precision RNNs for Language Recognition. ACL, 2018. arXiv:1805.04908
Pith/arXiv arXiv 2018
-
[24]
J. J. Michalenko, et al. Representing Formal Languages: A Comparison Between Finite Automata and Recurrent Neural Networks. ICLR, 2019. arXiv:1902.10297
Pith/arXiv arXiv 2019
-
[25]
R. Toro Icarte, et al. Learning Reward Machines: A Study in Partially Observable Reinforcement Learning. Artificial Intelligence, 323:103989, 2023. arXiv:2112.09477
Pith/arXiv arXiv 2023
-
[26]
R. I. Brafman, G. De Giacomo. Regular Decision Processes: A Model for Non-Markovian Domains. IJCAI, 2019
2019
-
[27]
M. L. Littman, R. S. Sutton, S. Singh. Predictive Representations of State. NeurIPS, 2001
2001
-
[28]
P. de Haan, D. Jayaraman, S. Levine. Causal Confusion in Imitation Learning. NeurIPS, 2019. arXiv:1905.11979
Pith/arXiv arXiv 2019
-
[29]
A. Pan, K. Bhatia, J. Steinhardt. The Effects of Reward Misspecification: Mapping and Mitigating Misaligned Models. ICLR, 2022. arXiv:2201.03544
Pith/arXiv arXiv 2022
-
[30]
J. Skalse, N. Howe, D. Krasheninnikov, D. Krueger. Defining and Characterizing Reward Gaming. NeurIPS, 2022. arXiv:2209.13085
Pith/arXiv arXiv 2022
-
[31]
B. F. van Dongen. BPI Challenge 2012 (loan application process). 4TU.ResearchData, 2012. doi:10.4121/uuid:3926db30-f712-4394-aebc-75976070e91f
work page doi:10.4121/uuid:3926db30-f712-4394-aebc-75976070e91f 2012
-
[32]
B. F. van Dongen. BPI Challenge 2015 Municipality 1 (building-permit applications). 4TU.ResearchData, 2015. doi:10.4121/uuid:a0addfda-2044-4541-a450-fdcc9fe16d17
work page doi:10.4121/uuid:a0addfda-2044-4541-a450-fdcc9fe16d17 2015
-
[33]
B. F. van Dongen. BPI Challenge 2017 (loan application process). 4TU.ResearchData, 2017. doi:10.4121/uuid:5f3067df-f10b-45da-b98b-86ae4c7a310b
work page doi:10.4121/uuid:5f3067df-f10b-45da-b98b-86ae4c7a310b 2017
-
[34]
B. F. van Dongen. BPI Challenge 2019 (purchase-order/procurement process). 4TU.ResearchData, 2019. doi:10.4121/uuid:d06aff4b-79f0-45e6-8ec8-e19730c248f1
work page doi:10.4121/uuid:d06aff4b-79f0-45e6-8ec8-e19730c248f1 2019
-
[35]
F. Mannhardt. Hospital Billing -- Event Log. 4TU.ResearchData, 2017. doi:10.4121/uuid:76c46b83-c930-4798-a1c9-4be94dfeb741
work page doi:10.4121/uuid:76c46b83-c930-4798-a1c9-4be94dfeb741 2017
-
[36]
M. de Leoni, F. Mannhardt. Road Traffic Fine Management Process. 4TU.ResearchData, 2015. doi:10.4121/uuid:270fd440-1057-4fb9-89a9-b699b47990f5
work page doi:10.4121/uuid:270fd440-1057-4fb9-89a9-b699b47990f5 2015
-
[37]
F. Mannhardt. Sepsis Cases -- Event Log. 4TU.ResearchData, 2016. doi:10.4121/uuid:915d2bfb-7e84-49ad-a286-dc35f063a460
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.