VASAE: Naming SAE Dictionary Directions with Vocabulary-Aligned Anchoring
Not on the map. Why?
Abstract (arXiv)
Sparse autoencoders (SAEs) provide useful decompositions of Transformer residual streams, but their learned features are usually named post hoc rather than directly connected to the Transformer's token vocabulary. We introduce Vocabulary-Aligned Sparse Autoencoder (VASAE), a method that trains SAE features under vocabulary-aligned anchoring and assigns each feature an intrinsic token name: the token string whose embedding is nearest to that feature. Without reducing reconstruction quality compared with a standard SAE, VASAE produces dictionaries with vocabulary-aligned features. Using a 0.8 cutoff on the nearest-token alignment score, dictionaries trained on GPT-2-small post-residual streams align about 90% of features in layers 0--10. In Llama-3.1-8B, representative shallow and middle-layer dictionaries contain strongly aligned features, including 92.8% in the shallow layer, while the representative final-layer dictionary shows limited alignment. After subtracting the sentence-level mean sparse code, case studies show that many remaining intrinsic token names are relevant to nearby input tokens. These results suggest that vocabulary-aligned anchoring can connect learned features to intrinsic token names during training, complementing post hoc interpretation of learned dictionaries.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 26 references that are in the map. First posted on arXiv 26 Jun 2026.
Builds on
19 of 26 references.
References · 19
- 2017Attention Is All You Needcited by 59,932
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
- 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 770
- 2019Designing and Interpreting Probes with Control Taskscited by 608
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
- 2018Analysis Methods in Neural Language Processing: A Surveycited by 276
7 more
- 2016Using the Output Embedding to Improve Language Modelscited by 266
- 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 225
- 2013k-Sparse Autoencoderscited by 209
- 2017All-but-the-Top: Simple and Effective Postprocessing for Word Representationscited by 205
- 2024Automatically Interpreting Millions of Features in Large Language Modelscited by 106
- 2016Tying Word Vectors and Word Classifiers: A Loss Framework for Language Modelingcited by 87
- 2024NNsight and NDIF: Democratizing Access to Open-Weight Foundation Model Internalscited by 43
Built on by
Not cited yet.