Explainable Structured Pruning of BERT via Mutual Information
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.
Authors
- Qiling Zhao
- Hanjuan Huang (ORCID: https://orcid.org/0009-0006-4691-8062)
- Hao-Jia Song (ORCID: https://orcid.org/0009-0000-2561-2423)
Institutions
- National Taiwan University of Science and Technology (TW)
- Wuyi University (CN)
- Wuyi University (CN)
Publication Details
- Journal
- Entropy
- Published
- 2025-12-02
- DOI
- https://doi.org/10.3390/e27121224
- Citations
- 1
- Primary Topic
- Explainable Artificial Intelligence (XAI)
- Type
- article
- Field-Weighted Citation Impact
- 2.48
Funders
- Wuyi University
- National Science and Technology Council