Get our free extension to see links to code for papers anywhere online!

 Add to Chrome

 Add to Firefox

CatalyzeX Code Finder - Browser extension linking code for ML papers across the web! | Product Hunt Embed

Three Dimensional Convolutional Neural Network Pruning with Regularization-Based Method

Nov 19, 2018
Yuxin Zhang, Huan Wang, Yang Luo, Roland Hu

In recent years, three-dimensional convolutional neural network (3D CNN) is intensively applied in video analysis and receives good performance. However, 3D CNN leads to massive computation and storage consumption, which hinders its deployment on mobile and embedded devices. In this paper, we propose a three-dimensional regularization-based pruning method to assign different regularization parameters to different weight groups based on their importance to the network. Experiments show that the proposed method outperforms other popular methods in this area.

* Accepted by NIPS 2018 Workshop CDNNRIA 

Share this with someone who'll enjoy it:

   Access Paper Source

Share this with someone who'll enjoy it: