Convolutional neural network pruning optimization method and device, electronic equipment and storage medium

• Main Authors: HE WEI, DONG YI, XIE XIAODAN, ZHAI JIA, CHEN FENG
• Format: Patent
• Language: Chinese; English
• Published: 12.04.2022
• Subjects: CALCULATING; COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS; COMPUTING; COUNTING; HANDLING RECORD CARRIERS; PHYSICS; PRESENTATION OF DATA; RECOGNITION OF DATA; RECORD CARRIERS

The invention provides a convolutional neural network pruning optimization method and apparatus, an electronic device and a storage medium. The method comprises the steps of obtaining a trained convolutional neural network model; for each category of the convolutional neural network model, determining a corresponding semantic information graph, and determining a filter importance factor of each filter in each category based on the semantic information graph; according to the importance factor of the filter and the pruning target, carrying out importance degree sorting on the filter; on the basis of the sorting result and a pruning target, the filters with the small importance degree are cut off step by step until the pruning target is achieved, and a convolutional neural network model after pruning optimization is obtained; and carrying out retraining on the convolutional neural network model after pruning optimization. According to the method, targeted convolutional neural network pruning compression with a