AlexNet 2 模型遷移


在alexnet的權(quán)值遷移成功后,下一步要做的就是將這個(gè)網(wǎng)絡(luò)fine-tune成適用于自己的數(shù)據(jù)集的網(wǎng)絡(luò)硼瓣。這里我主要把這部分工作分成三部分

1.重新訓(xùn)練最后一層fc層

顯然癣漆,最后一層fc層肯定和你想要輸出的分類信息有關(guān)的,所以這一層的權(quán)值必須要重新訓(xùn)練


2.確定需要fine-tune的網(wǎng)絡(luò)

最簡單的情況是固定所有的權(quán)值(在初始化權(quán)值變量的時(shí)候聲明此變量不可訓(xùn)練),只重新訓(xùn)練最后一層fc層禽炬,從實(shí)驗(yàn)結(jié)果來看這個(gè)結(jié)果比較差。fine-tune越多層網(wǎng)絡(luò)勤家,有可能得到更好的效果腹尖,但是耗時(shí)也越久

在我實(shí)現(xiàn)的例子中,我直接是在模型函數(shù)里確定哪些層是可訓(xùn)練的,這點(diǎn)只需要在聲明變量的時(shí)候設(shè)置trainable參數(shù)值就可以(True表示可訓(xùn)練热幔,F(xiàn)alse表示不可訓(xùn)練)

改進(jìn):一般來說不會將整個(gè)網(wǎng)絡(luò)都微調(diào)(fine-tune)來適應(yīng)自己的數(shù)據(jù)集乐设,也就是說總有一些網(wǎng)絡(luò)層是不可訓(xùn)練的。但是如果我們?nèi)稳辉谟?xùn)練的過程中不斷循環(huán)的把image傳遞(feed in)到這些層中绎巨,顯然這樣做很耗時(shí)而且是沒有意義的(因?yàn)檫@些層是不可學(xué)習(xí)的近尚,同一張照片傳遞1次和傳遞10000次都是一樣的效果),一種改進(jìn)方法是一次性將圖片傳遞進(jìn)這些網(wǎng)絡(luò)场勤,保存提取下來的信息戈锻,然后再循環(huán)的將這些信息傳遞到可訓(xùn)練的網(wǎng)絡(luò)層里面(相當(dāng)于用不能fine-tune的網(wǎng)絡(luò)來提取圖片的特征信息,然后再建立網(wǎng)絡(luò)來處理這些特征信息)

3.調(diào)整自己的數(shù)據(jù)集和媳,滿足網(wǎng)絡(luò)的輸入要求格遭,同時(shí)可以使用數(shù)據(jù)增強(qiáng)(data augment)技術(shù),擴(kuò)展數(shù)據(jù)集


實(shí)驗(yàn)結(jié)果:

這里我用的數(shù)據(jù)集是inception模型遷移學(xué)習(xí)中用的5種花的分類

分別比較fine-tune不同層網(wǎng)絡(luò)的效果


橙色微調(diào)網(wǎng)絡(luò)最多:從conv5-fc6-fc7-fc8

淺藍(lán)中間:從fc6-fc7-fc8

深藍(lán)微調(diào)最淺的網(wǎng)絡(luò):fc7-fc8

可以看到從效果來說微調(diào)的網(wǎng)絡(luò)越多效果就越好留瞳,但是耗時(shí)也會越多拒迅。

改進(jìn):

可以看到上面跑10,000個(gè)循環(huán)就花了差不多一個(gè)小時(shí),還是有點(diǎn)慢的她倘。主要優(yōu)化的點(diǎn)在于那些不微調(diào)的網(wǎng)絡(luò)層璧微。

假設(shè)一張圖片,在alexnet中要訓(xùn)練1000次硬梁,也就是每一層網(wǎng)絡(luò)都要前向前硫,反向這張圖片1000次。對于可學(xué)習(xí)的網(wǎng)絡(luò)層來說這樣做肯定是必須的荧止,但是對于不可學(xué)習(xí)的網(wǎng)絡(luò)層(如conv1-conv2-con3-con4等)开瞭,這樣就顯得沒有必要了,因?yàn)檫@些層的參數(shù)不可改變罩息,所以沒有必要反向計(jì)算。

所以一種很直觀的方法是將圖片集一次傳遞到這些不可訓(xùn)練的網(wǎng)絡(luò)中个扰,存儲起來作為特征輸入瓷炮。以后的訓(xùn)練輸入就是這些特征值,下面可以看到這樣做加速的訓(xùn)練大概6倍左右(改進(jìn)前訓(xùn)練10000次要50min,改進(jìn)后只需要8min)

微調(diào)不同層效果對比(用不能學(xué)習(xí)的網(wǎng)絡(luò)層來提取特征):


