mlsanityData to 7 Oct 2026

Causal Mechanism Reduction: Mechanism Replacement for Neural Network Pruning and Abstraction

arXiv 2602.24266 ↗First posted 27 Feb 2026cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Which internal mechanisms of a neural network can be replaced while preserving the computation it performs? Structured pruning asks for smaller deployable networks; causal abstraction asks for high-level models that commute with interventions. We introduce causal mechanism reduction (CMR), a framework that treats a trained network as a deterministic structural causal model and replaces selected internal variables by constants or affine functions of retained variables. These replacements compile exactly into smaller dense networks by bias and weight folding, and induce reduced causal models testable with interchange interventions. We derive a unified second-order replacement-risk objective whose special cases recover mean replacement, variance-based pruning (VBP), logit-distortion scoring, and affine neuron merging, together with a margin-based certificate linking logit distortion to interchange-intervention agreement. The framework also exposes a basic invariance requirement: functionally identical ReLU networks should induce the same reduction. Under exact positive-scaling reparameterizations, VBP's kept set collapses to chance-level overlap while the logit-distortion score is exactly invariant. Empirically, CMR variants are competitive with VBP under matched fine-tuning of DeiT-Tiny on ImageNet-100; the clearer separation appears in the invariance and interchange tests, where the logit-distortion score preserves kept sets and consistently improves distributional fidelity. CMR thus gives pruning, compilation, and causal-abstraction verification a common object to optimize and verify.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 31 of its 41 references that are in the map. First posted on arXiv 27 Feb 2026.

Builds on

31 of 41 references.

References · 31

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
  3. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  4. 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
  5. 2019Contrastive Multiview Codingcited by 1,297
  6. 2016Pruning Filters for Efficient ConvNetscited by 1,129
  7. 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
  8. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
  9. 2019Are Sixteen Heads Really Better than One?cited by 803
  10. 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
  11. 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 697
  12. 2017Learning Efficient Convolutional Networks through Network Slimmingcited by 673
19 more
  1. 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 643
  2. 2021Causal Abstractions of Neural Networkscited by 401
  3. 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
  4. 2023DepGraph: Towards Any Structural Pruningcited by 202
  5. 2023Finding Alignments Between Interpretable Causal Variables and Distributed Neural Representationscited by 197
  6. 2023Causal Abstraction: A Theoretical Foundation for Mechanistic Interpretabilitycited by 194
  7. 2022A Fast Post-Training Pruning Framework for Transformerscited by 161
  8. 2017Causal Consistency of Structural Equation Modelscited by 114
  9. 2021Inducing Causal Structure for Interpretable Neural Networkscited by 107
  10. 2018Abstracting Causal Modelscited by 104
  11. 2015Reducing Overfitting in Deep Networks by Decorrelating Representationscited by 94
  12. 2019BackPACK: Packing more into backpropcited by 77
  13. 2023X-Pruner: eXplainable Pruning for Vision Transformerscited by 66
  14. 2019Approximate Causal Abstractioncited by 62
  15. 2025The Non-Linear Representation Dilemma: Is Causal Abstraction Enough for Mechanistic Interpretability?cited by 34
  16. 2022Causal Abstraction with Soft Interventionscited by 23
  17. 2022Data-Efficient Structured Pruning via Submodular Optimizationcited by 22
  18. 2024Revisiting Scalable Hessian Diagonal Approximations for Applications in Reinforcement Learningcited by 5
  19. 2025Variance-Based Pruning for Accelerating and Compressing Trained Networkscited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.