Research / research/notes/4.1_observational_probing.md

documentPhase 1 notes — §4.1 Observational probingauthorSimon-Pierre BouchercreatedTue Aug 11 2026 20:00:00 GMT-0400 (heure avancée de l’Est)statusdraft

§4.1 — Observational probing

A1. Linear probes (probing classifiers)

  • Key papers: Alain & Bengio 2016, https://arxiv.org/abs/1610.01644 ; Belinkov, CL 2022, https://arxiv.org/abs/2102.12452
  • Reveals: whether a property is linearly decodable from hidden states per layer — an "information availability" profile.
  • Type: Observational (decodability ≠ use by the model).
  • Cost: Cheap — one forward pass per prompt + logistic/ridge on cached activations. Feasible 0.5B–14B on 16–64 GB (14B fp16 ≈ 28 GB weights → 64 GB tier or 4-bit). Main cost = activation storage; stream to disk.
  • Implementations: TransformerLens run_with_cache (MPS opt-in — some ops lack stable MPS kernels: https://transformerlensorg.github.io/TransformerLens/content/getting_started.html); baukit https://github.com/davidbau/baukit (device-agnostic). Trivial in MLX.
  • Failure modes: probe learns the task (Hewitt & Liang); decodable-but-unused information; dataset/seed sensitivity; high-capacity probes find structure in random models.
  • Epistemic status: Established as decodability measurement; contested as evidence of use.

A2. Logit lens

  • Key source: nostalgebraist 2020, https://www.lesswrong.com/posts/AcKRB8wDpdaN6v6ru/interpreting-gpt-the-logit-lens
  • Reveals: layer-by-layer refinement of the next-token prediction (final LN + unembedding applied to intermediate residual).
  • Cost: Near-free; ideal Mac technique; trivial in MLX.
  • Failure modes: works on GPT-2 family, poorly/misleadingly on BLOOM/OPT/some Llamas (intermediate residual bases differ from final — documented in Belrose et al. 2023); early layers unreadable.
  • Epistemic status: Established quick diagnostic; debunked-in-part as a quantitative claim about intermediate beliefs.

A3. Tuned lens

  • Key paper: Belrose et al. 2023, https://arxiv.org/abs/2303.08112 ; code https://github.com/AlignmentResearch/tuned-lens
  • Reveals: same as logit lens with trained per-layer affine translators — less biased prediction trajectories.
  • Cost: Moderate (hours on consumer hardware ≤7B; pretrained lenses exist). Model frozen — no backward through model.
  • Failure modes: translators can "do the work" (mitigated by affine-only capacity); lens must be retrained per checkpoint AND per quantization level — directly relevant to Experiment F.
  • Epistemic status: Established refinement; validated to 20B (Pythia/NeoX/OPT).

A4. Early decoding / vocabulary-space successors (DoLa, Future Lens, Patchscopes)

  • Key papers: DoLa: Chuang et al. 2023, https://arxiv.org/abs/2309.03883 ; Future Lens: Pal et al., CoNLL 2023, https://future.baulab.info/ ; Patchscopes: Ghandeharioun et al., ICML 2024, https://arxiv.org/abs/2401.06102
  • Reveals: DoLa: which layers hold "factual" signal. Future Lens: hidden states encode ≥2 tokens ahead. Patchscopes: unifying framework — patch a hidden state into another prompt/model and let the LLM verbalize it; subsumes logit/tuned lens and fixes early-layer unreadability.
  • Cost: Cheap–moderate; all ≤14B on Mac. Patchscopes ≈ 2× inference.
  • Failure modes: verbalized readouts inherit the model's confabulation; DoLa's premise unreliable across families ("Steering off Course", ACL 2025).
  • Epistemic status: Established tools; DoLa mechanism contested. Also: SimLens (2025) https://arxiv.org/pdf/2507.17618

A5. Representation reading / RepE

A6. Concept erasure — INLP, amnesic probing, LEACE

A7. Probing pitfalls: control tasks, selectivity, MDL

  • Key papers: Hewitt & Liang, EMNLP 2019, https://arxiv.org/abs/1909.03368 ; Voita & Titov 2020 (MDL), https://arxiv.org/pdf/2003.12298
  • Substance: selectivity = task accuracy − control-task accuracy; MDL replaces accuracy with codelength. Probes overfit to: word identity/frequency memorization, spurious dataset correlates, annotation artifacts, probe capacity, shared vocabulary between splits.
  • Implication for modelmap (Experiment A is exactly this): every probe map needs shuffled-label controls, selectivity or MDL reporting, ≥2 datasets, ≥3–5 seeds, randomly-initialized-model baselines.
  • Epistemic status: Established best practice; ignoring it is the field's canonical failure.