arXiv · 2209.05683
One-shot Network Pruning at Initialization with Discriminative Image Patches
Abstract
One-shot Network Pruning at Initialization (OPaI) is an effective method to decrease network pruning costs. Recently, there is a growing belief that data is unnecessary in OPaI. However, we obtain an opposite conclusion by ablation experiments in two representative OPaI methods, SNIP and GraSP. Specifically, we find that informative data is crucial to enhancing pruning performance. In this paper, we propose two novel methods, Discriminative One-shot Network Pruning (DOP) and Super Stitching, to prune the network by high-level visual discriminative image patches. Our contributions are as follows. (1) Extensive experiments reveal that OPaI is data-dependent. (2) Super Stitching performs significantly better than the original OPaI method on benchmark ImageNet, especially in a highly compressed model.
Explore related subjects
Keep this discovery
Yinan Yang, Yu Wang, Ying Ji, Heng Qi, Jien Kato. 2022-09-13. One-shot Network Pruning at Initialization with Discriminative Image Patches. https://arxiv.org/abs/2209.05683
Cite the original work for its findings. Save a collection to share your selection of sources.