1. 程式人生 > >梯度彌散和梯度爆炸

梯度彌散和梯度爆炸

問題描述

先來看看問題描述。

當我們使用sigmoid funciton 作為啟用函式時,隨著神經網路hidden layer層數的增加,訓練誤差反而加大了,如上圖所示。

下面以2層隱藏層神經網路為例,進行說明。

結點中的柱狀圖表示每個神經元引數的更新速率(梯度)大小,有圖中可以看出,layer2整體速度都要大於layer1.

我們又取每層layer中引數向量的長度來粗略的估計該層的更新速率,得到下圖。

可以看出,layer2的速率都要大於layer1.

然後我們繼續加深神經網路的層數。

可以得到下面的結論:

靠近輸出層的hidden layer 梯度大,引數更新快,所以很快就會收斂;

而靠近輸入層的hidden layer 梯度小,引數更新慢,幾乎就和初始狀態一樣,隨機分佈。

在上面的四層隱藏層網路結構中,第一層比第四層慢了接近100倍!!

這種現象就是梯度彌散(vanishing gradient problem)。而在另一種情況中,前面layer的梯度通過訓練變大,而後面layer的梯度指數級增大,這種現象又叫做梯度爆炸(exploding gradient problem)

總的來說,就是在這個深度網路中,梯度相當不穩定(unstable)。

直觀說明

那麼為何會出現這種情況呢?

現在我們來直觀的說明一下。

在上面的升級網路中,我們隨意更新一個引數,加上一個Δw,(我們知道可以使用引數變化量來估計偏導數的大小)這個引數的更新會隨著網路向前傳播。

而根據sigmoid的特點,它會將+∞~-∞之間的輸入壓縮到0~1之間。當input的值更新時,output會有很小的更新。

又因為上一層的輸出將作為後一層的輸入,而輸出經過sigmoid後更新速率會逐步衰減,直到輸出層只會有微乎其微的更新。

數學說明

如果上面的例子還不夠清楚,下面我們來看看,不是很嚴密的數學證明。

假設上面是一個三層hidden layer的神經網路,每一層只有一個neuron,我們下面的分析僅僅針對bias,w也是可以類比的。

C是損失函式。

每一層的輸入為z,輸出為a,其中有z = w*a + b。

上面的等式∂c/∂b1由每一層的導數乘上對應的w最後乘上∂c/∂a4組成。

我們給b1一個小的改變Δb1,它會在神經網路中起連鎖反應,影響最後的C。

我們使用變化率∂c/∂b1~Δc/Δb1來估計梯度。接下來可以進行遞推了。

先來計算Δb1對a1的影響。σ(z)為sigmoid函式。

結果正好是上面∂c/∂b1等式的第一項,然後影響下一層的輸出。

將上面推匯出來的兩個式子聯合起來,就得到b1對於z2的影響:

再和∂c/∂b1等式對比一下,驚喜!!

然後的推導就是完全一樣了,每個neuron的導數,乘上w,最終得到C的變化量:

兩邊除以Δb1:

sigmoid導數影象:

sigmoid導數在0取得最大值1/4。

如果我們使用均值為0,方差為1的高斯分佈初始化引數w,有|w| < 1,所以有:

可以看出隨著網路層數的加深的term也會變多,最後的乘積會指數級衰減,

這就是梯度彌散的根本原因。

而有人要問在train的時候如果引數w變得足夠大,就可能使|w|>1,就不滿足了。

的確這樣不會有梯度彌散問題,根據我們之前的分析,當|W|>1時,會使後面的layer引數指數級增加,從而引發梯度爆炸。

解決方法

梯度不穩定的方法就是,使用其他啟用函式替代sigmoid,比如Relu等等,這裡就不細說了。

參考文獻:http://neuralnetworksanddeeplearning.com/chap5.html#the_vanishing_gradient_problem