Research Article

Small Network for Lightweight Task in Computer Vision: A Pruning Method Based on Feature Representation

Table 1

Detail of experiments.

No.Original networkPretrainingRefiningSuperparameterPruning method

1Cifar10_fullCifar10Cifar10160 epoch 0.001lrMap-EX, Size-EX, scratch-EX, Map-NEX, Size-NEX, and scratch-NEX
Modified Cifar10
Modified Cifar10 with different data number
Cifar10160 epoch 0.001lrMultisample pruning with different ratio

2VGG16Cifar10Cifar10160 epoch 0.001lrMap-NEX, multisample pruning (0.8), and network slimming
ImageNet subdataset

3YoloV3ImageNetVOC80 epoch 0.001lr; 80 epoch 0.001lrMap-NEX; Size-NEX