mlsanityData to 7 Oct 2026

Hierarchical Sparse Circuit Extraction from Billion-Parameter Language Models through Scalable Attribution Graph Decomposition

arXiv 2601.12879 ↗First posted 19 Jan 2026cs.LG cs.AI cs.CLcited by 0

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

  1. 2017Graph Attention Networkscited by 8,854
  2. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  3. 2017Axiomatic Attribution for Deep Networkscited by 3,350
  4. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  5. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  6. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
  7. 2022In-context Learning and Induction Headscited by 930
  8. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  9. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
  10. 2024Scaling and evaluating sparse autoencoderscited by 530
  11. 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
  12. 2023How does GPT-2 compute greater-than?: Interpreting mathematical abilities in a pre-trained language modelcited by 199
11 more
  1. 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 145
  2. 2024Finding Transformer Circuits with Edge Pruningcited by 58
  3. 2025Weight-sparse transformers have interpretable circuitscited by 40
  4. 2021Defining and Quantifying the Emergence of Sparse Concepts in DNNscited by 36
  5. 2023Explaining Generalization Power of a DNN Using Interactive Conceptscited by 25
  6. 2024Automatically Identifying Local and Global Circuits with Linear Computation Graphscited by 22
  7. 2024Opening the AI black box: program synthesis via mechanistic interpretabilitycited by 19
  8. 2025Jacobian Sparse Autoencoders: Sparsify Computations, Not Just Activationscited by 9
  9. 2023Where We Have Arrived in Proving the Emergence of Sparse Symbolic Concepts in AI Modelscited by 8
  10. 2024The Local Interaction Basis: Identifying Computationally-Relevant and Sparsely Interacting Features in Neural Networkscited by 8
  11. 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

  1. 2026Language Model Circuits Are Sparse in the Neuron Basis
  2. 2026DifFRACT: Diffusion Feature Reconstruction and Attribution for Circuit Tracing
  3. 2025DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scale
  4. 2026When Trees Are Not Enough: Learning Mixed-Topology Feature Graphs with Adaptive Graph Sparse Autoencoders
  5. 2025TwIST: Rigging the Lottery in Transformers with Independent Subnetwork Training
  6. 2026CausalGate: Causal Importance Distillation for Transformer Module Pruning
  7. 2025GraphGhost: Tracing Structures Behind Large Language Models
  8. 2026Complexity Horizons of Compressed Models in Analog Circuit Analysis