Structured pruning for group regularized convolutional neural networks via dynamic regularization factor

High demand for computation and storage largely hinders the deployment of deep convolutional neural networks (CNNs) in resource constrained devices. Regularization-based pruning methods are effective for the model compression of CNNs by introducing a regularization term into the loss function. However, such methods typically compress weights to zero utilising a fixed regularization factor, which cannot impose different punishments on weights or structures with different importance. In this paper, a group regularization method is proposed to impose a dynamic regularization factor to each filter of convolutional layers, named GRDRF. The scaling factors of batch normalization (BN) layer are used to represent the importance of filters and construct updatable and dynamic regularization factors according to the importance representation. Moreover, a novel pruning strategy of ResNet is proposed, which can significantly improve the pruning performance. Numerical experiments on the popular CNN architectures and CIFAR-10 dataset are performed to verify the efficiency of the new group regularization method and the ResNet pruning strategy. The programs are publicly provided on https://github.com/xiaoli7766111/DRF.

Authors

Institutions

Publication Details

Journal
Journal of Industrial and Management Optimization
Published
2024-09-27
DOI
https://doi.org/10.3934/jimo.2024131
Citations
1
Primary Topic
Machine Learning and ELM
Type
article
Field-Weighted Citation Impact
0.35
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Structured pruning for group regularized convolutional neural networks via dynamic regularization factor

Jinlong Yuan, Junchi Ma, Bo Li, Feng Li et al.
1 citations
Journal of Industrial and Management Optimization
Machine Learning and ELM
0.35
article

Structured pruning for group regularized convolutional neural networks via dynamic regularization factor

Jinlong Yuan, Junchi Ma, Bo Li, Feng Li, Meijiao Zhu
article en
1 citations

Abstract

High demand for computation and storage largely hinders the deployment of deep convolutional neural networks (CNNs) in resource constrained devices. Regularization-based pruning methods are effective for the model compression of CNNs by introducing a regularization term into the loss function. However, such methods typically compress weights to zero utilising a fixed regularization factor, which cannot impose different punishments on weights or structures with different importance. In this paper, a group regularization method is proposed to impose a dynamic regularization factor to each filter of convolutional layers, named GRDRF. The scaling factors of batch normalization (BN) layer are used to represent the importance of filters and construct updatable and dynamic regularization factors according to the importance representation. Moreover, a novel pruning strategy of ResNet is proposed, which can significantly improve the pruning performance. Numerical experiments on the popular CNN architectures and CIFAR-10 dataset are performed to verify the efficiency of the new group regularization method and the ResNet pruning strategy. The programs are publicly provided on https://github.com/xiaoli7766111/DRF.

Journal of Industrial and Management OptimizationVol. 21(2)
Liaoning Normal University (CN), Dalian Maritime University (CN)
Openalex Percentile: Top 33%
Machine Learning and ELM
0.35
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.