arXiv:2609.07264cs.LGstat.ML2026-09

通过动态剪枝通道压缩Transformer,显著降低推理能耗且保持精度。

Dense Structural Compression of Transformers via Gauge-Correct Channel Removal

论文配图:Dense Structural Compression of Transformers via Gauge-Correct Channel Removal
图 1 · 摘自论文原文
  • 引入规范修正的通道惩罚机制,实现可物理移除的稀疏结构。
  • 在有限域多项式除法任务中,计算量压缩255倍仍保持精确。
  • 适合追求高效推理的模型部署者,尤其适用于算法类任务。

单标记推理能耗主导了部署型Transformer的成本与碳足迹,主要源于密集矩阵乘法带来的融合乘加运算和内存访问。为在保留密集张量以维持高GPU吞吐量的同时减少计算,我们从基础原理出发,提出一种训练期间自适应调整结构复杂度的方法,以最大化单位算力的推理效用。通过通道惩罚使整个张量切片归零,实现物理移除,同时保持密度与网络功能。自然方法(对各通道作用算子分量范数施加惩罚)因规范自由而被证明不稳定。我们提出GaugeLasso:添加对称组套索惩罚,在网络收敛至规范平衡时恢复乘积范数的单调函数。均衡分析支持逐通道校准,正确抑制单位算力下推理效用低的切片。在自适应压力下,网络重组为深度依赖的结构形态,其规模远小于完成任务所需的原始架构。在有限域𝔽₃₁上的多项式除法任务中,计算量从148降至255倍且精度完美;字符级语言建模中,压缩模型在同等乘加操作数下优于人工设计基线;掩码自编码压缩试验揭示了过配置与饱和轴,指导更优第二轮设计。压缩还单调加速训练过程。事后按同一效用排序剪枝无法达到此类结构,表明持续压力对发现高效模型至关重要。重训发现的架构可在统计任务上恢复基线性能,但在精确算法任务上失败。

原文摘要 · Abstract (English)

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 $\mathbb{F}_{31}$, 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.

模型压缩Transformer能量效率剪枝

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。