1. DDFP:A data driven filter pruning method with pruning compensation.
- Author
-
Li, Guoqiang, Liu, Bowen, and Chen, Anbang
- Subjects
- *
ARTIFICIAL neural networks , *SIGNAL convolution , *ROBUST control , *DEEP learning , *ARTIFICIAL intelligence - Abstract
Neural network pruning techniques can be effective in accelerating neural network models, making it possible to deploy them on edge devices. In this paper, we propose to prune neural networks using data variance. Unlike other existing methods, this method is somewhat robust and does not invalidate our criteria depending on the number of data batches and the number of training sessions. We also propose a pruning compensation technique. This technique fuses the pruned convolutional information into the remaining convolutional kernel close to it. This fusion operation can effectively help retain the pruned information. We evaluate the proposed method on a number of standard datasets and compare it with several current state-of-the-art methods. Our method always achieves better performance. For example, on Tiny ImageNet, our method can prune 54.2% FLOPs of ResNet50 while obtaining a 0.22% accuracy improvement. • A data-driven network pruning method with high robustness is proposed. • The proposed data-driven pruning method consumes less resources and is highly efficient. • A pruning compensation technique is proposed to reduce the negative impact of pruning. • The effectiveness of the proposed method is validated on several datasets. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF