Training Trajectories Determine Circuit Removability in Annealable Soft-Prior Transformers
On the map: Mechanistic interpretability → Circuits and mechanistic analysis
Abstract (arXiv)
Soft positional priors can help small Transformers learn retrieval circuits, but it is unclear whether the resulting circuits remain functional once the prior is removed. We test this with an annealable soft-prior Transformer whose attention biases can be learned, faded, or zeroed during training and evaluation. On associative recall, unforced models perform well with the prior active (0.772 ± 0.020) but collapse at zero gate (0.095 ± 0.009). Smooth fade-to-zero training preserves high zero-gate accuracy (0.734 ± 0.028), whereas forced-zero training, hard switching, and post hoc continuation fail to recover the same effect. The pattern also appears on Markov induction. Linear regression ICL provides a boundary case because zero-gate training can learn that task directly. Mechanistic traces show that circuit consolidation occurs after the gate reaches zero, even though the responsible heads vary across seeds. These results suggest that circuit removability in small discrete retrieval tasks depends on the training trajectory, not just the final architecture.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Circuits and mechanistic analysis, one of 13 directions in Mechanistic interpretability. Builds on 22 of its 27 references that are in the map. First posted on arXiv 9 Sep 2026.
Builds on
22 of 27 references.
Closest ancestors · 3
Same direction, newest first
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretability28 Apr
- 2023Progress measures for grokking via mechanistic interpretability12 Jan
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 small1 Nov
Other references · 19
- 2017Attention Is All You Needcited by 60,361
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 2018Self-Attention with Relative Position Representationscited by 1,120
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2016Deep Networks with Stochastic Depthcited by 979
- 2022In-context Learning and Induction Headscited by 964
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 684
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 522
- 2021ConViT: Improving Vision Transformers with Soft Convolutional Inductive Biasescited by 484
7 more
- 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 439
- 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 403
- 2021Early Convolutions Help Transformers See Bettercited by 396
- 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 327
- 2022KERPLE: Kernelized Relative Positional Embedding for Length Extrapolationcited by 86
- 2024Contextual Position Encoding: Learning to Count What's Importantcited by 54
- 2024DAPE: Data-Adaptive Positional Encoding for Length Extrapolationcited by 31
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Action Shaping: Policies Absorb What They Can Expresscited by 0
- 2026Action Shaping: Policies Absorb What They Can Express26 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026Routing Absorption in Sparse Attention: Why Random Gates Are Hard to Beat
- 2025Forgetting Transformer: Softmax Attention with a Forget Gate
- 2025How Transformers Learn In-Context Recall Tasks? Optimality, Training Dynamics and Generalization
- 2026Mixture of Chapters: Scaling Learnt Memory in Transformers
- 2026First-Order Predictable but Pairwise Fragile: Local Task Adaptation in Trained Transformers
- 2024Remembering Transformer for Continual Learning
- 2023The Transient Nature of Emergent In-Context Learning in Transformers
- 2026Learning to Remember, Learn, and Forget in Attention-Based Models