The Jacobian lens (J-lens) has been proposed as a way to read verbalizable representations from language models. However, its principle and meaning lack a detailed and theoretical discussion. We provide a mathematical view of this interpretation and of its assumed causal structure. Besides treating the J-lens as a heuristic probe, we further regard it as a first-order causal transfer operator from intermediate activations to expected future readouts. We study the Jacobian matrix as the optimal local linear approximation of the downstream mapping, analyze its global approximation behavior and bias, and identify its mathematical meaning as an expectation over anticipated future readouts. Further analysis of the Jacobian energy distribution reveals that its causal geometry is highly sparse. The energy decays with depth, concentrates in an extremely small proportion, and decomposes into diagonal pathways and specific critical positions. This decomposition further resolves the expectation of the J-lens over future outputs into short-horizon and sparse concept predictions, providing a more intuitive attribution and explanation for the ability of the J-lens to visualize concepts during the thinking process. Based on the theory, we propose a simple but effective improvement strategy and decoupling method for the J-lens, which significantly enhances the ability of the J-lens to read out correct intermediate concepts.
Figures & tables
Figure 1: Layer-wise bias analysis of J-lens for Qwen3-8B.
Figure 2: Jacobian energy analysis along the layers. The left figure indicates the relationship between absolute energy and the layers. The right figure indicates the energy concentration across layers, where the top x% of elements accumulate y% of the total energy of the layer.
Figure 3: A 3D heatmap for the energy distribution of the Jacobian at each layer and position.
association
multihop
multilingual
order-ops
poetry
typo
avg
Method
SHL
ICR
SHL
ICR
SHL
ICR
SHL
ICR
SHL
ICR
SHL
ICR
SHL
ICR
Qwen3-8B
Vanilla
1.01
4.70
1.65
16.58
0.66
40.07
0.66
12.82
0.99
19.54
0.62
14.40
0.94
18.71
Filtertopj=0.5
1.04
4.93
1.71
15.58
0.68
38.53
0.68
12.56
1.00
20.17
0.67
15.69
0.98
18.60
Filtertopj=0.2
1.05
6.36
1.73
16.35
0.69
48.40
0.67
17.36
1.06
21.13
0.69
19.03
1.00
22.14
Filtertopj=0.1
1.06
6.44
1.72
15.78
0.68
47.38
0.68
18.26
1.04
21.17
0.72
19.25
1.00
22.00
Table 1: Main results across several tasks. Bold denotes the best. Vanilla means the original J-lens. All J-lenses are trained on the same 200 examples with the same configuration.
Recent work from Anthropic introduced a new interpretability tool, called the Jacobian lens (J-lens, Gurnee et al., 2026). This method identified linear directions in the hidden states of frontier language models that had properties reminiscent of working memory-like representations. However, it is unclear what mechanistic properties of such features make them detectable to the J-lens. To this end, we introduce a mathematical theory of the J-lens, and show that it preferentially identifies coherent features -- directions that act like leading, spatially coherent Lyapunov modes. Small perturbations along these directions amplify across layers, remain roughly constant across token positions, and are consistent across different contexts. While we are agnostic to whether these properties are necessarily associated with a ``Global Workspace'', the analytically derived preferences of the J-lens nevertheless explain prior empirical applied to both model classes. We validate our analysis on a Bayes filter and a toy transformer, each optimally predicting next tokens emitted from a Hidden Markov Model (HMM). We show how modulating task parameters that change the coherence of features in the Bayes filter correspondingly and predictively modulates the response of the J-lens . We also introduce of a family of J-lenses built to emphasize different temporal scales of information processing in LLMs, which we call ω-J-lenses.
Mitchell Ostrow, Daoyuan Qian, Toni Liu +1
Department of Brain and Cognitive Sciences Massachusetts Institute of Technology · Department of Physics Cornell University
Lens methods inspect model computation by mapping intermediate activations to vocabulary tokens. Yet the concepts humans need to read out often span multiple tokens---entities, phrases, intermediate objects---making token-level readouts incomplete. Reliable multi-token readout with little model-specific preparation remains challenging. We introduce Concept Lens: token-level lens clues guide candidate concept search, then the model derives a representation for each candidate and scores it against the original activation. Across 2,400 multi-hop clozes on five LLMs (8B--70B), Concept Lens instantiated with J-lens and R-lens achieves average Rank@10 scores of 36.6% and 54.5%, respectively, compared with 21.7% for Template Lens. Concept-swap interventions on derived concept representations shift model answers toward those associated with the replacement concepts. Further experiments show that Concept Lens can also reveal what a model recognizes along the way, beyond what appears in its final answer. Our code is available at https://github.com/XijieGo/c-lens
Xijie Gong, Zimeng Huang, Tonghan Wang
College of AI, Tsinghua University, Beijing, China
A language model's prediction of its next token develops across layers, and lens methods track this process by decoding intermediate hidden states into tokens. But a lens reading reflects both the hidden state and the readout (the unembedding matrix) used to decode it. Many lenses are fit on a corpus, and we show that two lenses differing only in their fitting corpus can report different tokens for the same hidden states. We call this dependence corpus conditionality. To examine readout structure independently of the fitting corpus, we introduce Sparse Readout Prism (SRP), which decomposes the readout using only its weights and expresses any token logit or logit difference as a sum of contributions from sparse readout features. This reveals readout features as a new unit of analysis for lens readings, exposing structure that token identities can obscure and enabling comparisons across tokens, contexts, layers, and lenses. Replacing the original readout with SRP's sparse approximation reconstructs 8.9-17.3 percentage points more of the tested logit differences than the strongest of six baselines built on geometric relations among readout rows. Ablating features shifts logit differences in proportion to their SRP contributions. Although token readings vary with the fitting corpus, the dominant readout feature remains stable. Because SRP uses no corpus in its construction, it provides a control independent of the fitting corpus for lens analyses.
Matteo He, William F. Shen, Xinchi Qiu +1
Department of Computer Science and Technology, University of Cambridge