Other networks

December 7, 2021 ยท View on GitHub


Catalogue

1. Overview

In 2012, AlexNet network proposed by Alex et al. won the ImageNet competition by far surpassing the second place, and the convolutional neural network and even deep learning attracted wide attention. AlexNet used relu as the activation function of CNN to solve the gradient dispersion problem of sigmoid when the network is deep. During the training, Dropout was used to randomly lose a part of the neurons, avoiding the overfitting of the model. In the network, overlapping maximum pooling is used to replace the average pooling commonly used in CNN, which avoids the fuzzy effect of average pooling and improves the feature richness. In a sense, AlexNet has exploded the research and application of neural networks.

SqueezeNet achieved the same precision as AlexNet on Imagenet-1k, but only with 1/50 parameters. The core of the network is the Fire module, which used the convolution of 1x1 to achieve channel dimensionality reduction, thus greatly saving the number of parameters. The author created SqueezeNet by stacking a large number of Fire modules.

VGG is a convolutional neural network developed by researchers at Oxford University's Visual Geometry Group and DeepMind. The network explores the relationship between the depth of the convolutional neural network and its performance. By repeatedly stacking the small convolutional kernel of 3x3 and the maximum pooling layer of 2x2, the multi-layer convolutional neural network is successfully constructed and has achieved good convergence accuracy. In the end, VGG won the runner-up of ILSVRC 2014 classification and the champion of positioning.

DarkNet53 is designed for object detection by YOLO author in the paper. The network is basically composed of 1x1 and 3x3 kernel, with a total of 53 layers, named DarkNet53.

2. Accuracy, FLOPs and Parameters

ModelsTop1Top5Reference
top1
Reference
top5
FLOPs
(G)
Parameters
(M)
AlexNet0.5670.7920.57201.37061.090
SqueezeNet1_00.5960.8170.5751.5501.240
SqueezeNet1_10.6010.8190.6901.230
VGG110.6930.89115.090132.850
VGG130.7000.89422.480133.030
VGG160.7200.9070.7150.90130.810138.340
VGG190.7260.90939.130143.650
DarkNet530.7800.9410.7720.93818.58041.600

3. Inference speed based on V100 GPU

ModelsCrop SizeResize Short SizeFP32
Batch Size=1
(ms)
AlexNet2242561.176
SqueezeNet1_02242560.860
SqueezeNet1_12242560.763
VGG112242561.867
VGG132242562.148
VGG162242562.616
VGG192242563.076
DarkNet532562563.139

4. Inference speed based on T4 GPU

ModelsCrop SizeResize Short SizeFP16
Batch Size=1
(ms)
FP16
Batch Size=4
(ms)
FP16
Batch Size=8
(ms)
FP32
Batch Size=1
(ms)
FP32
Batch Size=4
(ms)
FP32
Batch Size=8
(ms)
AlexNet2242561.064471.704352.384021.449932.466963.72085
SqueezeNet1_02242560.971622.067193.674990.967362.532214.54047
SqueezeNet1_12242560.813781.629192.680440.760321.8773.15298
VGG112242562.244084.677947.65683.904129.5114717.14168
VGG132242562.585895.8270810.035914.6468412.6155823.70015
VGG162242563.132377.1925712.509135.6176916.4006432.03939
VGG192242563.699878.5916815.078666.6522120.433441.55902
DarkNet532562563.181015.8841910.149644.1082912.171422.15266