mlsanityData to 14 Oct 2026

When Does Reward Teach State? A Hidden-Automaton Instrument and a Group-Language Warning Signal

arXiv 2607.11953 ↗First posted 11 Jul 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Does a reinforcement-learning agent that earns reward learn its task's hidden state? We study this question with hidden finite automata that the agent partially controls. Because each automaton is known, we can normalize reward by the best achievable return and probe the network for the true state at every step. Together the two measurements separate failures that reward alone conflates. An agent can encode too little of a state its network could hold, or encode the state and still control poorly. Weak on-policy RL matches random play while the state probe stays at chance. State learning depends on the optimizer, the training budget, and the task's structure. Permutation automata provide a warning before training: no input symbol maps two distinct states to the same successor. On a stratified held-out set, 86 of 103 permutation automata fail the state probe, and the classification is stable across probe read-outs and recovery thresholds. Most of these failures come with weak reward. High reward without the state occurs but is rare. Non-permutation automata can also fail. Oracle-normalized reward alone therefore does not establish that the task's state was learned.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 29 of its 48 references that are in the map. First posted on arXiv 11 Jul 2026.

Builds on

29 of 48 references.

References · 29

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
  3. 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,258
  4. 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
  5. 2020Shortcut Learning in Deep Neural Networkscited by 1,969
  6. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  7. 2019Designing and Interpreting Probes with Control Taskscited by 640
  8. 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 436
  9. 2022Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Taskcited by 431
  10. 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 404
  11. 2022The Effects of Reward Misspecification: Mapping and Mitigating Misaligned Modelscited by 328
  12. 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 319
17 more
  1. 2019Causal Confusion in Imitation Learningcited by 288
  2. 2019Theoretical Limitations of Self-Attention in Neural Sequence Modelscited by 288
  3. 2022Transformers Learn Shortcuts to Automatacited by 268
  4. 2022Neural Networks and the Chomsky Hierarchycited by 217
  5. 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 207
  6. 2022Goal Misgeneralization: Why Correct Specifications Aren't Enough For Correct Goalscited by 126
  7. 2023What Formal Languages Can Transformers Express? A Surveycited by 124
  8. 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 122
  9. 2018On the Practical Computational Power of Finite Precision RNNs for Language Recognitioncited by 117
  10. 2021Chess as a Testbed for Language Model State Trackingcited by 72
  11. 2024Emergent World Models and Latent Variable Estimation in Chess-Playing Language Modelscited by 62
  12. 2024A Formal Framework for Understanding Length Generalization in Transformerscited by 45
  13. 2023Emergent Representations of Program Semantics in Language Models Trained on Programscited by 30
  14. 2024Language Models Need Inductive Biases to Count Inductivelycited by 25
  15. 2021Learning Reward Machines: A Study in Partially Observable Reinforcement Learningcited by 16
  16. 2025Transformers Provably Learn Chain-of-Thought Reasoning with Length Generalizationcited by 14
  17. 2019Representing Formal Languages: A Comparison Between Finite Automata and Recurrent Neural Networkscited by 9

Built on by

Not cited yet.