When ReAct Phase Probes Look Abstract
scale 3Mechanistic Interpretability ResearcherAI Interpretability @ Illinois (UIUC)Apr 2026 – PresentWork in progress
A preregistered test of whether a linear probe that decodes ReAct reasoning phase at 98% accuracy is reading an abstract reasoning state or a correlated surface cue.
Overview
Linear probing is the workhorse of interpretability: train a simple classifier on a model's internal activations and see whether some property can be read out. The permissiveness is the point — and also the trap. A probe that predicts a label has not proven the model uses that label; it may be latching onto anything correlated with it.
This paper aims that critique at a specific, tempting result. ReAct traces alternate labelled Thought / Action / Observation segments, and a probe can decode which phase a token belongs to almost perfectly. That invites the reading that the model maintains an internal "I am thinking now / acting now" state. But phase is confounded by construction: thoughts contain prose and arithmetic, actions contain calls, observations contain structured results. Marker text, position in the template, and content type all move together.
An earlier one-week pilot on GPT-2 Small — the ReAct Interp repository — had already found phase trivially decodable at every layer and flagged exactly this confound; this paper is the controlled follow-up that pilot called for.
The response is a decorrelation ladder — a procedurally generated dataset that crosses every phase with mathematics, code, and prose, shuffles order and repetition, and must clear frozen lexical, position, and content-balance gates before any activation is examined. Thresholds and headline layers were committed to version control before confirmatory runs. Accuracy is then normalised against a matched natural-trace denominator into a retained-signal score, and markers are stripped rung by rung.
The result inverts the initial story, and the paper reports the collapse rather than the headline.
Terms, in plain English
- Linear probe
- A simple classifier trained on a model’s internal activations to test whether some piece of information is present there in an easily readable form.
- ReAct
- A prompting format in which a model alternates between writing a Thought, taking an Action, and reading an Observation, rather than answering in one shot.
- Residual stream
- The running vector each transformer layer reads from and writes back into — effectively the model’s working memory as text passes through it.
- Base vs. instruction-tuned (IT)
- The same model before and after being fine-tuned to follow instructions. Comparing them isolates what that fine-tuning changed inside.
- Sparse autoencoder (SAE)
- A tool that decomposes a dense activation vector into a large set of mostly-inactive, more individually interpretable features.
- Activation patching
- Copying an internal activation from one run into another to test whether that component actually causes a behaviour, rather than merely correlating with it.
- Preregistration
- Committing the analysis plan and pass/fail thresholds in advance, so results cannot be reinterpreted after seeing them.
- Balanced accuracy
- Accuracy averaged per class, so a classifier cannot score well by favouring whichever label happens to be most common.
- Bootstrap interval
- A range of plausible values produced by repeatedly resampling the data; if it crosses zero, the effect is not distinguishable from none.
Figures
The headline result. Retained signal R at fixed relative-depth layers across the decorrelation ladder, with 95% paired-bootstrap intervals and the frozen illusion / partial / abstract bands. Removing markers roughly halves retention at both scales.
Paper · resources/116_When_ReAct_Phase_Probes_Lo.pdf p.5, Figure 2
Natural-trace probe accuracy stays near-ceiling even after markers and newlines are removed — the appearance of an abstract phase variable, before the controlled test undercuts it.
Paper · resources/116_When_ReAct_Phase_Probes_Lo.pdf p.5, Figure 1
Feature-level corroboration at 2B: the top phase-selective sparse-autoencoder features are overwhelmingly content detectors, and the causal patch effect is small and L20-driven.
Paper · resources/116_When_ReAct_Phase_Probes_Lo.pdf p.6, Figure 3
Matched-denominator retention for every scale × rung cell.
Paper · resources/116_When_ReAct_Phase_Probes_Lo.pdf p.6, Table 1
- mechanistic-interpretability
- probing
- sparse-autoencoders
- gemma-2
- preregistration
- transformerlens
- slurm
- negative-result