Papers
Topics
Authors
Recent
Search
2000 character limit reached

Post-Training N:M Activation Pruning

Updated 12 July 2026
  • The paper introduces a method that sparsifies input activations via an exact N:M structured mask, enabling dynamic, input-adaptive compression in large language models.
  • It employs weight-aware scoring and calibration techniques such as robust-norm scoring and mode-centering to preserve generative performance while reducing computation.
  • The approach achieves notable acceleration in compute and I/O efficiency compared to traditional weight pruning by leveraging hardware-friendly semi-structured sparsity patterns.

Post-training N ⁣: ⁣MN\!:\!M activation pruning is an inference-time sparsification regime in which a pretrained model is used without retraining or fine-tuning, while the input activation tensor XX of selected linear layers is pruned so that in every contiguous group of MM elements only NN are kept and the remaining MNM-N are set to zero. In LLMs, the resulting sparse activation is then multiplied by the original dense weight matrix, yielding a sparse-dense matrix multiplication; the intended benefits are dynamic, input-adaptive compression, reductions in I/O overhead, and compatibility with hardware-friendly semi-structured sparsity patterns such as 2 ⁣: ⁣42\!:\!4, 4 ⁣: ⁣84\!:\!8, and 8 ⁣: ⁣168\!:\!16 (An et al., 4 Aug 2025, Alanova et al., 26 Sep 2025).

1. Definition and mathematical setting

A generic starting point is the fully connected layer

Y=XW+b,Y = XW + b,

with XRN×ICX\in\mathbb{R}^{N\times IC}, XX0, XX1, and XX2. In post-training activation pruning, the pruning target is the input activation tensor XX3, not the weight matrix XX4. This framing is explicit in "Post-Training Statistical Calibration for Higher Activation Sparsity" (Chua et al., 2024).

Within this setting, unstructured post-training activation pruning is typically written as elementwise thresholding: XX5 This rule produces dynamic, input-dependent sparsity, but it does not enforce exact local structure. By contrast, "Amber Pruner: Leveraging N:M Activation Sparsity for Efficient Prefill in LLMs" (An et al., 4 Aug 2025) studies strict structured masking of the input activations to linear projections during inference, under the standard semi-structured definition that within every XX6 consecutive elements, at most XX7 are nonzero.

The same topic is framed at a broader level in "Lightweight error mitigation strategies for post-training N:M activation sparsity in LLMs," whose abstract describes post-training XX8 activation pruning as a hardware-friendly, minimally calibrated alternative that can preserve generative behavior better than weight pruning at equivalent sparsity levels (Alanova et al., 26 Sep 2025). This places XX9 activation pruning at the intersection of structured compression, dynamic sparsity, and deployment-oriented LLM inference.

2. Distinction from adjacent pruning paradigms

A central conceptual distinction is between pruning activations and using activations only to score weights. "EGGS-PTP: An Expander-Graph Guided Structured Post-training Pruning Method for LLMs" prunes weights in linear layers under an MM0 semi-structured pattern, while input activation norms appear only in the RIA importance metric used to decide which weights to keep (Bazarbachi et al., 13 Aug 2025). The pruned object is therefore MM1, not a runtime-sparse activation tensor.

"Symmetric Pruning of LLMs" is similar in this respect. Its theoretical contribution is a saliency analysis for weight pruning, with activations MM2 serving as calibration signals in formulas such as

MM3

and, as a special case, the Wanda score

MM4

These are activation-aware weight-pruning criteria, not activation-pruning rules (Yi et al., 31 Jan 2025).

By contrast, SCAP and Amber Pruner both sparsify the activation tensor itself at inference time. SCAP does so through dynamic magnitude thresholding of FC inputs, while Amber Pruner performs structured top-MM5-within-MM6 pruning on the input activations of selected linear layers (Chua et al., 2024, An et al., 4 Aug 2025). A recurrent misconception is therefore to treat all activation-aware pruning methods as activation pruning methods; the recent literature sharply separates activation-aware weight scoring from direct activation masking.

3. Mask construction, scoring, and calibration

SCAP is the clearest example of post-training activation pruning without exact MM7 structure. Its runtime criterion is elementwise MM8 magnitude thresholding, with thresholds calibrated from a small calibration set. The framework also introduces Mode-Centering, which shifts activations by a scalar MM9 before thresholding: NN0

NN1

and then fuses NN2 into the bias: NN3 The stated purpose is to pre-calibrate activation distributions for maximizing post-training sparsity when the activation distribution is not already centered near zero (Chua et al., 2024).

Amber Pruner implements exact structured activation sparsity. It begins from naïve top-NN4 magnitude selection within each NN5-sized group, then replaces raw magnitude ranking with a weight-aware activation score

NN6

To make the channel multipliers more discriminative, it introduces Robust-Norm Scoring: first discard weight values outside the global NN7th–NN8th percentile range,

NN9

then standardize,

MNM-N0

