| 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. | ||