前言
Batch Normalization是由google提出的一種訓練優化方法。參考論文:Batch Normalization Accelerating Deep Network Training by Reducing Internal Covariate Shift
個人覺得BN層的作用是加快網絡學習速率,論文中提及其它的優點都是這個優點的副產品。
網上對BN解釋詳細的不多,大多從原理上解釋,沒有說出實際使用的過程,這里從what, why, how三個角度去解釋BN。
What is BN
Normalization是數據標准化(歸一化,規范化),Batch 可以理解為批量,加起來就是批量標准化。
先說Batch是怎么確定的。在CNN中,Batch就是訓練網絡所設定的圖片數量batch_size。
Normalization過程,引用論文中的解釋:
輸入:輸入數據x1..xm(這些數據是准備進入激活函數的數據)
計算過程中可以看到,
1.求數據均值;
2.求數據方差;
3.數據進行標准化(個人認為稱作正態化也可以)
4.訓練參數γ,β
5.輸出y通過γ與β的線性變換得到新的值
在正向傳播的時候,通過可學習的γ與β參數求出新的分布值
在反向傳播的時候,通過鏈式求導方式,求出γ與β以及相關權值
Why is BN
解決的問題是梯度消失與梯度爆炸。
關於梯度消失,以sigmoid函數為例子,sigmoid函數使得輸出在[0,1]之間。
事實上x到了一定大小,經過sigmoid函數的輸出范圍就很小了,參考下圖
如果輸入很大,其對應的斜率就很小,我們知道,其斜率(梯度)在反向傳播中是權值學習速率。所以就會出現如下的問題,
在深度網絡中,如果網絡的激活輸出很大,其梯度就很小,學習速率就很慢。假設每層學習梯度都小於最大值0.25,網絡有n層,因為鏈式求導的原因,第一層的梯度小於0.25的n次方,所以學習速率就慢,對於最后一層只需對自身求導1次,梯度就大,學習速率就快。
這會造成的影響是在一個很大的深度網絡中,淺層基本不學習,權值變化小,后面幾層一直在學習,結果就是,后面幾層基本可以表示整個網絡,失去了深度的意義。
關於梯度爆炸,根據鏈式求導法,
第一層偏移量的梯度=激活層斜率1x權值1x激活層斜率2x…激活層斜率(n-1)x權值(n-1)x激活層斜率n
假如激活層斜率均為最大值0.25,所有層的權值為100,這樣梯度就會指數增加。
How to use BN
先解釋一下對於圖片卷積是如何使用BN層。
這是文章卷積神經網絡CNN(1)中5x5的圖片通過valid卷積得到的3x3特征圖(粉紅色)。特征圖里的值,作為BN的輸入,也就是這9個數值通過BN計算並保存γ與β,通過γ與β使得輸出與輸入不變。假設輸入的batch_size為m,那就有m*9個數值,計算這m*9個數據的γ與β並保存。正向傳播過程如上述,對於反向傳播就是根據求得的γ與β計算梯度。
這里需要着重說明2個細節:
1.網絡訓練中以batch_size為最小單位不斷迭代,很顯然,新的batch_size進入網絡,機會有新的γ與β,因此,在BN層中,有總圖片數/batch_size組γ與β被保存下來。
2.圖像卷積的過程中,通常是使用多個卷積核,得到多張特征圖,對於多個的卷積核需要保存多個的γ與β。
結合論文中給出的使用過程進行解釋
輸入:待進入激活函數的變量
輸出:
1.對於K維的輸入,假設每一維包含m個變量,所以需要K個循環。每個循環中按照上面所介紹的方法計算γ與β。這里的K維,在卷積網絡中可以看作是卷積核個數,如網絡中第n層有64個卷積核,就需要計算64次。
需要注意,在正向傳播時,會使用γ與β使得BN層輸出與輸入一樣。
2.在反向傳播時利用γ與β求得梯度從而改變訓練權值(變量)。
3.通過不斷迭代直到訓練結束,求得關於不同層的γ與β。如網絡有n個BN層,每層根據batch_size決定有多少個變量,設定為m,這里的mini-batcherB指的是特征圖大小*batch_size,即m=特征圖大小*batch_size,因此,對於batch_size為1,這里的m就是每層特征圖的大小。
4.不斷遍歷訓練集中的圖片,取出每個batch_size中的γ與β,最后統計每層BN的γ與β各自的和除以圖片數量得到平均直,並對其做無偏估計直作為每一層的E[x]與Var[x]。
5.在預測的正向傳播時,對測試數據求取γ與β,並使用該層的E[x]與Var[x],通過圖中11:所表示的公式計算BN層輸出。
注意,在預測時,BN層的輸出已經被改變,所以BN層在預測的作用體現在此處
至此,BN層的原理與使用過程就解釋完畢,給出的解釋都是本人覺得值得注意或這不容易了解的部分,如有錯漏,請指正。