Bag of Dims: Training-Free Mechanistic Interpretability via Dimension-Level Sign Patterns
Abstract
The standard basis of transformer hidden states serves as a training-free, architecture-general feature representation where individual dimensions encode semantic content through signs and confidence through magnitudes, functioning as independent binary registers without requiring learned rotations or optimization.
We show the standard basis of transformer hidden states already provides a training-free, architecture-general feature basis. Individual dimensions encode semantic content via their signs (+/-1) and confidence via their magnitudes, acting as independent binary registers; a feature is a subset of dimensions with a consistent sign pattern, read by counting sign agreements with no learned rotation. We validate this Bag of Dims framework across seven models spanning language (Qwen 3.5-4B, Gemma 3-4B, Mistral 7B, Qwen3-32B), vision (DINOv2, ViT-Base), and audio (AST). Signs alone carry predictive content: unit-magnitude sign patterns preserve 60-93% top-5 next-token accuracy through the LM head, and decoder-free Hamming scoring reaches 80-90% top-4096. From a single-token cache (one forward pass per token, no context, no labels), we detect 175 categories at AUC 0.97-0.99 by sign agreement; a trained probe adds only +0.018 AUC and converges to axis-aligned weights. These features are causally operative: they survive the K/V attention projections, trace to the FFN neuron coalitions that write them (random-weight controls never reproduce this), and flipping a feature's signs during the live forward pass suppresses its concept across four language models, magnitude-matched and concept-specific. Dimensions stay independent throughout (pairwise mutual information below 0.006 bits). The structure is not specific to language: the same per-dimension signs appear in self-supervised vision (DINOv2, 9/12 ImageNet superclasses), supervised vision (ViT-Base, 11/12), and audio (AST, 50/50 ESC-50 categories), so it reflects transformer training in general, not the language-modeling objective. The standard basis already suffices for feature reading at one forward pass, no optimization, no GPU-days. The open problem shifts from finding the right rotation to cataloging what each dimension encodes.
Community
TL;DR: you can read interpretable features out of transformer hidden states with
no trained SAE and no probe, straight from the standard basis.
The idea is to treat each dimension as an independent register, its sign carries
content, its magnitude carries confidence. A "feature" is then just a subset of
dims with a consistent sign pattern, read by counting sign agreements. One forward
pass, zero training.
Look at the figures in the paper, they directly triggered the idea to look the raw dims.
Some results that surprised me:
- Sign alone (all magnitudes set to 1) preserves 60–93% of top-5 next-token accuracy
- 175 semantic categories deteted from a single-token cache with zero labels;
a trained probe adds only +0.018 AUC and converges to axis-aligned weights - Same structure appears in vision (DINOv2, ViT) and audio (AST), looks like a
property of transformer training, not language - Flipping a feature's signs mid-forward-pass causally suppresses the concept
It's deliberately simple, and simple enough that you can hand the paper to a coding agent, and reproduce in ~15-20Mins on a model of your choice on your laptop, I kept the code aways it would be better testament of the methodology both if it work or fails
Would love scrutiny — especially from folks deep in SAEs
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Architecture Determines Observability in Transformers (2026)
- fmxcoders: Factorized Masked Crosscoders for Cross-Layer Feature Discovery (2026)
- Perplexity Can Miss SAE Feature Damage Under Quantization (2026)
- Gated Subspace Inference for Transformer Acceleration (2026)
- Polymorphism Is Rotation: Operational Mechanistic Interpretability from a Two-Layer Transformer to Pythia-70m (2026)
- Where does Absolute Position come from in decoder-only Transformers? (2026)
- Sign-Aware Gated Sparse Autoencoders: Modeling Anticorrelated Features with Bi-Jump-ReLU Activations (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2606.12629 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper

