arXiv Machine Learning

Markovian Circuit Tracing: Ground-Truth Tests of Transformer State Abstractions

Markovian Circuit Tracing (MCT) is a benchmark that tests whether transformer activations encode a compact state abstraction with recoverable transition structure. In experiments with small causal transformers trained on sequences from four‑state hidden Markov models, models achieve near‑optimal validation loss and recover transition matrices accurately under easy and medium observability, but performance collapses to control levels under hard observability. Various controls—including shuffled data, untrained activations, and sparse autoencoders—demonstrate that the recovered transition structure is genuinely causally useful rather than a trivial artifact.

arXiv Machine Learning
Sep 1

The Intervention Gap in Latent World Models

The paper introduces the concept of intervention fidelity in latent world models, measuring whether a model’s open‑loop transitions align with actual environment interventions. Experiments on TD‑MPC2, Cheetah, and DreamerV3 show that high reward fit does not guarantee fidelity, and that self‑supervised models can outperform task‑anchored ones in preserving intervention effects. The authors propose a capture‑gated audit to localize failures and argue that fidelity must be directly audited on the model’s native interface.

By Donna Vakalis
arXiv AI
Sep 28

FLIP: Final Layer Inference-Time Probing for Vision-Language Models

FLIP is a final‑layer inference‑time probe designed to test whether a logit‑facing intervention site in an open‑weight vision‑language model (VLM) supports structured, task‑linked computation rather than generic perturbation. The probe applies elementwise flooring to the final normalized hidden state before logit computation, leaving other model components unchanged. By sweeping intervention strength on a controlled detection/counting task, FLIP identifies three regimes—negligible change, a bounded interior regime with improved detection recall and reduced counting error, and over‑suppression—while a four‑criterion protocol ensures the observed effects are mechanistically interpretable.

By Drandreb Earl O. Juanico, Rowel O. Atienza
arXiv AI
Oct 9

Reward Observability and the Limits of Offline Checkpoint Selection in RSSM World Models

The paper investigates the closed‑loop behavior of a recurrent state‑space model (RSSM) trained on human demonstrations in LunarLander‑v3. Using the model for zero‑shot CEM MPC and actor‑critic training in imagination, the authors find that a model‑based A2C policy trained on a single checkpoint achieves returns comparable to the best model‑free A2C policy while requiring far fewer real transitions. They introduce the Reward Observability Fraction (ROF) to quantify how much of the reward gradient lies in the observable subspace of the latent dynamics, but show that ROF does not predict the collapse of MPC in offline settings, leaving the problem of predicting closed‑loop failure unresolved.

By Nikolai Smolyanskiy, Jonathan Shock
arXiv Computation and Language
Aug 28

SCIT: Testing Causal Cache Carriers in Latent Chain-of-Thought Models

SCIT (Suffix Cache Interchange Test) is a causal protocol designed to identify which transformer components carry counterfactual computations in latent chain-of-thought models. By constructing exact source‑recipient counterfactuals and applying sufficiency tests, K/V splits, hidden‑state controls, and semantic source controls, SCIT demonstrates that counterfactual arithmetic primarily transfers through value‑cache suffix trajectories rather than hidden states or keys. The method reveals carrier‑regime shifts across different GPT‑2 checkpoints, providing a cache‑level diagnostic and a competence‑gated carrier map for arithmetic mechanisms.

By Yi Ding, Lijun Huang, Menglin Yang