English

Post-Training Statistical Calibration for Higher Activation Sparsity

Machine Learning 2024-12-11 v1 Artificial Intelligence

Abstract

We present Statistical Calibrated Activation Pruning (SCAP), a post-training activation pruning framework that (1) generalizes sparsification by input activations of Fully-Connected layers for generic and flexible application across Transformers, and (2) features a simple Mode-Centering technique to pre-calibrate activation distributions for maximizing post-training sparsity. Our results demonstrate robust Pareto efficiency compared to prior methods, translating to a 1.5x additional LLM decoding speedup against CATS at iso model quality. SCAP effectiveness is empirically verified across a wide range of models, including recent Transformer Decoders, MoE, Mamba2, Encoding Transformer, and pre-quantized models, highlighting its practicality and scalability. The code is available at: https://github.com/IntelLabs/SCAP.

Cite

@article{arxiv.2412.07174,
  title  = {Post-Training Statistical Calibration for Higher Activation Sparsity},
  author = {Vui Seng Chua and Yujie Pan and Nilesh Jain},
  journal= {arXiv preprint arXiv:2412.07174},
  year   = {2024}
}

Comments

ENLSP-IV NeurIPS Workshop 2024

R2 v1 2026-06-28T20:28:57.373Z