AlexNet 2 模型迁移


在alexnet的权值迁移成功后,下一步要做的就是将这个网络fine-tune成适用于自己的数据集的网络。这里我主要把这部分工作分成三部分

1.重新训练最后一层fc层

显然,最后一层fc层肯定和你想要输出的分类信息有关的,所以这一层的权值必须要重新训练


2.确定需要fine-tune的网络

最简单的情况是固定所有的权值(在初始化权值变量的时候声明此变量不可训练),只重新训练最后一层fc层,从实验结果来看这个结果比较差。fine-tune越多层网络,有可能得到更好的效果,但是耗时也越久

在我实现的例子中,我直接是在模型函数里确定哪些层是可训练的,这点只需要在声明变量的时候设置trainable参数值就可以(True表示可训练,False表示不可训练)

改进:一般来说不会将整个网络都微调(fine-tune)来适应自己的数据集,也就是说总有一些网络层是不可训练的。但是如果我们任然在训练的过程中不断循环的把image传递(feed in)到这些层中,显然这样做很耗时而且是没有意义的(因为这些层是不可学习的,同一张照片传递1次和传递10000次都是一样的效果),一种改进方法是一次性将图片传递进这些网络,保存提取下来的信息,然后再循环的将这些信息传递到可训练的网络层里面(相当于用不能fine-tune的网络来提取图片的特征信息,然后再建立网络来处理这些特征信息)

3.调整自己的数据集,满足网络的输入要求,同时可以使用数据增强(data augment)技术,扩展数据集


实验结果:

这里我用的数据集是inception模型迁移学习中用的5种花的分类

分别比较fine-tune不同层网络的效果


橙色微调网络最多:从conv5-fc6-fc7-fc8

浅蓝中间:从fc6-fc7-fc8

深蓝微调最浅的网络:fc7-fc8

可以看到从效果来说微调的网络越多效果就越好,但是耗时也会越多。

改进:

可以看到上面跑10,000个循环就花了差不多一个小时,还是有点慢的。主要优化的点在于那些不微调的网络层。

假设一张图片,在alexnet中要训练1000次,也就是每一层网络都要前向,反向这张图片1000次。对于可学习的网络层来说这样做肯定是必须的,但是对于不可学习的网络层(如conv1-conv2-con3-con4等),这样就显得没有必要了,因为这些层的参数不可改变,所以没有必要反向计算。

所以一种很直观的方法是将图片集一次传递到这些不可训练的网络中,存储起来作为特征输入。以后的训练输入就是这些特征值,下面可以看到这样做加速的训练大概6倍左右(改进前训练10000次要50min,改进后只需要8min)

微调不同层效果对比(用不能学习的网络层来提取特征):


可以看到训练1000*10次,总体来说各种方法的区别并不是很大。微调越多层的网络效果稍微好一点,但是也要花更多的时间训练。同时注意conv4-fc8方案,在训练到300次的时候表现大幅下滑,我觉得可能是因为学习率过高的关系。这也说明了微调越多层,就需要越多训练网络的技巧,训练出一个好的网络也越困难。

结论:

1.对于AlexNet模型对于没有在训练中将的图片也是有用的,特别是网络越靠近输入,其提取的特征就越具有普适性。表现在这次试验中的是微调越多的网络,得到的效果就越好

2.如果不是微调整个模型,就不要将整个模型拿来训练。也就是用不训练的层拿来做特征提取,对所有的图片只前向运算一次,将运算得到的特征作为后面网络层的输入。在本次试验中可以看到这样子做可以大大的加快训练过程。

3.关于Inception-v3与AlexNet模型在同样的一种训练集上的迁移学习效果对比:

可以看到Inception模型的效果确实是比AlexNet的效果好很多,用Inception模型提取最后一层网络的特征再重新分类,在1000次循环(大概用时1min)就能达到90%的准确率(测试集),而对于AlexNet,即使是改进后的网络训练10000次(大概8min)大概准确率(测试集)在83%左右.但是两个网络只单纯的增加训练时间是不能再提高准确率了(AlexNet维持在85%左右)



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

推荐阅读更多精彩内容