『Benchmark Analysis of Representative Deep Neural Network Architecture』论文笔记

一 为什么读这篇

蛙神推荐的,对截止2018年10月出现的各种经典CNN网络的一个测评,算是比较新的一篇综述,通过阅读这篇,搞清楚CNN最新的发展脉络。

二 截止阅读时这篇论文的引用次数

0次。毕竟10月才出的。。

三 相关背景介绍

来自意大利和法国的学者,arXiv上的时间为2018年10月

四 论文的主要贡献

1 分析对比了40多种网络在普通平台和嵌入式平台上的识别准确率,模型复杂度,计算复杂度,内存使用,预测时间

2 让研究者知道目前网络已发展到什么程度,有哪些方向值得继续探索

3 让实践者知道如何在资源限制下选择最好的网络

五 详细解读

1 介绍

其他文章都是分析网络层的组成和识别性能,对计算代价方面(内存使用,预测时间)及其对识别效果的影响的分析较少。Canziani等人在2016年年中的时候对嵌入式系统上的网络做过一些分析,不过比较的网络个数较少。Huang等人在2017年的cvpr上分析过一批SOTA的目标检测网络的速度与准确率之间的权衡。本文的目的是进行更全面的比较,同时也在两种平台上进行比较。

几个重要发现

  1. 识别准确率并不随着运算的增加而增加
  2. 模型复杂度和准确率之前并不是线性关系
  3. 期望的吞吐量是可达准确率的上界???
  4. 不是所有的网络都有效的使用了它们的参数
  5. 几乎所有网络在高配GPU上实时性能都不错,但只有少数在嵌入式上表现过关
  6. 即使网络的模型复杂度很低也得占至少0.6GB的显存

2 benchmark

软件环境:PyTorch,cuDNN-v5.1,CUDA-v9.0

硬件环境-工作站:Titan X,3840 CUDA cores

硬件环境-嵌入式:Jetson TX1 board,Maxwell 256 CUDA cores

3 网络架构

AlexNet 2012-NIPS

VGG家族:VGG-11, VGG-13, VGG-16, VGG-19,分别使用和不使用BN的版本 2014

BN-Inception 2015-ICML

GoogLeNet 2015-CVPR

SqueezeNet-v1.0, SqueezeNet-v1.1 2016

ResNet-18, ResNet-34, ResNet-50, ResNet-101, ResNet-152 2016-CVPR

Inception-v3 2016-CVPR, Inception-v4, Inception-ResNet-v2 2016-ICLR

DenseNet-121, DenseNet-169, DenseNet-201, growth rate=32 2017-CVPR

DenseNet-161 growth rate=48 2017-CVPR

ResNeXt-101(32x4d), ResNeXt-101(64x4d) 2017-CVPR

Xception 2017-CVPR

DualPathNet-68, DualPathNet-98, DualPathNet-131 2017-NIPS

SE-ResNet-50, SENet-154, SE-ResNet-101, SE-ResNet-152, SE-ResNeXt-50(32x4d), SE-ResNeXt-101(32x4d) 2018-CVPR

NASNet-A-Large, NASNet-A-Mobile 2018-CVPR

MobileNet-v1 2017, MobileNet-v2 2018-CVPR

ShuffleNet 2018

4 性能指标

NASNet-A-Large的图像大小为331x331

InceptionResNet-v2,Inception-v3,Inception-v4,Xception的图像大小为229x229

其余为224x224

A 准确率

ImageNet-1k验证集上的TOP1和TOP5准确率

B 模型复杂度

需要学习的总参数量

C 内存使用

为网络分配的内存和不同batch下处理需要的内存(batch1,2,4,8,16,32,64)

D 计算复杂度

floating-point operations(FLOPs)

乘法和加法的浮点运算。乘法和加法被记为两次浮点运算,因为许多最近的模型卷积都没有bias。

E 预测时间

5 评测结果

A 准确率 VS 计算复杂度 VS 模型复杂度

准确率VS模型复杂度VS计算复杂度.png

横轴表示计算复杂度,纵轴表示准确率,球的大小表示模型复杂度

NASNet-A-Large有最高的准确率,同时计算复杂度也最高

在低计算复杂度下,SE-ResNeXt-50(32x4d)有最高的准确率同时模型复杂度也不高(2.76M个参数)

计算复杂度和准确率之间没直接关系(SENet-154 VS SE-ResNeXt-101(32x4d))。模型复杂度和准确率之间也没直接关系(VGG-13 VS ResNet-18)

B 准确率 VS 学习能力

准确率VS学习能力.png

网络并未充分利用其所有的参数,所以有许多方法研究怎么压缩模型还能保持原模型的准确率。用TOP1准确率密度来衡量模型利用其参数的效率(TOP1准确率除其所有参数)

这点很明显,专门针对移动端研发的网络参数利用率更高,比如SqueezeNets,ShuffleNet,MobileNets和NASNet-A-Mobile

NASNet-A-Mobile和MobileNet-v2在准确率方面胜出

除此之外,Inception-v4和SE-ResNeXt-101胜出

C 预测时间

预测时间.png

绿色越重性能越好,红色越重性能越差

D 准确率 VS 预测时间

准确率VS预测时间.png

E 内存使用

内存使用.png

当batch为1时,绝大部分模型需要的显存小于1GB

绿色越重需要的内存越少

从这个图可以看出,绝大多数网络batch不超过64显存也没占多少,为啥batch到128就不行了呢

F 内存使用 VS 模型复杂度

两者呈线性关系

G 限制下的最佳网络

有问题查这个表就行了

内存使用:高(<=1.4GB)中(<=1GB)低(<=0.7G)

计算时间:半实时(15FPS),实时(30FPS),超实时(60FPS)

限制下的最佳网络.png

6 结论

把Introduction提到的几个重要发现复述了一遍

六 读后感

根据这篇文章的结论,感觉综合下来用SE-ResNeXt-50或101(32x4d)就行了。另外,对于网络的评估指标有了更进一步的认识。不是所有网络都充分利用了它们的参数,模型的准确率并不直接与其复杂度相关。

七 补充

关于复杂度分析可以看下这两篇

以VGG为例,分析深度网络的计算量和参数量

卷积神经网络的复杂度分析

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,194评论 6 490
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,058评论 2 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 156,780评论 0 346
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,388评论 1 283
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,430评论 5 384
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,764评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,907评论 3 406
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,679评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,122评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,459评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,605评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,270评论 4 329
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,867评论 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,734评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,961评论 1 265
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,297评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,472评论 2 348

推荐阅读更多精彩内容