ABSTRACT Demands to deploy deep neural network (DNN) models on mobile devices and embedded systems have drastically grown in recent years. When transplanting DNN models to such platforms, requirements pertaining to computation and memory use are bottlenecks. To overcome them, network pruning has been carefully studied as a method of network compression. The effectiveness of network pruning is significantly affected by incorrect pruning on important connections. In this paper, we propose a network pruning method based on the modified L 1/2 penalty that reduces incorrect pruning by increasing the sparsity of the pretrained models. The modified L 1/2 penalty yields better sparsity than the L 1 penalty at a similar computational cost. Compared with the past work that numerically defines the importance of connections and re-establishes important weights when incorrect pruning occurs, our method achieves faster convergence by using a simpler pruning strategy. The results of experiments show that our method can compress LeNet300-100, LeNet-5, ResNet, AlexNet, and VGG16 by factors of 66×, 322×, 26×, 21×, and 16×, respectively, with negligible loss of accuracy. INDEX TERMS Deep learning, loss function, neural network, network pruning.
REF proposed a weight-based network pruning approach using the modified l 1/2 penalty to increase the sparsity of the pre-trained models.
Prune Deep Neural Networks With the Modified $L_{1/2}$ Penalty
IEEE Access
IEEE Access
Computer Science