iT邦幫忙

2022 iThome 鐵人賽

DAY 15
0
AI & Data

菜鳥工程師第一個電腦視覺(CV)專案-農作物影像辨識競賽系列 第 15

D15-影像識別經典模型簡介1st_AlexNet

  • 分享至 

  • xImage
  •  

Part0:前言

明天有工作進度大報告,在今天簡短的兩小時內要快速了解一個新模型並做總結,有點壓力呀~
廢話不多說,Let's go!


Part1:今日目標

1.認識AlexNet模型架構
2.學習該模型提倡的幾個模型設計


Part2:內容

AlexNet為Alex Krizhevsky等人在2012年在ImageNet影像識別競賽獲得冠軍的模型(論文連結),AlexNet top-5測試集錯誤率比前一年冠軍(SIFT+FVs)下降了10%(下圖),該模型的出現使得深度學習在影像識別等領域的研究開啟興盛的時代。

圖片出處


圖片出處


1.AlexNet模型架構:


圖片出處


圖片出處
根據上圖AlexNet的架構共八層,第一層到第五層是卷積層(CNN)和池化層(pooling),第六層到第八層是全連接層(fully connected Layers),較LeNet有更深的神經網路結構,共有5層卷積和3層全連接層,論文中提到幾點詳述如下。


2-1.提倡新的模型架構設計:

(1)激活函數ReLU(ReLU Nonlinearity)

解決當時流行的激活函數(sigmoid, tanh)的梯度消失(vanishing gradient)問題。梯度消失發生原因:當卷積神經網路CNN層數太深時,參數權重變化對損失(loss)的影響就會隨著神經元的傳遞而衰減,造成參數無法被更新,訓練將不繼續收斂。

(2)在多顆GPU上訓練(Training on Multiple GPUs)

受限於當時GPU記憶體容量,作者將神經網路(net)分散在兩顆GPU。平行運算方式將所有卷積核(kernel)或神經元(neurons)分成一半個放在一顆GPU上,透過此方式能相較只在一顆GPU上稍微減少運算時間。

(3)局部響應歸一化(Local Response Normalization, LRN)

對局部區域神經元間創造競爭機制,使得其中反應較大的數值變得更大,並抑制反饋較小的神經元,目標是提升預測準確性,但後續研究發現該方法幫助有限,現今已很少見。

(4)重疊池化(Overlapping Pooling)

池化層目的在蒐集特定區域特徵並統整整張圖片資訊,傳統的池化掃過的範圍並不重複,作者在實驗中顯示重疊池化能降低錯誤率並且防止過度擬合。


2-2.減少過度擬合發生的方法(Reducing Overfitting)

(1)資料擴充(Data Augmentation)

最常見且最容易防止模型過度擬合的方法就是「資料擴充」,作者提出兩種方式:

  • 圖像平移和水平鏡像反射(image translations and horizontal reflections)
  • 改變訓練圖片的RGB通道強度(altering the intensities of the RGB channels in training images),透過主成分(PCA)方法。
(2)丟棄/隱藏部分神經元(Dropout)

透過每次丟棄不同神經元(每個神經元(neuron)會有0.5機率不參與訓練(同時不參與前向傳播(forwark)和反向傳播(back propagation),造成每回合訓練的神經網路架構有些微差異,強迫模型學到更泛化的資料特徵。

以上幾項神經網路設計,後續被證明能強健模型的穩定性和提升預測表現,在現今許多複雜模型仍可看到上述相關設計。

Part3:專案進度

認識在影像辨識預測準確率大幅提升的經典模型AlexNet,透過學習該模型架構和閱讀其論文,能了解到有哪些有用的模型架構設計能應用,將當作工具在此次的農作物分類競賽上視需求使用。

Part4:下一步

明天將介紹影像識別競賽上,下一個超越AlexNet模型的經典模型VGG。


參考:

心得小語:
終於完成啦~~ 因為是距今約十年前發表的模型架構,網路上蠻多資源可學習統整,也才發現原來蠻多現今常見的技巧(dropout、ReLU)等等,居然都已經存在許久,能站在巨人的肩膀上學習真好!
感謝今日的學習,也希望明天工作的會議順利呀,明天見!
今日工時: 50min*3

/images/emoticon/emoticon30.gif

努力學習,勤奮工作,讓生命更加耀眼
Study hard, work hard, make the life more glorious.


上一篇
D14-Pytorch_模型搭建技巧1st
下一篇
D16-中繼站:前半段賽程回顧&後半段賽程規劃
系列文
菜鳥工程師第一個電腦視覺(CV)專案-農作物影像辨識競賽32
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言