A Hierarchical Importance-Guided Multi-objective Evolutionary Framework for Deep Neural Network Pruning
Organizations: Department of Computer Science and Physics, Wilfrid Laurier University, Waterloo, N2L 3C5, Ontario, Canada
Abstract
The optimization of over-parameterized deep neural networks represents a large-scale, high-dimensional, and strongly non-convex decision problem that challenges existing optimization frameworks. Current evolutionary and gradient-based pruning methods often struggle to scale to such dimensionalities, as they rely on flat search spaces, scalarized objectives, or repeated retraining, leading to premature convergence and prohibitive computational cost. This paper introduces a hierarchical importance-guided evolutionary framework that reformulates convolutional network pruning as a tractable large-scale multi-objective optimization problem. In the first phase, a continuous evolutionary search performs coarse exploration of weight-wise pruning thresholds to shrink the search space and identify promising regions of the Pareto set. The second phase applies a fine-grained binary evolutionary optimization constrained to the surviving weights, where importance-aware sampling and adaptive variation operators refine local search in the sparse region of the Pareto set. This hierarchical design combines global exploration and localized exploitation to achieve a well-distributed Pareto set of networks balancing compactness and accuracy. Empirical results on CIFAR-10 and CIFAR-100 using ResNet-56 and ResNet-110 confirm the method's effectiveness compared to existing evolutionary approaches: pruning achieves up to 51.9% and 38.9% parameter reductions with almost no accuracy loss compared to state-of-the-art evolutionary DNN pruning methods. The proposed method contributes a scalable evolutionary approach for solving very-large-scale multi-objective optimization problems, offering a general paradigm extendable to other domains where the decision space is exponentially large, objective functions are conflicting, and efficient trade-off discovery is essential.
Figures & tables
| Dataset | Subset | Size | Purpose | Usage Phase |
|---|---|---|---|---|
| CIFAR-10 | Training Set | 47,500 | Model fine-tuning | Pre-pruning |
| Validation Set | 1,000 | Fitness evaluation | Phase 1 & 2 | |
| Optimization Subset | 500 | Fast EA evaluation | Phase 1 & 2 | |
| Test Set | 10,000 | Final model evaluation | Post-optimization | |
| CIFAR-100 | Training Set | 47,000 | Model fine-tuning | Pre-pruning |
| Validation Set | 2,000 | Fitness evaluation | Phase 1 & 2 |
| Model | Depth | Params (M) | Variant |
|---|---|---|---|
| ResNet-18 | 18 | 11.18 | ImageNet |
| ResNet-50 | 50 | 23.53 | ImageNet |
| ResNet-56 | 56 | 0.855 | CIFAR |
| ResNet-101 | 101 | 42.52 | ImageNet |
| ResNet-110 | 110 | 1.731 | CIFAR |
| ResNet-152 | 152 | 58.16 | ImageNet |
| Parameter | Setting |
|---|---|
| Optimizer | Adam |
| Learning rate | |
| Loss function | Cross-Entropy Loss |
| Epochs | 30 |
| Batch size | 32 |
| Scheduler | None (constant learning rate) |
| Parameter | Setting |
|---|---|
| Algorithm | NSGA-II |
| Population size / Generations | 50 / 50 |
| Sampling | Latin Hypercube Sampling (LHS) |
| Crossover | SBX ( , ) |
| Mutation | Polynomial ( , ) |
| Search space | Continuous |
| Parameter | Setting |
|---|---|
| Algorithm | NSGA-II |
| Population size / Generations | 50 / 50 |
| Sampling | Importance-Guided Probabilistic Sampling |
| Crossover | Uniform ( ) |
| Mutation | Bit-flip ( ) |
| Search space | Binary pruning masks |
| Architecture | Test Error CIFAR-10 | Test Error CIFAR-100 |
|---|---|---|
| ResNet18 | 6.32 | 23.74 |
| ResNet50 | 6.03 | 22.08 |
| ResNet56 | 7.33 | 31.88 |
| ResNet101 | 5.34 | 20.76 |
| ResNet110 | 6.82 | 36.62 |
| ResNet152 | 5.17 | 20.56 |
| Metric | ResNet18 | ResNet50 | ResNet56 | ResNet101 | ResNet110 | ResNet152 |
|---|---|---|---|---|---|---|
| Baseline (M) | ||||||
| Baseline Weights (M) | 11.18 | 23.53 | 0.855 | 42.52 | 1.731 | 58.16 |
| Baseline Accuracy (%) | 93.68 | 93.97 | 92.67 | 94.66 | 93.18 | 94.83 |
| Phase 1 Results | ||||||
| Best Solution-Non-Zero Weights (M) | 6.970 | 12.02 | 0.506 | 17.14 | 1.575 | 22.62 |
| Best Solution Accuracy (%) | 93.14 | 93.49 | 92.18 | 93.64 | 92.85 | 93.35 |
| Metric | ResNet18 | ResNet50 | ResNet56 | ResNet101 | ResNet110 | ResNet152 |
|---|---|---|---|---|---|---|
| Baseline (M) | ||||||
| Baseline Weights (M) | 11.23 | 23.71 | 0.862 | 42.70 | 1.731 | 58.35 |
| Baseline Accuracy (%) | 76.26 | 77.92 | 68.12 | 79.24 | 63.38 | 79.44 |
| Phase 1 Results | ||||||
| Best Solution-Non-Zero Weights (M) | 7.810 | 8.145 | 0.819 | 18.06 | 1.764 | 24.32 |
| Best Solution Accuracy (%) | 76.11 | 77.31 | 67.94 | 79.08 | 63.02 | 78.98 |
| Method | CNN architecture | Baseline Error (%) | Error after pruning (%) | Difference (%) | Pruned (%) |
|---|---|---|---|---|---|
| Greedy-Filter Li et al. (2016) | ResNet56 | 6.96 | 6.94 | -0.02 | 27.67 |
| ResNet110 | 6.47 | 6.70 | -0.23 | 38.63 | |
| Auto-Balanced Filter Ding et al. (2018) | ResNet56 | 6.96 | 8.56 | 1.60 | 41.51 |
| ResNet110 | – | 8.27 | – | 40.00 | |
| DeepPruningES Fernandes Junior and Yen (2020) | ResNet56 | 6.63 | 8.12 | 1.49 | 21.30 |
| ResNet110 | 6.20 | 7.42 | 1.22 | 16.74 |
| Method | CNN architecture | Baseline Error (%) | Error after pruning (%) | Difference (%) | Pruned (%) |
|---|---|---|---|---|---|
| DeepPruningES Fernandes Junior and Yen (2020) | ResNet56 | 41.94 | 40.20 | -1.74 | 16.18 |
| ResNet110 | 50.17 | 50.98 | 0.81 | 17.78 | |
| Bi-CNN-Pruning Louati et al. (2024) | ResNet56 | 41.86 | 24.83 | -17.03 | 16.38 |
| ResNet110 | 50.47 | 38.44 | -12.03 | 18.13 | |
| Two-Phase Pruning | ResNet56 | 31.88 | 32.84 | 0.86 | 22.13 |
| ResNet110 | 36.62 | 37.89 | 1.27 | 20.62 |
Neural Network Pruning
Neural network pruning removes weights, neurons, channels, heads, or layers to reduce model size and computation while retaining useful behavior.
See all 278 papers