Xilinx, Inc.
METHOD OF PRUNING CONVOLUTIONAL NEURAL NETWORK BASED ON FEATURE MAP VARIATION

Abstract:

Provided in the present disclosure is a method of pruning a convolutional neural network based on feature map variation. The present invention enables compression of an entire network by means of removing a portion of filters in a convolutional layer, and such process is called pruning. A main contribution of the present invention is determining a pruning rule for filters in a single convolutional layer according to a feature map variation condition, using the rule to analyze network sensitivity, and pruning the entire network according to the network sensitivity.