ExplainabilityTransparencyAccountabilitycs.LG
Pruning by Explaining: A Novel Criterion for Deep Neural Network Pruning
Seul-Ki Yeom, Philipp Seegerer, Sebastian Lapuschkin, Alexander Binder, Simon Wiedemann, Klaus-Robert Müller, Wojciech Samek
December 18, 2019
Pattern Recognition
280 citations
17 influential
Key Takeaways
-
Interpretability can guide model compression: The paper demonstrates that using explainability techniques to identify which parts of neural networks are most important for their decisions can also help determine which components are safe to remove, creating a direct link between understanding AI systems and making them more efficient.
-
Pruned models remain trustworthy: By removing only the least relevant units identified through explainability methods, the approach maintains or even improves accuracy while reducing model size, suggesting that interpretability-guided pruning preserves the model's decision-making quality rather than degrading it.
-
Efficiency gains without extensive retraining: The method performs particularly well in resource-constrained scenarios where organizations have limited data or computational budget for fine-tuning, achieving model compression "while maintaining or even improving accuracy" without requiring extensive retraining—making responsible AI practices more accessible to organizations with limited resources.
-
Computational efficiency of the approach: The authors note their method has "computational cost in the order of gradient computation and is comparatively simple to apply," meaning organizations can implement this interpretability-based pruning without requiring specialized expertise or excessive computational overhead.
-
Broader applicability across vision tasks: The method was "evaluated on a broad range of computer vision datasets," suggesting the approach generalizes beyond single use cases and could be adopted across different applications where model efficiency and interpretability are both priorities.