arrow
返回

A Dynamic Pruning Method on Multiple Sparse Structures in Deep Neural Networks

delete2023-01-01
delete6
delete
OA
AI
J
Jie Hu
P
Peng Lin
H
Huajun Zhang
Z
Zining Lan
W
Wenxin Chen
K
Kailiang Xie
陈思耘 封面图
陈思耘 (Siyun Chen)
H
Hao Wang
常
常胜 (Sheng Chang) *
DOI:10.1109/ACCESS.2023.3267469delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
It is well known that significant computational power and a large amount of memory are required for deep neural networks, which makes them difficult to apply in resource-limited environments. So, many network compression and acceleration technologies have emerged, of which connection pruning is widely applied due to its effectiveness and convenience. A novel connection pruning method for full model capacity on multiple sparse structures is proposed in this paper. We design a simple and efficient function called Dynamic Processing Unit (DPU) for handling the evaluated weights. Our method has the following features: 1) Instead of being pruned directly or set to 0, the weights are controlled by the DPU to determine whether they will be used during subsequent forward passes of the network during the iteration of pruning training. 2) It supports the traditional multi-steps prune method as well as the end-to-end training mode that can get a compressed network in a single stage by fusing training and pruning. 3) It can learn multiple useful sparse structures, including, but not limited to, depth-wise, filter-wise, channel-wise, 2D-filter-wise, row-wise, column-wise, connection-wise and mixed sparse structures. Our method is tested on various widely-used datasets and models, such as the LeNet and the ResNet on MNIST and CIFAR-10. Importantly, it demonstrates good performance in all these cases.
Keyword:
Convolutional neural networks
dynamic channel pruning
network compression and acceleration
structured pruning

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

W
wuhan university
学者数:
8.1W
论文数: 5.8W
被引数: 70
引用论文

引用论文

Dynamical Channel Pruning by Conditional Accuracy Change for Deep Neural Networks
err2021-02-01
err52
PREAI
errChen, Zhiqiang; Xu, Ting-Bing; Du, Changde; Liu, Cheng-Lin; He, Huiguang
err分享
err收藏
A new approach to the neurological assessment of the preterm and full-term newborn infant
err1980-01-01
err0
PREAI
errLilly MS Dubowitz; Victor Dubowitz; Penelope Palmer; M Verghote
err分享
err收藏
err分享
err收藏
Asymptotic Soft Filter Pruning for Deep Convolutional Neural Networks
err2020-08-01
err135
errOAAI
errHe, Yang; Dong, Xuanyi; Kang, Guoliang; Fu, Yanwei; Yan, Chenggang; Yang, Yi
err分享
err收藏
学者 查看更多内容