12组 - 《Very Deep Convolutional Networks For Large-Scale Image Recognition》

版权声明:本文为博主原创文章,欢迎转载分享,请注明本文出处。文章内容如有侵权,请联系删除。


《Very Deep Convolutional Networks For Large-Scale Image Recognition》是牛津大学 Visual Geometry Group(VGG) K. Simonyan 和 A. Zisserman 于2014年撰写的论文。这篇文章主要研究了,在大规模图像的识别问题中,卷积神经网络的深度对准确率的影响。文中采用核宽度为 3 的卷积结构(3 × 3 convolution filter),构建了一系列不同深度的卷积网络。该模型在 ImageNet Large-Scale Visual Recognition Challenge 2014 (ILSVRC-2014)中,取得了定位项目第一名,分类项目第二名的成绩。

数据预处理工作

在模型的训练过程中,卷积网络的输入是大小为 224 × 224 的RGB图像。文中对每一幅图像的预处理工作仅有,对每一个像素减去RGB的均值(mean RGB value)。

卷积神经网络的配置

卷积部分(Convolutional Layers):卷积层使用大小为 3 × 3 的接受域(receptive field)(它是可以表示“左右”、“上下”、“中心”这些概念的最小单元),在结构C中也使用到大小为 1 × 1 的接受域(它可以被看作是对输入通道的线性变换;作用是在不影响输入输出维数的情况下,对输入进行线性变换,然后通过线性整流函数(ReLU)进行非线性处理,增加网络的非线性表达能力(引用自系统学习深度学习(十七)--VGG模型));卷积步长(stride)为 1 个像素;接受域大小为 3 × 3 的卷积层的填充(padding)为 1 个像素;部分卷积层后是池化(pooling)层,池化层采用最大池化(max-pooling)函数,共有五层,max-pooling的窗口大小为 2 × 2,步长为 2 个像素,池化层的作用在于特征加强,同时缩小Feature Map的尺寸(引用自VGG网络详解及代码实现)。

全连接部分(Fully-Connected Layers):文中提到的一系列卷积网络,均只有 3 个全连接层;前两个全连接层各有4096个通道(channel),第三个全连接层有1000个通道(在ILSVRC任务中,这1000个通道用于分类,每个通道对应一个类别);所有网络中的全连接层配置相同。

VGG 16 的微体系结构如图 1 所示(原图取自MACROARCHITECTURE OF VGG16)。

图 1  -  VGG 16 的微体系结构

表 1 给出了全部卷积网络的配置,这些网络遵循上述设计原则,区别仅在于深度不同。VGG采用的是一种Pre-training的方式,这种方式在经典的神经网络中经常能够见到,就是先训练一部分小网络,然后在确定这部分网络达到稳定后,再在这基础上逐渐加深。表 1 从结构A到结构E体现的就是这个过程,并且当网络处于阶段D的时候,效果是最优的,因此阶段D的网络也就是最终的VGGNet了(引用自ConvNets Typical Architecture与VGGNet)。

表 1  -  ConvNet Configurations

文章中共构建了六个不同结构的网络。

结构 A:与AlexNet结构相似,存在5个卷积层和3个全连接层,不同的是,卷积层使用大小为 3 × 3 的接受域。

结构 A-LRN:保留AlexNet中的局部响应归一化层(Local Response Normalization, LRN),其他与结构A相同。

结构 B:在结构A的基础上,Stage1和Stage2分别增加了一个接受域大小为 3 × 3 的卷积层,共有10个卷积层,总计13层。

结构 C:在结构B的基础上,Stage3、Stage4和Stage5分别增加了一个接受域大小为 1 × 1 的卷积层,共13个卷积层,总计16层。

结构 D:在结构B的基础上,Stage3、Stage4和Stage5分别增加了一个接受域大小为 3 × 3 的卷积层,共13个卷积层,总计16层。

结构 E:在结构D的基础上,Stage3、Stage4和Stage5分别增加了一个接受域大小为 3 × 3 的卷积层,共16个卷积层,总计19层。

表 2 给出了不同网络模型中参数的数量。可见,虽然从A结构到E结构,网络深度由11层增加至19层,但参数的数量并未出现太大变化,这是因为在网络中,参数主要集中在全连接层(引用自VGG网络详解及代码实现)。

表 2  -  Number of Parameters

对卷积结构的讨论

与AlexNet(Krizhevsky et al., 2012)和ZFNet(Zeiler & Fergus, 2013; Sermanet et al., 2014)相比,VGGNet使用了更小的卷积结构。在第一个卷积层中,AlexNet使用了大小为 11 × 11、步长为 4 个像素的卷积结构,ZFNet使用了大小为 7 × 7、步长为 2 个像素的卷积结构,而VGGNet则使用了大小为 3 × 3、步长为 1 个像素的卷积结构。

可见,相邻的两个 3 × 3 的卷积和一个 5 × 5 的卷积拥有相同的有效接受域,两两相邻的三个 3 × 3 的卷积和一个 7 × 7 的卷积拥有相同的有效接受域。这让VGGNet包含了三个ReLU层而非一个,使得决策函数更具识别能力,同时减少了参数的数量。

模型训练

VGGNet的训练过程与AlexNet基本一致。

其中,优化方法采用含有动量的小批量梯度下降法(mini-batch gradient descent with momentum);批尺寸(batch size)设置为256;采用L2正则化,惩罚系数设置为 5e-4;同时为前两个全连接层添加“Dropout正则”,dropout比率设置为 0.5。

文中指出,尽管VGGNet比AlexNet拥有更多的参数和更深的层次,但是VGGNet只需要很少次迭代就能够收敛,作者猜想到两方面原因:第一,更深的层次与更小的卷积核尺寸起到了隐式正则化的作用;第二,特定层的预初始化(pre-initialisation)。

关于Pre-Initialisation。网络A的权值 W ~ (0, 0.01) 的高斯分布,偏差(bias)为0;由于存在大量的线性整流函数,不好的权值初始值对于网络训练影响较大。为了绕开这个问题,作者先通过随机的方式训练最浅的网络A;然后在训练其他网络时,把A的前4个卷积层和最后全连接层的权值当做其他网络的初始值,未赋值的中间层通过随机初始化的方式得到。(引用自Very Deep Convolutional Networks for Large-Scale Image Recognition(VGG模型)

文中通过对原始图像进行各向同性的放缩,并采取随机裁剪的方式,获得固定大小为 224 × 224 的输入图像。为了扩充数据集,对裁剪后的图像进行了随机的水平翻转及随机的RGB颜色转换。同时,文中采用了两种多规模(multi-scale)的训练方式。第一种方法是将原始图像依据不同的规模S放缩后进行裁剪训练(S = 256; S = 384),第二种方法是给定S的范围 [Smin, Smax],让每一幅图像随机选取区间内的S进行训练,相当于通过尺度抖动增强数据。



欢迎大家参与讨论!如有错误,请批评指正!

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

推荐阅读更多精彩内容