1. 程式人生 > >神經網路訓練時,出現NaN loss

神經網路訓練時,出現NaN loss

1、梯度爆炸

原因:在學習過程中,梯度變得非常大,使得學習的過程偏離了正常的軌跡

症狀:觀察輸出日誌(runtime log)中每次迭代的loss值,你會發現loss隨著迭代有明顯的增長,最後因為loss值太大以至於不能用浮點數去表示,所以變成了NaN。

可採取的方法:1.降低學習率,比如solver.prototxt中的base_lr,降低一個數量級(至少)。如果在你的模型中有多個loss層,就不能降低基礎的學習率base_lr,而是需要檢查日誌,找到產生梯度爆炸的層,然後降低train_val.prototxt中該層的loss_weight.

2、錯誤的學習率策略及引數

原因:在學習過程中,caffe不能得出一個正確的學習率,相反會得到inf或者nan的值。這些錯誤的學習率乘上所有的梯度使得所有引數變成無效的值。

症狀:觀察輸出日誌(runtime log),你應該可以看到學習率變成NaN,例如:

... sgd_solver.cpp:106] Iteration 0, lr = -nan
可採取的方法:修改solver.prototxt檔案中所有能影響學習率的引數。比如,如果你設定的學習率策略是lr_policy:"poly", 而你又忘了設定最大迭代次數max_iter,那麼最後你會得到lr=NaN...

關於caffe學習率及其策略的內容,可以在github的/caffe-master/src/caffe/proto/caffe.proto檔案中看到(傳送門

3、錯誤的損失函式

原因:有時,在損失層計算損失值時會出現NaN的情況。比如,向InfogainLoss層沒有歸一化輸入值,使用自定義的損失層等。

症狀:觀察輸出日誌(runtime log)的時候,你可能不會發現任何異常:loss逐漸下降,然後突然出現NaN。

可採取的方法:重新嘗試該錯誤,列印損失層的值並除錯。

舉個栗子:有一回,我根據批量資料中標籤出現的 頻率去歸一化懲罰值並以此計算loss。如果有個label並沒有在批量資料中出現,頻率為0,結果loss出現了NaN的情況。在這種情況先,需要用足夠大的batch來避免這個錯誤。

4、錯誤的輸入

原因:你的輸入中存在NaN!

症狀:一旦學習過程中碰到這種錯誤的輸入,輸出就會變成NaN。觀察輸出日誌(runtime log)的時候,你可能也不會發現任何異常:loss逐漸下降,然後突然出現NaN。

可採取的方法:重建你的輸入資料集(Imdb/leveldn/hdf5...),確保你的訓練集/驗證集中沒有髒資料(錯誤的圖片檔案)。除錯時,使用一個簡單的網路去讀取輸入,如果有一個輸入有錯誤,這個網路的loss也會出現NaN。

5、Pooling層的步長大於核的尺寸

由於一些原因,步長stride>核尺寸kernel_size的Pooling層會出現NaN。比如:

layer {
  name: "faulty_pooling"
  type: "Pooling"
  bottom: "x"
  top: "y"
  pooling_param {
    pool: AVE
    stride: 5
    kernel: 3
  }
}

結果y會出現NaN。

一般來說,出現NaN有以下幾種情況:

1.如果在迭代的100輪以內,出現NaN,一般情況下的原因是因為你的學習率過高,需要降低學習率。可以不斷降低學習率直至不出現NaN為止,一般來說低於現有學習率1-10倍即可。

2.如果當前的網路是類似於RNN的迴圈神經網路的話,出現NaN可能是因為梯度爆炸的原因,一個有效的方式是增加“gradient clipping”(梯度截斷來解決)

3.可能用0作為了除數;

4.可能0或者負數作為自然對數

5.需要計算loss的陣列越界(尤其是自己,自定義了一個新的網路,可能出現這種情況)

6.在某些涉及指數計算,可能最後算得值為INF(無窮)(比如不做其他處理的softmax中分子分母需要計算exp(x),值過大,最後可能為INF/INF,得到NaN,此時你要確認你使用的softmax中在計算exp(x)做了相關處理(比如減去最大值等等))


原文連結

參考