Dissociating Decodability and Causal Use in Bracket-Sequence Transformers

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Sharma, Aryan, Dawes, Cutter, Raval, Shivam
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913058317664256
author Sharma, Aryan
Dawes, Cutter
Raval, Shivam
author_facet Sharma, Aryan
Dawes, Cutter
Raval, Shivam
contents When trained on tasks requiring an understanding of hierarchical structure, transformers have been found to represent this hierarchy in distinct ways: in the geometry of the residual stream, and in stack-like attention patterns maintaining a last-in, first-out ordering. However, it remains unclear whether these representations are causally used or merely decodable. We examine this gap in transformers trained on the Dyck language (a formal language of balanced bracket sequences), where the hierarchical ground truth is explicit. By probing and intervening on the residual stream and attention patterns, we find that depth, distance, and top-of-stack signals are all decodable, yet their causal roles diverge. Specifically, masking attention to the true top-of-stack position causes a sharp drop in long-distance accuracy, while ablating low-dimensional residual stream subspaces has comparatively little effect. These results, which extend to a templated natural language setting, suggest that even in a controlled setting where the relevant hierarchical variables are known, decodability alone does not imply causal use.
format Preprint
id arxiv_https___arxiv_org_abs_2604_22128
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Dissociating Decodability and Causal Use in Bracket-Sequence Transformers
Sharma, Aryan
Dawes, Cutter
Raval, Shivam
Computation and Language
Machine Learning
When trained on tasks requiring an understanding of hierarchical structure, transformers have been found to represent this hierarchy in distinct ways: in the geometry of the residual stream, and in stack-like attention patterns maintaining a last-in, first-out ordering. However, it remains unclear whether these representations are causally used or merely decodable. We examine this gap in transformers trained on the Dyck language (a formal language of balanced bracket sequences), where the hierarchical ground truth is explicit. By probing and intervening on the residual stream and attention patterns, we find that depth, distance, and top-of-stack signals are all decodable, yet their causal roles diverge. Specifically, masking attention to the true top-of-stack position causes a sharp drop in long-distance accuracy, while ablating low-dimensional residual stream subspaces has comparatively little effect. These results, which extend to a templated natural language setting, suggest that even in a controlled setting where the relevant hierarchical variables are known, decodability alone does not imply causal use.
title Dissociating Decodability and Causal Use in Bracket-Sequence Transformers
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2604.22128