Training, Reading, and Editing Legible Transformers
Not on the map. Why?
Abstract (arXiv)
A transformer can be built from operators that are legible by construction -- bounded, named units that read as fuzzy set operations rather than dense activations -- but legibility must be pressed for during training, and the pressure has a failure mode. A crispness penalty meant to sharpen a bounded operator into a decisive detector instead collapses it into a dead constant. An identity, E[v(1-v)] = mu(1-mu) - var, shows why -- the penalty is a variance-minimizer blind to the difference between a live detector and a constant -- and names the fix: a per-channel variance floor, the target legibility metric written as a loss, which recovers both legibility and quality. A learned per-unit fraction then retires the hand-set reserved-GELU partition of prior work: given the choice the model keeps no unit as pure GELU and routes 87% of its load-bearing computation through crisp operators. The result is the most legible transformer we have built -- 78% of its feed-forward operands and 50% of its attention value channels are crisp-and-contextual detectors, and per-head legibility rises from 18% in shallow layers to 78% in deep ones. Read in the correct rotated per-layer frame, these units separate a clean detection (what a unit responds to) from a harder naming (what its output decodes to); and because the objective makes each unit crisp and sparse, edits to them are far more local -- 50-184x in the deep layers where the edit sites concentrate -- and can target explicit conjunctions a single neuron cannot express. Finally, a between-unit decorrelation pressure exposes a legibility dial: it trades a circuit's reuse for independence at no quality cost, turning concepts into single, surgically editable units and a prediction into a short explanation read off a handful of named operations. Quality holds at parity with a conventional baseline throughout.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 63 of its 75 references that are in the map. First posted on arXiv 9 Jul 2026.
Builds on
63 of 75 references.
References · 63
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2020GLU Variants Improve Transformercited by 1,724
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,239
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2020Concept Bottleneck Modelscited by 1,022
38 more
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
- 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
- 2022Mass-Editing Memory in a Transformercited by 976
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2024Refusal in Language Models Is Mediated by a Single Directioncited by 960
- 2022Toy Models of Superpositioncited by 928
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
- 2024KAN: Kolmogorov-Arnold Networkscited by 634
- 2020Towards Faithfully Interpretable NLP Systems: How should we define and evaluate faithfulness?cited by 590
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2019BLiMP: The Benchmark of Linguistic Minimal Pairs for Englishcited by 553
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
- 2018Towards Robust Interpretability with Self-Explaining Neural Networkscited by 505
- 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
- 2023Towards Best Practices of Activation Patching in Language Models: Metrics and Methodscited by 328
- 2023In-Context Learning Creates Task Vectorscited by 302
- 2023Function Vectors in Large Language Modelscited by 292
- 2020Whitening for Self-Supervised Representation Learningcited by 278
- 2019Understanding Straight-Through Estimator in Training Activation Quantized Neural Netscited by 225
- 2024Not All Language Model Features Are One-Dimensionally Linearcited by 201
- 2020Logic Tensor Networkscited by 197
- 2026Scaling Monosemanticity: Extracting Interpretable Features from Claude 3 Sonnetcited by 175
- 2020Compositional Explanations of Neuronscited by 172
- 2018Categorizing Variants of Goodhart's Lawcited by 132
- 2022Analyzing Transformers in Embedding Spacecited by 131
- 2020Logical Neural Networkscited by 118
- 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 116
- 2024Automatically Interpreting Millions of Features in Large Language Modelscited by 106
- 2020Analyzing Differentiable Fuzzy Logic Operatorscited by 100
- 2017Residual Connections Encourage Iterative Inferencecited by 74
- 2024Identifying Functionally Important Features with End-to-End Sparse Dictionary Learningcited by 64
- 2016Highway and Residual Networks learn Unrolled Iterative Estimationcited by 52
- 2023Codebook Features: Sparse and Discrete Interpretability for Neural Networkscited by 45
- 2024Decomposing The Dark Matter of Sparse Autoencoderscited by 42
- 2024Mathematical Models of Computation in Superpositioncited by 34
- 2022Interpreting Neural Networks through the Polytope Lenscited by 33
50 most cited of 63 All 63 (JSON)
Built on by
Not cited yet.