BaCP: Backbone Contrastive Pruning for Preserving Representations in Extremely Sparse Neural Networks
Organizations: Lahore University of Management Sciences Lahore, Pakistan
Abstract
Unstructured pruning at extreme sparsity often suffers from representational collapse, causing sharp drops in accuracy. To address this, we study Backbone Contrastive Pruning (BaCP), which regularizes the sparse network's embedding space by aligning it with pretrained, fine-tuned, and historical snapshot models. Building on the contrastive decomposition of the CAP framework (Xu et al., 2022), we provide a rigorous matched-budget characterization of this approach across multiple pruning criteria. Evaluated across 90 settings, BaCP improves accuracy substantially in extreme sparsity regimes where standard pruning fails, and is close to baseline where representations remain intact.
Figures & tables
| ResNet-34 | ResNet-50 | VGG-11 | VGG-19 | |||||
| Sparsity | I.P. | I.P. | I.P. | I.P. | ||||
| Magnitude ( Han et al., 2015 ) | ||||||||
| 0.95 | 91.44 | +0.36 | 92.82 | +0.10 | 90.04 | +0.32 | 91.30 | +0.24 |
| 0.97 | 91.51 | +0.15 | 92.65 | 0.14 | 90.31 | 0.10 | 91.64 | +0.40 |
| 0.99 | 90.66 | +0.38 | 91.59 | +0.01 | 90.10 | 0.09 | 91.18 | +0.47 |
| 0.999 | 76.32 | +5.33 | 69.84 | +10.76 | 85.31 | 0.84 | 84.91 | +1.44 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Criterion | Sparsity | ||||
|---|---|---|---|---|---|---|
| 0.95 | 0.97 | 0.99 | 0.999 | 0.9995 | ||
| ResNet-34 | magnitude | |||||
| SNIP-it | ||||||
| WANDA | ||||||
| Model | Data | Criterion | I.P. | BaCP | surviving | ||
|---|---|---|---|---|---|---|---|
| ResNet-34 | C-10 | magnitude | 0.95 | 91.44 0.57 | 91.80 0.21 | ||
| ResNet-34 | C-10 | magnitude | 0.97 | 91.51 0.68 | 91.66 0.23 | ||
| ResNet-34 | C-10 | magnitude | 0.99 | 90.66 0.42 | 91.04 0.53 | ||
| ResNet-34 | C-10 | magnitude | 0.999 | 76.32 0.67 | 81.65 0.22 | ||
| ResNet-34 | C-10 | magnitude | 0.9995 | 61.09 1.06 | 75.11 0.14 | ||
| ResNet-34 | C-10 | SNIP-it | 0.95 | 91.26 0.65 | 91.60 0.14 |
| Model | Criterion | I.P. | CE-only | BaCP | aug LR | contrastive | regime | |
|---|---|---|---|---|---|---|---|---|
| MobileNetV2 | 0.99 | magnitude | 85.73 0.22 | 79.55 1.44 | 80.81 0.34 | intact | ||
| ResNet-34 | 0.95 | magnitude | 91.44 0.57 | 90.45 0.42 | 91.80 0.21 | intact | ||
| ResNet-34 | 0.999 | magnitude | 76.32 0.67 | 80.09 1.04 | 81.65 0.22 | damaged | ||
| ResNet-34 | 0.999 | SNIP-it | 79.92 1.12 | 77.67 0.77 | 79.41 0.17 | damaged | ||
| ResNet-34 | 0.999 | WANDA | 38.25 1.17 | 56.25 3.01 | 55.13 0.52 | damaged | ||
| ResNet-50 | 0.95 | magnitude | 92.82 0.42 | 91.55 0.00 | 92.92 0.32 | intact |
| Arm | Model | acc. | seeds | vs. control | |
|---|---|---|---|---|---|
| head128 | MobileNetV2 | 0.95 | 87.95 | 3 | control |
| head16 | MobileNetV2 | 0.95 | 87.08 | 1 | |
| head32 | MobileNetV2 | 0.95 | 87.56 | 1 | |
| head64 | MobileNetV2 | 0.95 | 87.74 | 1 | |
| headfrozen | MobileNetV2 | 0.95 | 87.74 | 1 | |
| headnone | MobileNetV2 | 0.95 | 87.95 | 1 |
| Arm | Model | acc. | vs. control | |
|---|---|---|---|---|
| snapeven2 | ResNet-34 | 0.97 | 92.32 | |
| snapeven3 | ResNet-34 | 0.97 | 92.17 | |
| snapeven5 | ResNet-34 | 0.97 | 92.03 | |
| snapfirst2 | ResNet-34 | 0.97 | 92.04 | |
| snapeven2 | ResNet-34 | 0.999 | 81.14 | |
| snapeven3 | ResNet-34 | 0.999 | 81.19 |
| Model | I.P. @ 0.01 | BaCP @ 0.05 | BaCP @ 0.1 | 0.1 vs 0.05 | |
|---|---|---|---|---|---|
| VGG-11 | 0.95 | 90.04 0.15 | 90.36 0.02 | 87.29 1.92 | |
| VGG-11 | 0.999 | 85.31 0.12 | 84.47 0.26 | 83.05 2.00 |
| Model | Data | Criterion | I.P. | BaCP | |
|---|---|---|---|---|---|
| MobileNetV2 | C-10 | magnitude | 0.999 | 10.00 | 10.00 |
| MobileNetV2 | C-10 | magnitude | 0.9995 | 10.00 | 10.00 |
| MobileNetV2 | C-10 | SNIP-it | 0.999 | 10.00 | 24.05 |
| MobileNetV2 | C-10 | SNIP-it | 0.9995 | 10.00 | 10.00 |
| MobileNetV2 | C-10 | WANDA | 0.99 | 10.00 | 10.00 |
| MobileNetV2 | C-10 | WANDA | 0.999 | 10.00 | 10.00 |
| I.P. | BaCP | |
| Initialisation | ImageNet-pretrained, identical checkpoint | |
| Dense reference | identical, per seed | |
| Criterion | identical (magnitude / SNIP-it / WANDA, global threshold) | |
| Sparsity schedule | cubic, ramp over 80% of steps, | |
| Prunable scope | identical; classifier head pruned, projection head never | |
| Pruning epochs | 50 | 50 |
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 292 papers