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

Institutions

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

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Explainable Structured Pruning of BERT via Mutual Information

Qiling Zhao, Hanjuan Huang, Hao-Jia Song
1 citations
Entropy
Explainable Artificial Intelligence (XAI)
2.48
article

Explainable Structured Pruning of BERT via Mutual Information

Qiling Zhao, Hanjuan Huang, Hao-Jia Song
article en
1 citations

Abstract

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.

EntropyVol. 27(12)
National Taiwan University of Science and Technology (TW), Wuyi University (CN), Wuyi University (CN)
Wuyi University, National Science and Technology Council
Openalex Percentile: Top 8%
Explainable Artificial Intelligence (XAI)
2.48
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.