基于滤波器裁剪的卷积神经网络加速算法

李浩,赵文杰,韩波

Convolutional neural network acceleration algorithm based on filters pruning

Hao LI,Wen-jie ZHAO,Bo HAN

表 1 裁剪CIFAR-10数据集上的预训练模型VGG-16

Tab.1 Pruning of pre-training model VGG-16 on CIFAR-10 dataset

卷积层 Fb/106 Fa/106 ΔF/% conv 1_1 1.8 0.9 50 conv 1_2 38 19 50 conv 2_1 19 14.25 25 conv 2_2 38 28.5 25 conv 3_1 19 14.25 25 conv 3_2 38 28.5 25 conv 3_3 38 38 0 conv 4_1 19 9.5 50 conv 4_2 38 9.5 75 conv 4_3 38 9.5 75 conv 5_1 9.4 2.35 75 conv 5_2 9.4 2.35 75 conv 5_3 9.4 2.35 75 FC6 0.26 0.13 50 FC7 5.1×10−3 5.1×10−3 0 总量 315 179 43.2