and finally score activations as

MNM-N1

The weight-side channel multipliers are fixed per layer and can be precomputed offline, so the runtime procedure reduces to elementwise scoring followed by top-MNM-N2-within-MNM-N3 selection (An et al., 4 Aug 2025).

The abstract of (Alanova et al., 26 Sep 2025) places additional emphasis on lightweight, plug-and-play error mitigation techniques and pruning criteria, and states that these can establish strong hardware-friendly baselines that require minimal calibration. This suggests that post-training MNM-N4 activation pruning is not only a masking problem but also an error-control problem, especially when local structural constraints are imposed without retraining.

4. Selective deployment across layers and stages

The current literature does not treat all linear layers as equally prunable. Amber Pruner is explicitly designed for the prefill stage, where linear projection layers are highly compute-dense in LLM inference. The target modules include MNM-N5, MNM-N6, MNM-N7, MNM-N8, MNM-N9, 2 ⁣: ⁣42\!:\!40, and 2 ⁣: ⁣42\!:\!41, but the actual pruning policy is selective rather than uniform (An et al., 4 Aug 2025).

Its skip policy is defined by an end-to-end perturbation measure. For example, after pruning the input activation to the query projection, the relative perturbation is

2 ⁣: ⁣42\!:\!42

Using this criterion, the method reports that 2 ⁣: ⁣42\!:\!43 and 2 ⁣: ⁣42\!:\!44 are not worth pruning because with GQA they contribute less compute; 2 ⁣: ⁣42\!:\!45 and 2 ⁣: ⁣42\!:\!46 are sensitive and are preserved; 2 ⁣: ⁣42\!:\!47 is the least sensitive and is pruned in all layers; and 2 ⁣: ⁣42\!:\!48 and 2 ⁣: ⁣42\!:\!49 are selectively pruned layer by layer. The reported fraction of total linear computations accelerated is 4 ⁣: ⁣84\!:\!80 for LLaMA3.1-8B, 4 ⁣: ⁣84\!:\!81 for Qwen2-7B, and 4 ⁣: ⁣84\!:\!82 for Qwen3-30B-A3B (An et al., 4 Aug 2025).

SCAP addresses a broader architecture set but with a different abstraction. Rather than fixing an 4 ⁣: ⁣84\!:\!83 pattern, it generalizes sparsification by input activations of fully connected layers for application across recent Transformer Decoders, MoE, Mamba2, Encoding Transformer, and pre-quantized models. The common object remains the FC input activation tensor, but the masking rule is threshold-based rather than exact semi-structured selection (Chua et al., 2024).

5. Empirical behavior and sparsity-pattern tradeoffs

The strongest direct evidence for structured post-training activation pruning presently comes from Amber Pruner. Under the 4 ⁣: ⁣84\!:\!84 pattern, using Amber-P (all), the reported zero-shot average changes are from 4 ⁣: ⁣84\!:\!85 to 4 ⁣: ⁣84\!:\!86 on LLaMA3.1-8B, from 4 ⁣: ⁣84\!:\!87 to 4 ⁣: ⁣84\!:\!88 on Qwen2-7B, and from 4 ⁣: ⁣84\!:\!89 to 8 ⁣: ⁣168\!:\!160 on Qwen3-30B-A3B, corresponding to drops of 8 ⁣: ⁣168\!:\!161, 8 ⁣: ⁣168\!:\!162, and 8 ⁣: ⁣168\!:\!163, respectively. The same tables show that naïve top-8 ⁣: ⁣168\!:\!164 activation pruning loses substantially more: 8 ⁣: ⁣168\!:\!165, 8 ⁣: ⁣168\!:\!166, and 8 ⁣: ⁣168\!:\!167 at 8 ⁣: ⁣168\!:\!168. The paper further reports favorable generative and long-context behavior, including LLaMA3.1-8B GSM8K 8 ⁣: ⁣168\!:\!169 versus baseline Y=XW+b,Y = XW + b,0, and LongBench average Y=XW+b,Y = XW + b,1 versus Y=XW+b,Y = XW + b,2 under Y=XW+b,Y = XW + b,3 (An et al., 4 Aug 2025).

The abstract of (Alanova et al., 26 Sep 2025) places these results in a broader comparative frame. It states that pruning activations enables superior preservation of generative capabilities compared to weight pruning at equivalent sparsity levels, and that sparsity patterns beyond NVIDIA's standard Y=XW+b,Y = XW + b,4 are important: the Y=XW+b,Y = XW + b,5 pattern achieves performance nearly on par with unstructured sparsity, while the Y=XW+b,Y = XW + b,6 pattern is emphasized as a superior candidate when balancing flexibility against hardware implementation complexity. Within the limits of the available abstract, this identifies pattern selection as a first-order algorithmic variable rather than a purely hardware-driven formatting choice (Alanova et al., 26 Sep 2025).

