揭示视觉Transformer中可解码性与因果性的分离现象
Causality $\neq$ Decodability, and Vice Versa: Lessons from Interpreting Counting ViTs
- 通过激活替换实验检验空间与CLS token的因果作用
- 中层物体token因果性强但可解码性弱,末层则相反
- 发现隐藏计算回路,适合研究模型可解释性的人看
机制可解释性旨在揭示神经网络内部组件如何导致预测结果。一个长期挑战是区分两个常被混淆的概念:可解码性(信息从隐藏状态中恢复的能力)和因果性(这些状态对输出的功能影响程度)。本文研究了微调用于目标计数的视觉变换器(ViTs)中二者的关系。采用激活修补法,通过在干净-损坏图像对间移植激活,测试空间与CLS token的因果作用。同时,训练线性探测器评估不同深度下计数信息的可解码性。结果显示系统性错位:中层物体token虽可解码性弱,但因果影响强;而末层物体token可解码性强,却功能上无用。类似地,CLS token在中层变得可解码,仅在最后层获得因果力。这表明可解码性与因果性反映表征的不同维度——信息是否存在,与信息是否被使用——它们的差异可揭示隐藏计算通路。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。