Architecture-aware Network Pruning for Vision Quality Applications

08/05/2019
by   Wei-Ting Wang, et al.
6

Convolutional neural network (CNN) delivers impressive achievements in computer vision and machine learning field. However, CNN incurs high computational complexity, especially for vision quality applications because of large image resolution. In this paper, we propose an iterative architecture-aware pruning algorithm with adaptive magnitude threshold while cooperating with quality-metric measurement simultaneously. We show the performance improvement applied on vision quality applications and provide comprehensive analysis with flexible pruning configuration. With the proposed method, the Multiply-Accumulate (MAC) of state-of-the-art low-light imaging (SID) and super-resolution (EDSR) are reduced by 58 drop, respectively. The memory bandwidth (BW) requirements of convolutional layer can be also reduced by 20

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset