PerturbPrune: Guiding Sparsity Allocation by Probing Layer Stability Along Principal Directions

Ningxin Wu, Zhengxu Hou, Ziran Zhao, Tao Jing, Pengyuan Liu, Chen Kang, Yangshijie Zhang, Luxi Xing, Bin-Bin Yang, Feng Xiao, Lu Liu, Dong Yu


Abstract
Large language models (LLMs) exhibit significant layer-wise redundancy, motivating non-uniform pruning for efficient compression. However, existing sparsity allocation methods rely on indirect statistical heuristics derived from weight magnitudes or activation distributions, without directly assessing how robustly each layer preserves semantic information under input variation. We propose PerturbPrune, a framework that allocates layer-wise sparsity by probing the intrinsic stability of hidden representations. PerturbPrune estimates the principal subspace of each layer’s activations via PCA and injects structured perturbations along these directions. Layer sensitivity is quantified through matrix difference entropy (DiME): stable layers are identified as encoding compact, essential features and thus preserved, while highly sensitive layers are deemed over-parameterized and pruned more aggressively. Extensive experiments across diverse LLM families (LLaMA, LLaMA-2, Qwen, OPT) demonstrate that PerturbPrune broadly matches or outperforms existing non-uniform methods. Compared to uniform pruning, it prevents catastrophic collapse, reducing perplexity by up to 99.7% and improving zero-shot accuracy by up to 22.7%. The method is also highly efficient, completing sparsity allocation on Qwen3.5-9B in 7 minutes—achieving 24× and 17.4× speedups over AlphaPruning and MRP. Finally, PerturbPrune generalizes to Mixture-of-Experts architectures, achieving up to 19.3% perplexity reduction on Qwen3.5-35B-A3B.
Anthology ID:
2026.findings-emnlp.1122
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
23753–23780
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1122/
DOI:
Bibkey:
Cite (ACL):
Ningxin Wu, Zhengxu Hou, Ziran Zhao, Tao Jing, Pengyuan Liu, Chen Kang, Yangshijie Zhang, Luxi Xing, Bin-Bin Yang, Feng Xiao, Lu Liu, and Dong Yu. 2026. PerturbPrune: Guiding Sparsity Allocation by Probing Layer Stability Along Principal Directions. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 23753–23780, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
PerturbPrune: Guiding Sparsity Allocation by Probing Layer Stability Along Principal Directions (Wu et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1122.pdf