DEV Community

AI OpenFree
AI OpenFree

Posted on

LLMs carry a judgment signal you can read with a linear probe (and why steering it fails)

LLMs carry a judgment signal you can read with a linear probe (and why steering it fails)

LLM hidden states encode reward-related information: roughly, whether the current trajectory is heading toward a correct answer.

We reproduced this on our own hardware, added the controls we felt were missing, and got a clear picture of what the signal can and cannot do.

Setup

  • Generate N answers, label each by final correctness
  • Take the last-token hidden state at layer L
  • Fit a linear probe (ridge) predicting correctness
  • Report AUC on a held-out set

No fine-tuning, no extra model.

Result 1: the signal is real across scales

Model Probe AUC
1.5B 0.81
4B (Gemma-4 based) 0.96
9B 0.72
27B 0.95
26B MoE (Gemma-4) 0.987

Note it is not monotone in size. Architecture family matters more than parameter count.

Result 2: it is causal, not decorative

Zeroing the top 1% of dimensions by probe weight at one layer:

  • accuracy 0.567 to 0.108 (minus 45.8 points)
  • zeroing random dimensions, same count: minus 0.8 points

The control is the important half. Without it you cannot distinguish "this direction matters" from "perturbing anything hurts".

Result 3: reading works, steering does not

This is where most of our compute went, and where the result was negative:

Intervention Held-out effect
Add alpha * w to activations minus 4.7 points
Sigma-normalized dose no gain
Clamp and gate (self-limiting) no gain
Best-of-N selection by probe loses to plain majority vote

And the killer control: injecting a random direction at matched magnitude produced an identical score (0.575 vs 0.575). The value direction was not special; only the magnitude was.

What it is genuinely good for

Selective prediction. Rank answers by probe score, answer only the top X percent:

Coverage Accuracy
100% 73.0%
80% 85.0%
70% 92.9%
60% 100%

Held-out detection AUC: 0.940.

The probe is an excellent thermometer and a useless heater. You can read whether the model knows. You cannot push it into knowing.

The caveat that bit us

The probe is domain-dependent. Same model, same method: 0.94 AUC on math reasoning, 0.66 on factual/legal questions.

Models that are confused leave a trace. Models that are confidently ignorant do not.

If you deploy this, recalibrate per domain and report per-domain numbers. A single headline AUC is misleading.


From VIDRAFT, a Korean deep-tech company running an **AI Foundry* — we diagnose, breed and optimize AI models for specific industries. Open models: Hugging Face · vidraft.net*

Top comments (0)