Hierarchical Sparse Circuit Extraction from Billion-Parameter Language Models through Scalable Attribution Graph Decomposition
Not on the map. Why?
Abstract (arXiv)
Extracting sparse circuits from billion-parameter transformers is constrained by O(2n) search cost and pervasive feature reuse across co-active pathways. Hierarchical Attribution Graph Decomposition (HAGD) addresses this through four stages: cross-layer transcoder training, spectral coarsening of attribution graphs, graph-neural-network (GNN)-guided hierarchical traversal, and causal intervention verification, reducing worst-case complexity to O(n2 log n). Per-layer transcoders trained on the RedPajama corpus yield monosemantic dictionaries; gradient-activation products form weighted attribution graphs; normalized-Laplacian spectral clustering builds multi-resolution hierarchies; an attention-based GNN assigns circuit-membership scores at successive coarsening stages. Evaluation spans GPT-2 (117M-774M), Pythia (1.4B-6.9B), and Llama (7B-70B) across modular arithmetic, parity computation, integer sorting, coreference resolution (WinoGrande), commonsense reasoning (HellaSwag), and factual recall. Behavioral preservation reaches 91% (±2.3%) on modular arithmetic with 49-347-node circuits, while ACDC exhausts memory beyond 1.4B parameters. Cross-architecture transfer coefficients span 0.38-0.82, with within-family pairs (Llama-7B → Llama-70B) attaining 0.82. Limitations include omitted attention-head circuits, 15-20% unexplained reconstruction variance, ablation-based validation circularity, and uncertain interpretability of circuits exceeding several hundred nodes.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 23 of its 33 references that are in the map. First posted on arXiv 19 Jan 2026.
Builds on
23 of 33 references.
References · 23
- 2017Graph Attention Networkscited by 8,854
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2022In-context Learning and Induction Headscited by 930
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
- 2023How does GPT-2 compute greater-than?: Interpreting mathematical abilities in a pre-trained language modelcited by 199
11 more
- 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 145
- 2024Finding Transformer Circuits with Edge Pruningcited by 58
- 2025Weight-sparse transformers have interpretable circuitscited by 40
- 2021Defining and Quantifying the Emergence of Sparse Concepts in DNNscited by 36
- 2023Explaining Generalization Power of a DNN Using Interactive Conceptscited by 25
- 2024Automatically Identifying Local and Global Circuits with Linear Computation Graphscited by 22
- 2024Opening the AI black box: program synthesis via mechanistic interpretabilitycited by 19
- 2025Jacobian Sparse Autoencoders: Sparsify Computations, Not Just Activationscited by 9
- 2023Where We Have Arrived in Proving the Emergence of Sparse Symbolic Concepts in AI Modelscited by 8
- 2024The Local Interaction Basis: Identifying Computationally-Relevant and Sparsely Interacting Features in Neural Networkscited by 8
- 2025Discovering Transformer Circuits via a Hybrid Attribution and Pruning Frameworkcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Language Model Circuits Are Sparse in the Neuron Basis
- 2026DifFRACT: Diffusion Feature Reconstruction and Attribution for Circuit Tracing
- 2025DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scale
- 2026When Trees Are Not Enough: Learning Mixed-Topology Feature Graphs with Adaptive Graph Sparse Autoencoders
- 2025TwIST: Rigging the Lottery in Transformers with Independent Subnetwork Training
- 2026CausalGate: Causal Importance Distillation for Transformer Module Pruning
- 2025GraphGhost: Tracing Structures Behind Large Language Models
- 2026Complexity Horizons of Compressed Models in Analog Circuit Analysis