Compression of Convolutional Neural Networks With Divergent Representation of Filters. 2022

Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang

Convolutional neural networks (CNNs) have made remarkable achievements in many tasks. However, most of them are hardly applied to embedded systems directly because of the requirement of huge memory space and computing power. In this article, we propose a pruning framework, namely, FiltDivNet, to accelerate and compress CNN models for their applicability to small or portable devices. The correlations among filters are taken into account and measured by the goodness of fit. On this basis, a hybrid-cluster pruning strategy is designed with dynamic pruning ratios for different clusters in CNN models. It aims at representing its filters in their diversity by removing redundant ones cluster by cluster. In addition, a new loss function with adaptive sparsity constraints is introduced for the retraining and fine-tuning in the FiltDivNet. Finally, some comparative experiments based on classical CNN models are carried out to demonstrate its effectiveness in compression performance and its adaptability with different CNN architectures.

UI MeSH Term Description Entries

Related Publications

Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
July 2022, IEEE transactions on pattern analysis and machine intelligence,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
September 2018, Neural networks : the official journal of the International Neural Network Society,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
July 2020, Biomedical optics express,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
October 2017, IEEE transactions on cybernetics,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
December 2019, PLoS computational biology,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
January 2021, Frontiers in big data,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
March 2022, Sensors (Basel, Switzerland),
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
February 2023, Abdominal radiology (New York),
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
April 2016, Computer methods and programs in biomedicine,
Peng Lei, and Jiawei Liang, and Tong Zheng, and Jun Wang
May 2021, Sensors (Basel, Switzerland),
Copied contents to your clipboard!