新智元推薦
【新智元導讀】反向傳播算法(BP算法)是目前用來訓練人工神經網絡的最常用且最有效的算法。作為谷歌機器學習速成課程的配套材料,谷歌推出一個演示網站,直觀地介紹了反向傳播算法的工作原理。
網站地址:
https://google-developers.appspot.com/machine-learning/crash-course/backprop-scroll/
反向傳播算法對於快速訓練大型神經網絡來說至關重要。本文將介紹該算法的工作原理。
簡單的神經網絡
如上圖,你會看到一個神經網絡,其中包含一個輸入節點、一個輸出節點,以及兩個隱藏層(分別有兩個節點)。
相鄰的層中的節點通過權重
激活函數
每個節點都有一個總輸入 x、一個激活函數 f(x) 以及一個輸出 y=f(x)。
f(x)必須是非線性函數,否則神經網絡就只能學習線性模型。
常用的激活函數是 S 型函數:
誤差函數
目標是根據數據自動學習網絡的權重,以便讓所有輸入
為了衡量與該目標的差距,我們使用了一個誤差函數
正向傳播
首先,我們取一個輸入樣本
為了保持一致性,我們將輸入視為與其他任何節點相同,但不具有激活函數,以便讓其輸出與輸入相等,即
現在,我們更新第一個隱藏層。我們取上一層節點的輸出 y,並使用權重來計算下一層節點的輸入 x。
然後,我們更新第一個隱藏層中節點的輸出。 為此,我們使用激活函數 f(x)。
使用這兩個公式,我們可以傳播到網絡的其餘內容,並獲得網絡的最終輸出。
誤差導數
反向傳播算法會對特定樣本的預測輸出和理想輸出進行比較,然後確定網絡的每個權重的更新幅度。 為此,我們需要計算誤差相對於每個權重
獲得誤差導數後,我們可以使用一種簡單的更新法則來更新權重:
其中,
[注意] 該更新法則非常簡單:如果在權重提高後誤差降低了 (
其他導數
為了幫助計算
反向傳播
我們開始反向傳播誤差導數。 由於我們擁有此特定輸入樣本的預測輸出,因此我們可以計算誤差隨該輸出的變化情況。 根據我們的誤差函數
現在我們獲得了
其中,當 f(x) 是 S 型激活函數時,
一旦得出相對於某節點的總輸入的誤差導數,我們便可以得出相對於進入該節點的權重的誤差導數。
根據鏈式法則,我們還可以根據上一層得出
接下來,只需重複前面的 3 個公式,直到計算出所有誤差導數即可。
結束。
【加入社群】