Explainable Structured Pruning of BERT via Mutual Information
Hanjuan Huang1, Hao-Jia Song2, Qiling Zhao3
1College of Mechanical and Electrical Engineering, Wuyi University, Wuyishan 354300, China.
None:
Bidirectional Encoder Representations from Transformers (BERT) excels in natural language processing (NLP) but is costly on edge devices. We introduce an unsupervised, retraining-free structured pruning scheme for BERT, guided by mutual information (MI). Leveraging Rényi α-order entropy, we design a representation-aware MI estimator and a principled kernel-bandwidth selection, producing stable, sample-efficient neuron-level pruning signals. This method removes redundant units while preserving representational capacity, reduces memory and latency, and deploys readily on commodity hardware. Explainable-AI visualizations clarify how compression reshapes intermediate features and predictions. Across benchmarks, the compressed models maintain minimal accuracy loss, outperform or match strong unsupervised baselines, and remain competitive with supervised alternatives.
Related Concept Videos
Survival Tree
Building a Survival Tree
Constructing a...
Quantifying and Rejecting Outliers: The Grubbs Test
Woodward–Hoffmann Selection Rules and Microscopic Reversibility
Generalization, Discrimination, and Extinction
Generalization occurs when a behavior reinforced in one context is performed in similar situations. For instance, a student who studies diligently for calculus and receives excellent grades might apply the same study habits to psychology and history, expecting similar results. Generalization shows how learning in one setting can influence behavior in...
