Paper ID | SS-NNC.8 | ||
Paper Title | HYBRID PRUNING AND SPARSIFICATION | ||
Authors | Hamed Rezazadegan Tavakoli, Nokia Technologies, Finland; Joachim Wabnig, Nokia Bell Labs, Finland; Francesco Cricri, Honglei Zhang, Emre Aksu, Nokia Technologies, Finland; Iraj Saniee, Nokia Bell Labs, Finland | ||
Session | SS-NNC: Special Session: Neural Network Compression and Compact Deep Features | ||
Location | Area B | ||
Session Time: | Tuesday, 21 September, 08:00 - 09:30 | ||
Presentation Time: | Tuesday, 21 September, 08:00 - 09:30 | ||
Presentation | Poster | ||
Topic | Special Sessions: Neural Network Compression and Compact Deep Features: From Methods to Standards | ||
IEEE Xplore Open Preview | Click here to view in IEEE Xplore | ||
Abstract | A hybrid approach based on the combination of saliency-based neural pruning and regularization-based sparsification is proposed. We propose using a graph diffusion process for determining the neuron importance for pruning. Then, we use a regularization loss based on weighted L1-norm and L2-norm during fine-tuning to recover the lost performance. This is followed by a threshold step to further impose sparsification. We demonstrate such a hybrid approach achieves significantly better performance in comparison to purely regularization-based sparsification for large neural networks. To this end, we assessed our proposed method on three tasks, including: image classification (3 network architectures), audio classification and image compression. |