0
雷鋒網按：本文作者Professor ho，原文載于其知乎主頁，雷鋒網獲其授權發(fā)布。
傳統(tǒng)的“提拉米蘇”式卷積神經網絡模型，都以層疊卷積層的方式提高網絡深度，從而提高識別精度。但層疊過多的卷積層會出現(xiàn)一個問題，就是梯度彌散(Vanishing)，backprop無法有效地把梯度更新到前面的網絡層，導致前面的層參數(shù)無法更新。
而BatchNormalization（BN）、ResNet的skip connection就是為了解決這個問題，BN通過規(guī)范化輸入數(shù)據(jù)改變數(shù)據(jù)分布，在前傳過程中消除梯度彌散。而skip connection則能在后傳過程中更好地把梯度傳到更淺的層次中。那么問題來了，
為什么加了一個捷徑就能把梯度傳到淺層網絡？
這個要從神經網絡梯度更新的過程說起，如果讀者已經非常熟悉神經網絡的梯度更新，可以快進這部分，但這個梯度更新的原理才是整個問題的關鍵。
這里主要引用cs231n的講義，個人認為這是理解神經網絡梯度更新講得最好的課程，建議直接看cs231n 2016的視頻講解，Andrej Karpathy在視頻里講得非常清晰，配合他的PPT看非常易懂，這里是地址:CS231n Winter 2016: Lecture 4: Backpropagation, Neural Networks 1。這里是YouTube視頻鏈接，國內的朋友請百度：cs231n 2016視頻。
由于篇幅原因，就不像Andrej在視頻里從實例講起那么詳細了，這里就只講梯度在中間層傳播時的計算過程，如果還沒完全了解神經網絡梯度更新原理的，建議先把視頻看完。
圖1 來自斯坦福cs231n課程slides
當梯度傳播到中間層的神經元f時，如圖1所示，來自上一層的梯度dLdz從右邊z進入，傳到中間的神經元。此神經元在左邊有兩個輸入，分別是x和y，為了計算L對于x和y的梯度dLdx和dLdy，就必須先計算dzdx和dzdy，根據(jù)復合函數(shù)求導公式，dLdx = dLdz * dzdx，dLdy = dLdz * dzdy，這樣就能算出傳播到x和y的梯度了。也就是說通過這個方法，來自深一層的梯度就能傳播到x和y當中。
圖2
讓我們來考慮一個新的情況。圖2虛線框為一個神經元block，假設輸入x為10，權重w1=0.1，w2=0.1，w3=0.1，w4=0.1，每個神經元對輸入的操作均為相乘。我們對它進行前傳和后傳的計算，看看梯度的變化情況：
前向傳播：
首先x與w1相乘，得到1；1與w2相乘，得到0.1，以此類推，如下面的gif圖綠色數(shù)字表示
圖3 前向傳播
后向傳播：
假設從下一層網絡傳回來的梯度為1（最右邊的數(shù)字），后向傳播的梯度數(shù)值如下面gif圖紅色數(shù)字表示：
圖4 后向傳播
那么這里可以看到，本來從上一層傳過來的梯度為1，經過這個block之后，得到的梯度已經變成了0.0001和0.01，也就是說，梯度流過一個blcok之后，就已經下降了幾個量級，傳到前一層的梯度將會變得很小！
這就是梯度彌散。假如模型的層數(shù)越深，這種梯度彌散的情況就更加嚴重，導致淺層部分的網絡權重參數(shù)得不到很好的訓練，這就是為什么在Resnet出現(xiàn)之前，CNN網絡都不超過二十幾層的原因。
既然梯度經過一層層的卷積層會逐漸衰減，我們來考慮一個新的結構，如圖5：
圖5
假如，我們在這個block的旁邊加了一條“捷徑”（如圖5橙色箭頭），也就是常說的“skip connection”。假設左邊的上一層輸入為x，虛線框的輸出為f(x)，上下兩條路線輸出的激活值相加為h(x)，即h(x) = F(x) + x，得出的h(x)再輸入到下一層。
圖6
當進行后向傳播時，右邊來自深層網絡傳回來的梯度為1，經過一個加法門，橙色方向的梯度為dh(x)/dF(x)=1，藍色方向的梯度也為1。這樣，經過梯度傳播后，現(xiàn)在傳到前一層的梯度就變成了[1, 0.0001, 0.01]，多了一個“1”！正是由于多了這條捷徑，來自深層的梯度能直接暢通無阻地通過，去到上一層，使得淺層的網絡層參數(shù)等到有效的訓練！
這個想法是何等的簡約而偉大，不得不佩服作者的強大的思維能力！
圖7
如圖7，左邊來了一輛裝滿了“梯度”商品的貨車，來領商品的客人一般都要排隊一個個拿才可以，如果排隊的人太多，后面的人就沒有了。于是這時候派了一個人走了“快捷通道”，到貨車上領了一部分“梯度”，直接送給后面的人，這樣后面排隊的客人就能拿到更多的“梯度”。
圖8
ResNet正是有了這樣的Skip Connection，梯度能暢通無阻地通過各個Res blocks，作者何凱明說到，唯一影響深度的就是內存不足，因此只要內存足夠，上千層的殘差網絡也都能實現(xiàn)。
而DenseNet更為極端，它的skip connection不僅僅只連接上下層，直接實現(xiàn)了跨層連接，每一層獲得的梯度都是來自前面幾層的梯度加成。
圖9 DenseNet結構
DenseNet在增加深度的同時，加寬每一個DenseBlock的網絡寬度，能夠增加網絡識別特征的能力，而且由于DenseBlock的橫向結構類似 Inception block的結構，使得需要計算的參數(shù)量大大降低。因而此論文獲得了CVPR2017最佳論文獎項！
圖10 DenseNet詳細結構 來自原論文
ResBlock與BN的結合能夠完美解決梯度彌散的問題，這使得更深的網絡成為可能。卷積神經網絡除了不斷往深度發(fā)展，在寬度上也不斷拓展，兩者結合起來可以創(chuàng)造出更強大的CNN模型。期待更多杰出的工作！
由于筆者水平尚淺，對上面的概念理解或許有偏差，歡迎各位指正，不勝感激。如果覺得這篇文章對您有幫助，請分享給您的朋友，讓更多人一起學習。
雷峰網版權文章，未經授權禁止轉載。詳情見轉載須知。
