cs.CLAug 26, 2026

Short Horizons and Sparse Concepts: a Mathematical View of the Readout in the J-lens

Authors: Shi-Qi Yan, Kai-Xuan Ding, Chao-Hong Tan, Qian Chen, Wen Wang, Xiangang Li, Zhen-Hua Ling

Organizations: Alibaba Token Hub, Alibaba Group

Abstract

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

Explore similar work

CardsList
  1. A Mathematical Theory of the Jacobian Lens

    Oct 22, 2026Mitchell Ostrow, Daoyuan Qian, Toni Liu +1LLM InterpretabilityRepresentation Geometry in Language Models

  2. Verbalizing Multi-Token Concepts in LLMs

    Aug 31, 2026Xijie Gong, Zimeng Huang, Tonghan WangLLM Interpretability

  3. Sparse Readout Prism: Explaining Logit-Lens Scores in Features Instead of Tokens

    Sep 1, 2026Matteo He, William F. Shen, Xinchi Qiu +1Transformer InterpretabilityLLM Interpretability