導讀

如何去掉batch normalization層來加速神經網絡。

介紹

Batch Normalization是將各層的輸入進行歸一化，使訓練過程更快、更穩定的一種技術。在實踐中，它是一個額外的層，我們通常添加在計算層之后，在非線性之前。它包括兩個步驟：

首先減去其平均值，然后除以其標準差

進一步通過γ縮放，通過β偏移，這些是batch normalization層的參數，當網絡不需要數據的時候，均值為0、標準差為1。

Batch normalization在神經網絡的訓練中具有較高的效率，因此得到了廣泛的應用。但它在推理的時候有多少用處呢？

一旦訓練結束，每個Batch normalization層都擁有一組特定的γ和β，還有μ和σ，后者在訓練過程中使用指數加權平均值進行計算。這意味著在推理過程中，Batch normalization就像是對上一層（通常是卷積）的結果進行簡單的線性轉換。

由于卷積也是一個線性變換，這也意味著這兩個操作可以合并成一個單一的線性變換！這將刪除一些不必要的參數，但也會減少推理時要執行的操作數量。

在實踐中怎么做？

用一點數學知識，我們可以很容易地重新對卷積進行排列來處理batch normalization。提醒一下，對一個輸入x進行卷積之后再進行batch normalization的運算可以表示為：

那么，如果我們重新排列卷積的W和b，考慮batch normalization的參數，如下：

我們可以去掉batch normalization層，仍然得到相同的結果！

注意：通常，在batch normalization層之前的層中是沒有bias的，因為這是無用的，也是對參數的浪費，因為任何常數都會被batch normalization抵消掉。

這樣做的效果怎樣？

我們將嘗試兩種常見的架構：

使用batch norm的VGG16

ResNet50

為了演示，我們使用ImageNet dataset和PyTorch。兩個網絡都將訓練5個epoch，看看參數數量和推理時間的變化。

1. VGG16

我們從訓練VGG16 5個epoch開始(最終的準確性并不重要)：

參數的數量：

單個圖像的初始推理時間為：

如果使用了batch normalization折疊，我們有：

以及：

8448個參數被去掉了，更好的是，幾乎快了0.4毫秒！最重要的是，這是完全無損的，在性能方面絕對沒有變化：

讓我們看看它在Resnet50的情況下是怎么樣的！

2. Resnet50

同樣的，我們開始訓練它5個epochs：

初始參數量為：

推理時間為：

使用batch normalization折疊后，有：

和：

現在，我們有26,560的參數被移除，更驚訝的hi，推理時間減少了1.5ms，性能一點也沒降。

責任編輯：lq！