Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size

About

Recent research on deep neural networks has focused primarily on improving accuracy. For a given accuracy level, it is typically possible to identify multiple DNN architectures that achieve that accuracy level. With equivalent accuracy, smaller DNN architectures offer at least three advantages: (1) Smaller DNNs require less communication across servers during distributed training. (2) Smaller DNNs require less bandwidth to export a new model from the cloud to an autonomous car. (3) Smaller DNNs are more feasible to deploy on FPGAs and other hardware with limited memory. To provide all of these advantages, we propose a small DNN architecture called SqueezeNet. SqueezeNet achieves AlexNet-level accuracy on ImageNet with 50x fewer parameters. Additionally, with model compression techniques we are able to compress SqueezeNet to less than 0.5MB (510x smaller than AlexNet). The SqueezeNet architecture is available for download here: https://github.com/DeepScale/SqueezeNet

Forrest N. Iandola, Song Han, Matthew W. Moskewicz, Khalid Ashraf, William J. Dally, Kurt Keutzer• 2016

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet-1k (val)
Top-1 Accuracy57.5
1469
Image ClassificationImageNet--
431
Image ClassificationImageNet ILSVRC-2012 (val)
Top-1 Accuracy57.5
405
Image ClassificationImageNet (test)--
235
Image ClassificationImageNet (test val)
Accuracy57.5
58
Aerial Image ClassificationAIDER v2 (test)
F1 Score0.845
41
Perceptual SimilarityBAPPS (val)
2AFC (Overall)68.6
39
Image ClassificationImageNet-1K
Top-1 Accuracy58.2
20
Aerial Image ClassificationAIDER (test)
F1 Score75.25
12
Image ClassificationLDD (test)
F1 Score89.34
12
Showing 10 of 22 rows

Other info

Code

Follow for update