Batch Normalization

批量归一化(BatchNormalization)是深度学习中用于优化训练过程的技术,它解决了不同维度数值差异大导致的梯度消失或爆炸问题,使训练更加稳定。通过输入特征的标准化,减小了内部协变量位移,加速了网络收敛。在前向传播中,对每个mini-batch的数据进行归一化,并在激活函数前后皆可应用。此外,通过引入可学习的参数β和γ,允许网络学习到不同的分布。在测试阶段,使用训练时所有mini-batch的指数加权平均值进行归一化。批量归一化是现代神经网络架构中不可或缺的部分,显著提升了模型的性能和训练效率。

Batch Normalization

要解决的问题

如果在不同的维度中的数值差异较大,会产生如图所示的情况:

  • 如果x1很小,当w1有改变Δw1时,结果y有改变Δe就会很小,于是所计算出来的梯度就会相对很小;
  • 如果x2很大,当w2有改变Δw2时,结果y有较大的改变,于是计算出来的梯度就会很大

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YjcyIJW0-1629901870384)(C:\Users\nth12\AppData\Roaming\Typora\typora-user-images\image-20210825212513953.png)]

以上的情况会使得训练变得十分困难!!

针对输入的特征归一化

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-A0Luf0V9-1629901870386)(C:\Users\nth12\AppData\Roaming\Typora\typora-user-images\image-20210825213220286.png)]

如图所示,假设现有R个训练样本的特征向量每个都是n维的:
x1,x2,x3⋯xR∈Rn x^1,x^2,x^3\cdots x^R \in R^n x1,x2,x3xRRn
对所有训练样本d的每个维度做如下的操作:
x^ir←xir−miδi \hat{x}_{i}^r \leftarrow \frac{x_i^r - m_i}{\delta_i} x^irδixirmi

前向传播过程中的值归一化

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-kNHCzox3-1629901870387)(C:\Users\nth12\AppData\Roaming\Typora\typora-user-images\image-20210825214117656.png)]

  • 如图所示,假设三个训练样本正通过同一个神经元,产生输出z1,z2,z3。与输入中的归一化执行相同的方式即可。
    • 注意此时左边的每个黑色的x长条是三个训练样本(经过输入的归一化后的),右边的三个z实际上是三个标量为三个训练样本通过权值为w1的神经元后的输出结果。
    • 激活前后执行均可
  • 但显然只可一个Batch计算一次上述结果,因为GPU没有足够多的内存容量一次装载所有的数据

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8UxZV5Ia-1629901870389)(C:\Users\nth12\AppData\Roaming\Typora\typora-user-images\image-20210825220543820.png)]

  • 除此之外,为了让归一化后的结果有不同的分布,可以做如下的修正

  • z^i=β+γz~i \hat{z}^i = \beta + \gamma \tilde{z}^i z^i=β+γz~i

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-rlx2wh4W-1629901870389)(C:\Users\nth12\AppData\Roaming\Typora\typora-user-images\image-20210825222153094.png)]

  • 其中求和表示对一个midibatch中得到的所有向量进行求和,z(i)表示minibatch中第i个样本对应的向量(维度为当前层神经元的个数)。

测试时的归一化

  • 使用涵盖所有训练时minibatch的指数加权平均进行计算

    [外链图片转存中...(img-cvZKN3ET-1629901870390)]

  • 使用训练集的结果的指数加权平均来更新测试集

批归一化(Batch Normalization)是深度学习中一种重要的技术,用于加速神经网络的训练过程并提高模型的稳定性。其核心思想是在训练过程中对每一层的输入进行标准化处理,使得数据分布更加稳定,从而缓解内部协变量偏移(Internal Covariate Shift)问题[^2]。 ### 概念 批归一化的基本步骤包括以下几个方面: 1. **标准化输入**:对于某一层的输入 $ x $,计算其均值 $ \mu_B $ 和方差 $ \sigma_B^2 $,然后对输入进行标准化: $$ \hat{x}^{(k)} = \frac{x^{(k)} - \mu_B}{\sqrt{\sigma_B^2 + \epsilon}} $$ 其中 $ \epsilon $ 是一个很小的常数,用于防止除零错误。 2. **可学习的参数**:在标准化之后,引入两个可学习的参数 $ \gamma $ 和 $ \beta $,用于缩放和偏移标准化后的值: $$ y^{(k)} = \gamma^{(k)} \hat{x}^{(k)} + \beta^{(k)} $$ 这一步确保了网络能够学习到适合当前任务的数据分布。 ### 应用 批归一化在深度学习中的应用非常广泛,尤其在卷积神经网络(CNN)和全连接网络(FCN)中效果显著。以下是一些典型的应用场景和优势: - **加速训练**:通过减少内部协变量偏移,批归一化可以显著加快模型的训练速度。实验表明,使用批归一化的模型可以在更少的迭代次数内达到相同的准确率。 - **提高模型稳定性**:批归一化有助于缓解梯度消失和梯度爆炸问题,从而提高模型的稳定性。它使得网络对初始化的敏感度降低,从而更容易训练深层模型。 - **减少对正则化的需求**:由于批归一化本身具有一定的正则化效果,因此在使用该技术时可以减少对其他正则化方法(如Dropout)的依赖。 - **提升模型性能**:在许多任务中,例如图像分类和目标检测,批归一化可以显著提升模型的性能。例如,在ImageNet数据集上,使用批归一化的模型通常可以获得更高的准确率。 ### 示例代码 以下是一个简单的示例,展示了如何在PyTorch中使用批归一化: ```python import torch import torch.nn as nn class SimpleNet(nn.Module): def __init__(self): super(SimpleNet, self).__init__() self.layer = nn.Sequential( nn.Linear(100, 256), nn.BatchNorm1d(256), nn.ReLU(), nn.Linear(256, 10) ) def forward(self, x): return self.layer(x) ``` 在这个示例中,`nn.BatchNorm1d(256)` 对全连接层的输出进行批归一化处理。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值