SCAP provides an important unstructured baseline for interpreting these structured results. It reports robust Pareto efficiency compared to prior methods, translating to a Y=XW+b,Y = XW + b,7 additional LLM decoding speedup against CATS at iso model quality. On Mistral-7B, the geometric mean decoding speedup over dense is reported as Y=XW+b,Y = XW + b,8 for CATS and Y=XW+b,Y = XW + b,9 for SCAP, with both at XRN×ICX\in\mathbb{R}^{N\times IC}0 zero-shot task performance. This does not establish XRN×ICX\in\mathbb{R}^{N\times IC}1 behavior directly, but it shows that post-training activation sparsity can yield measurable wall-clock gains while preserving model quality (Chua et al., 2024).

6. Hardware implications, sparse kernels, and quantized deployment

The systems rationale for post-training activation pruning differs from that of weight pruning. In SCAP, the intended execution model is sparse FC or GEMV under batch size XRN×ICX\in\mathbb{R}^{N\times IC}2 decoding, where zero entries in the activation vector permit skipping corresponding weight channels and therefore reduce both memory traffic and multiply-accumulate work. The paper explicitly notes that decoding is memory-bandwidth-bound, that a single sparse vector is the sweet spot, and that larger batches or beam search reduce exploitable structure because masks from different vectors overlap less (Chua et al., 2024).

Amber Pruner addresses a different operating point: prefill rather than decode. It targets sparse-dense GEMM or SpMM in compute-heavy projection layers and is candid that current general-purpose hardware has limited support for fine-grained sparse activations. The paper therefore reports coverage of linear computations that can be sparsified, rather than claiming a fully realized wall-clock deployment path on commodity stacks. Experiments were run on XRN×ICX\in\mathbb{R}^{N\times IC}3 Ascend 910B processors, and the main practical caveat is the lack of optimized SpMM kernels for this setting (An et al., 4 Aug 2025).

The same work introduces Outstanding-sparse, which composes Amber Pruner with SmoothQuant-based W8A8 PTQ. Standard SmoothQuant scaling is written as

XRN×ICX\in\mathbb{R}^{N\times IC}4

while Outstanding-sparse flips this factor,

XRN×ICX\in\mathbb{R}^{N\times IC}5

to expand activation range rather than compress it; the reported choice is XRN×ICX\in\mathbb{R}^{N\times IC}6. The quantization setup uses XRN×ICX\in\mathbb{R}^{N\times IC}7 samples from BoolQ, with activations quantized per tensor and weights quantized per channel. At XRN×ICX\in\mathbb{R}^{N\times IC}8, the reported average change from the SQ-W8A8 baseline to O-sparse is from XRN×ICX\in\mathbb{R}^{N\times IC}9 to XX00 on LLaMA3.1-8B, from XX01 to XX02 on Qwen2-7B, and from XX03 to XX04 on Qwen3-30B-A3B (An et al., 4 Aug 2025).

7. Limitations, misconceptions, and open problems

The most immediate limitation is that direct work on post-training XX05 activation pruning remains narrow relative to the much larger literature on structured weight pruning. SCAP is directly about post-training activation pruning, but its mask is dynamic unstructured threshold sparsity rather than exact XX06; calibration controls expected global sparsity, not deterministic local structure (Chua et al., 2024). Amber Pruner is directly about training-free XX07 activation sparsity, but its current scope is mainly prefill, its realized gains are constrained by sparse kernel availability, and its strongest scoring variant is said to be not applicable to MoE because tokens are dynamically routed to experts (An et al., 4 Aug 2025).

A second limitation is conceptual confusion between direct activation pruning and activation-aware weight pruning. EGGS-PTP and Symmetric Pruning are highly relevant for scoring design, relative importance, and structured selection, but neither constructs token-dependent or runtime activation masks. In EGGS-PTP, activations enter the RIA metric used to rank weights, and the central structural idea is connectivity-aware XX08 weight masking inspired by expander graphs (Bazarbachi et al., 13 Aug 2025). In Symmetric Pruning, activations appear in reconstruction-based saliency formulas for deleting weights, and the paper explicitly evaluates XX09 and XX10 structured pruning only on the weight side (Yi et al., 31 Jan 2025).

Open problems follow directly from these gaps. The EGGS-PTP discussion identifies activation-pruning-relevant questions such as how to define an activation-side analogue of the bipartite expansion objective, whether dynamic XX11 activation masks can be generated cheaply enough at inference time, whether sparse activation kernels deliver the same practical gains as static sparse weights, and how activation pruning should be combined with weight XX12 pruning (Bazarbachi et al., 13 Aug 2025). The abstract of (Alanova et al., 26 Sep 2025) adds a complementary systems agenda by motivating future hardware to support more flexible sparsity patterns. A plausible implication is that the mature form of post-training XX13 activation pruning will depend not only on better saliency rules, but also on co-design between calibration, structured masking, sparse kernel support, and pattern-specific hardware execution.

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Post-Training N:M Activation Pruning.