批归一化(Batch Normalization)

批归一化是现在越来越多的神经网络采用的方法,其具有加快训练速度、防止过拟合等优点,尤其在深度神经网络中效果非常好。现将BN的学习整理一篇文章备忘。

1.为什么要采用BN?

随着神经网络的层数加深,研究者发现神经网络训练起来越困难,收敛越慢。BN就是为解决这一问题提出的。
首先明确神经网络之所以可以先训练,再预测并取得较好效果的前提假设:

我们假设神经网络的训练数据和测试数据是独立同分布

在神经网络的训练过程中,如果输入数据的分布不断变化,神经网络将很难稳定的学习规律,这也是one example SGD训练收敛慢的原因(随机得到的数据前后之间差距可能会很大)。而网络的每一层通过对输入数据进行线性和非线性变换都会改变数据的分布,随着网络层数的加深,每层接收到的数据分布都不一样,这还怎么学习规律呀,这就使得深层网络训练困难。

BN的启发来源是:之前的研究表明如果在图像处理中对输入图像进行白化操作的话(所谓白化,就是对输入数据分布变换到0均值,单位方差的正态分布)那么神经网络会较快收敛。神经网络有很多隐藏层,图像只是第一层的输入数据,对于每一个隐藏层来说,都有一个输入数据,即前一层的输出。BN将每一层的输入都进行了类似于图像白化的操作,将每层的数据都控制在稳定的分布内,并取得了很好的效果。

2.怎么进行BN?

BN算法是专门针对mini-batch SGD进行优化的,mini-batch SGD一次性输入batchsize个数据进行训练,相比one example SGD,mini-batch SGD梯度更新方向更准确,毕竟多个数据的分布和规律更接近整体数据的分布和规律,类似于多次测量取平均值减小误差的思想,所以收敛速度更快。


BN步骤

3.BN的本质思想

BN究竟对数据的分布做了什么处理,我们来看下面的示意图:



在概率论中我们都学过,数据减去均值除以方差后,将变成均值为0,方差为1的标准正态分布。如果数据分布在激活函数(图中假设为sigmoid)梯度比较小的范围,在深层神经网络训练中将很容易出现梯度消失的现象,这也是深度网络难训练的原因。通过规范化处理后的数据分布在0附近,图中为激活函数梯度最大值附近,较大的梯度在训练中收敛速度自然快。

但是,关键问题出现了,分布在0附近的数据,sigmoid近似线性函数,这就失去了非线性激活操作的意义,这种情况下,神经网络将降低拟合性能,如何解决这一问题呢?作者对规范化后的(0,1)正态分布数据x又进行了scale和shift操作:y = scale * x + shift,即对(0,1)正态分布的数据进行了均值平移和方差变换,使数据从线性区域向非线性区域移动一定的范围,使数据在较大梯度和非线性变换之间找到一个平衡点,在保持较大梯度加快训练速度的同时又不失线性变换提高表征能力。这两个参数需要在训练中由神经网络自己学习,即公式中的γ和β。如果原始数据的分布就很合适,那么即使经过BN,数据也可以回到原始分布状态,这种情况下就相当于恒等变换了,当然这是特殊情况。

4.预测时的BN实现:

在训练时,BN的操作步骤如第一张图所示,那么在预测时,每次只输入一张图的情况下,无法进行均值和方差的计算,此时该怎么实现BN呢?
正是因为训练数据和测试数据是独立同分布的,所以我们可用训练时的所有均值和方差来对测试数据的均值和方差进行无偏估计 。本来mini-batch SGD就是在整体数据量大无法一次性操作的情况下,把数据切割成几部分,用部分近似整体的解决方案。在训练时,将每一个mini-batch的均值和方差记录下,估计出整体的均值和方差如下:


最后一步的计算就是:

这样,就完成了预测时的BN计算。

5.BN计算细节:

首先要明确一点:BN是沿着batch方向计算的,每个神经元都会有一组(γ,β)在非线性激活前面约束数据。

(1)全连接层的BN计算方法:

假设batch_size=m, 输入的每一个样本有d维,记为x_i=(x_i^{(1)},x_i^{(2)},x_i^{(3)},…,x_i^{(d)}), 其中i=1,2,3,…,m
下标表示batch,上标表示一个样本中第几个维度,即第几个神经元。
那么BN计算如下:
μ_1 = \frac{1}{m} (x_1^{(1)}+x_2^{(1)}+x_3^{(1)}+…+x_m^{(1)})
……
\hat{x}_i=\gamma_ix+\beta
中间的过程省略了,其核心思想就是BN是对每一个batch的某一固定维度规范化的,一个样本中有d维,就会求出d组(γ,β),即每一个神经元都有一组(γ,β)。
(2)卷积层的BN计算方法:
在卷积层中,数据在某个卷积层中的维度是[batch, w, h, c],其中batch表示batch_size,w是feature map的宽,h是feature map的高,c表示channels。在沿着batch的方向,每个channel的feature map就相当于一个神经元,经过BN后会得到c组(γ,β)。此时的BN算法可表示如下:


图中的, 表示一个batch中每个样本相同位置的feature map中的元素,共有batchwh个元素。计算结束后会有c对()。

6.BN的反向传播

7.BN究竟用在什么位置?

原论文中BN操作是放在线性计算后,非线性激活前,即:
g(BN(WX))
其中g()表示激活函数。
这里建议参考一下ResNet_v1和ResNet_v2的用法:

最后一点还需要注意的是,在使用BN后,神经网络的线性计算(WX + b)中的偏置b将不起作用,因为在(WX + b)求均值后b作为常数均值还是b,在规范化的过程中原数据要减去均值,所以b在这两步计算中完全抵消了。但由于BN的算法中有一个偏置项β,它完全可以代替b的作用,所以有BN的计算中可不用b。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,607评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,047评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,496评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,405评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,400评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,479评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,883评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,535评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,743评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,544评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,612评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,309评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,881评论 3 306
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,891评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,136评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,783评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,316评论 2 342

推荐阅读更多精彩内容