可以看到訓(xùn)練1000*10次递宅,總體來說各種方法的區(qū)別并不是很大娘香。微調(diào)越多層的網(wǎng)絡(luò)效果稍微好一點(diǎn),但是也要花更多的時(shí)間訓(xùn)練办龄。同時(shí)注意conv4-fc8方案烘绽,在訓(xùn)練到300次的時(shí)候表現(xiàn)大幅下滑,我覺得可能是因?yàn)閷W(xué)習(xí)率過高的關(guān)系俐填。這也說明了微調(diào)越多層安接,就需要越多訓(xùn)練網(wǎng)絡(luò)的技巧,訓(xùn)練出一個(gè)好的網(wǎng)絡(luò)也越困難英融。

結(jié)論:

1.對于AlexNet模型對于沒有在訓(xùn)練中將的圖片也是有用的盏檐,特別是網(wǎng)絡(luò)越靠近輸入歇式,其提取的特征就越具有普適性。表現(xiàn)在這次試驗(yàn)中的是微調(diào)越多的網(wǎng)絡(luò)胡野,得到的效果就越好

2.如果不是微調(diào)整個(gè)模型材失,就不要將整個(gè)模型拿來訓(xùn)練。也就是用不訓(xùn)練的層拿來做特征提取硫豆,對所有的圖片只前向運(yùn)算一次龙巨,將運(yùn)算得到的特征作為后面網(wǎng)絡(luò)層的輸入。在本次試驗(yàn)中可以看到這樣子做可以大大的加快訓(xùn)練過程熊响。

3.關(guān)于Inception-v3與AlexNet模型在同樣的一種訓(xùn)練集上的遷移學(xué)習(xí)效果對比:

可以看到Inception模型的效果確實(shí)是比AlexNet的效果好很多旨别,用Inception模型提取最后一層網(wǎng)絡(luò)的特征再重新分類,在1000次循環(huán)(大概用時(shí)1min)就能達(dá)到90%的準(zhǔn)確率(測試集)耘眨,而對于AlexNet,即使是改進(jìn)后的網(wǎng)絡(luò)訓(xùn)練10000次(大概8min)大概準(zhǔn)確率(測試集)在83%左右.但是兩個(gè)網(wǎng)絡(luò)只單純的增加訓(xùn)練時(shí)間是不能再提高準(zhǔn)確率了(AlexNet維持在85%左右)



最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末昼榛,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子剔难,更是在濱河造成了極大的恐慌胆屿,老刑警劉巖,帶你破解...
    沈念sama閱讀 222,729評論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件偶宫,死亡現(xiàn)場離奇詭異非迹,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)纯趋,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,226評論 3 399
  • 文/潘曉璐 我一進(jìn)店門憎兽,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人吵冒,你說我怎么就攤上這事纯命。” “怎么了痹栖?”我有些...
    開封第一講書人閱讀 169,461評論 0 362
  • 文/不壞的土叔 我叫張陵亿汞,是天一觀的道長。 經(jīng)常有香客問我揪阿,道長疗我,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 60,135評論 1 300
  • 正文 為了忘掉前任南捂,我火速辦了婚禮吴裤,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘溺健。我一直安慰自己麦牺,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,130評論 6 398
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著枕面,像睡著了一般愿卒。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上潮秘,一...
    開封第一講書人閱讀 52,736評論 1 312
  • 那天琼开,我揣著相機(jī)與錄音,去河邊找鬼枕荞。 笑死柜候,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的躏精。 我是一名探鬼主播渣刷,決...
    沈念sama閱讀 41,179評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼矗烛!你這毒婦竟也來了辅柴?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 40,124評論 0 277
  • 序言:老撾萬榮一對情侶失蹤瞭吃,失蹤者是張志新(化名)和其女友劉穎碌嘀,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體歪架,經(jīng)...
    沈念sama閱讀 46,657評論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡股冗,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,723評論 3 342
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了和蚪。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片止状。...
    茶點(diǎn)故事閱讀 40,872評論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖攒霹,靈堂內(nèi)的尸體忽然破棺而出怯疤,到底是詐尸還是另有隱情,我是刑警寧澤催束,帶...
    沈念sama閱讀 36,533評論 5 351
  • 正文 年R本政府宣布旅薄,位于F島的核電站,受9級特大地震影響泣崩,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜洛口,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,213評論 3 336
  • 文/蒙蒙 一矫付、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧第焰,春花似錦买优、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,700評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽烘跺。三九已至,卻和暖如春脂崔,著一層夾襖步出監(jiān)牢的瞬間滤淳,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,819評論 1 274
  • 我被黑心中介騙來泰國打工砌左, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留脖咐,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 49,304評論 3 379
  • 正文 我出身青樓汇歹,卻偏偏與公主長得像屁擅,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子产弹,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,876評論 2 361

推薦閱讀更多精彩內(nèi)容