Causality $\neq$ Decodability, and Vice Versa: Lessons from Interpreting Counting ViTs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Huang, Lianghuan, Chang, Yingshan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915546028572672
author Huang, Lianghuan
Chang, Yingshan
author_facet Huang, Lianghuan
Chang, Yingshan
contents Mechanistic interpretability seeks to uncover how internal components of neural networks give rise to predictions. A persistent challenge, however, is disentangling two often conflated notions: decodability--the recoverability of information from hidden states--and causality--the extent to which those states functionally influence outputs. In this work, we investigate their relationship in vision transformers (ViTs) fine-tuned for object counting. Using activation patching, we test the causal role of spatial and CLS tokens by transplanting activations across clean-corrupted image pairs. In parallel, we train linear probes to assess the decodability of count information at different depths. Our results reveal systematic mismatches: middle-layer object tokens exert strong causal influence despite being weakly decodable, whereas final-layer object tokens support accurate decoding yet are functionally inert. Similarly, the CLS token becomes decodable in mid-layers but only acquires causal power in the final layers. These findings highlight that decodability and causality reflect complementary dimensions of representation--what information is present versus what is used--and that their divergence can expose hidden computational circuits.
format Preprint
id arxiv_https___arxiv_org_abs_2510_09794
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Causality $\neq$ Decodability, and Vice Versa: Lessons from Interpreting Counting ViTs
Huang, Lianghuan
Chang, Yingshan
Machine Learning
Computer Vision and Pattern Recognition
Mechanistic interpretability seeks to uncover how internal components of neural networks give rise to predictions. A persistent challenge, however, is disentangling two often conflated notions: decodability--the recoverability of information from hidden states--and causality--the extent to which those states functionally influence outputs. In this work, we investigate their relationship in vision transformers (ViTs) fine-tuned for object counting. Using activation patching, we test the causal role of spatial and CLS tokens by transplanting activations across clean-corrupted image pairs. In parallel, we train linear probes to assess the decodability of count information at different depths. Our results reveal systematic mismatches: middle-layer object tokens exert strong causal influence despite being weakly decodable, whereas final-layer object tokens support accurate decoding yet are functionally inert. Similarly, the CLS token becomes decodable in mid-layers but only acquires causal power in the final layers. These findings highlight that decodability and causality reflect complementary dimensions of representation--what information is present versus what is used--and that their divergence can expose hidden computational circuits.
title Causality $\neq$ Decodability, and Vice Versa: Lessons from Interpreting Counting ViTs
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.09794