Dense Structural Compression of Transformers via Gauge-Correct Channel Removal
Not on the map. Why?
Abstract (arXiv)
Inference energy per token drives the cost and carbon footprint of deployed transformers. It is dominated by dense matrix products that incur fused multiply-accumulate (FMA) operations and memory traffic. To reduce these computations while retaining dense tensors for high GPU throughput, we develop a methodology from first principles to adapt structural complexity during training to maximize inference utility per unit compute. Channel penalties drive entire tensor slices to zero to enable physical removal while preserving density and the network function. The natural approach, penalizing the norm of operator components acting through each channel, is provably destabilized by gauge freedom. We resolve this pathology with GaugeLasso: additive symmetric group-lasso penalties that recover a monotone function of product-norms when the network converges to gauge balance. Our equilibrium analysis enables per-channel calibration to correctly suppress slices that under-perform in inference utility per unit compute. Under adaptive pressure, the network reorganizes into depth-dependent structural profiles that can be far smaller than the architecture required to learn the task. On polynomial long division over mathbbF31, compute compresses from 148 to 255 times with perfect accuracy. On character-level language modeling, compressed models outperform the hand-designed baseline at equal FMA. On masked autoencoding, a compression trial exposes which axes were over-provisioned and which saturated, guiding a better second design. Compaction also accelerates training monotonically as the model progresses. Post-hoc pruning with the same utility ranking cannot reach these structures, showing that sustained pressure is central to discovery of efficient models. Retraining a discovered architecture recovers baseline quality on our statistical tasks, but fails on our exact algorithmic task.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 28 references that are in the map. First posted on arXiv 7 Sep 2026.
Builds on
26 of 28 references.
References · 26
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 2020GLU Variants Improve Transformercited by 1,724
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2016Pruning Filters for Efficient ConvNetscited by 1,129
- 2019Are Sixteen Heads Really Better than One?cited by 803
14 more
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
- 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 596
- 2016Learning Structured Sparsity in Deep Neural Networkscited by 538
- 2019Importance Estimation for Neural Network Pruningcited by 531
- 2017Sharp Minima Can Generalize For Deep Netscited by 462
- 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 434
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 398
- 2024SliceGPT: Compress Large Language Models by Deleting Rows and Columnscited by 378
- 2015Norm-Based Capacity Control in Neural Networkscited by 299
- 2020DynaBERT: Dynamic BERT with Adaptive Width and Depthcited by 246
- 2022Structured Pruning Learns Compact and Accurate Modelscited by 167
- 2020On the Effect of Dropping Layers of Pre-trained Transformer Modelscited by 118
- 2019Equi-normalization of Neural Networkscited by 16
Built on by
Not cited yet.