4 个月前

SqueezeNet:参数量减少50倍且模型大小小于0.5MB的AlexNet级精度

SqueezeNet:参数量减少50倍且模型大小小于0.5MB的AlexNet级精度

摘要

近期关于深度神经网络的研究主要集中在提高准确性上。对于给定的准确率水平,通常可以找到多种能够达到该准确率的DNN架构。在相同准确率的情况下,较小的DNN架构至少具有三个优势:(1)较小的DNN在分布式训练过程中需要较少的服务器间通信。(2)较小的DNN在从云端向自动驾驶汽车传输新模型时需要较少的带宽。(3)较小的DNN更容易部署在FPGA和其他内存有限的硬件上。为了提供这些优势,我们提出了一种称为SqueezeNet的小型DNN架构。SqueezeNet在ImageNet数据集上的准确率达到了AlexNet水平,但参数数量减少了50倍。此外,通过使用模型压缩技术,我们能够将SqueezeNet压缩到小于0.5MB(比AlexNet小510倍)。SqueezeNet架构可在此处下载:https://github.com/DeepScale/SqueezeNet

代码仓库

jiweibo/imagenet
pytorch
GitHub 中提及
avoroshilov/tf-squeezenet
tf
GitHub 中提及
Mind23-2/MindCode-80
mindspore
GitHub 中提及
zjZSTU/LightWeightCNN
pytorch
GitHub 中提及
rcmalli/keras-squeezenet
tf
GitHub 中提及
xin-w8023/SqueezeNet-PyTorch
pytorch
GitHub 中提及
mtmd/Mobile_ConvNet
GitHub 中提及
maxemerling/COVID_CT
tf
GitHub 中提及
ejlb/squeezenet-chainer
官方
GitHub 中提及
Element-Research/dpnn
官方
pytorch
GitHub 中提及
adeely9/experiment_2_python3
pytorch
GitHub 中提及
johngear/eecs504
tf
GitHub 中提及
DeepScale/SqueezeNet
官方
pytorch
GitHub 中提及
cmasch/squeezenet
tf
GitHub 中提及
m1lhaus/SimpleSqueezeNet
GitHub 中提及
osmr/imgclsmob
mxnet
GitHub 中提及
modelhub-ai/squeezenet
mxnet
GitHub 中提及
matteo-rizzo/fc4-pytorch
pytorch
GitHub 中提及
haria/SqueezeNet
官方
mxnet
GitHub 中提及
brianjychan/landuse
GitHub 中提及
gsp-27/pytorch_Squeezenet
pytorch
GitHub 中提及
vonclites/squeezenet
tf
GitHub 中提及
Dawars/SqueezeNet-tf
tf
GitHub 中提及
dividiti/ck-caffe
tf
GitHub 中提及
Banus/caffe-demo
caffe2
GitHub 中提及
DT42/squeezenet_demo
官方
GitHub 中提及

基准测试

基准方法指标
image-classification-on-imagenet-9SqueezeNet + Simple Bypass
Top 1 Accuracy: 60.4%
image-classification-on-imagenet-pSqueezeNet + Simple Bypass
Top 5 Accuracy: 82.5%
network-pruning-on-imagenetSqueezeNet (6-bit Deep Compression)
Accuracy: 57.5%
MParams: 1.24

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SqueezeNet:参数量减少50倍且模型大小小于0.5MB的AlexNet级精度 | 论文 | HyperAI超神经