【深度学习】BatchNorm、LayerNorm

本文详细介绍了BatchNorm和LayerNorm两种归一化技术的概念、作用、计算过程以及在深度学习特别是NLP中的应用。BatchNorm主要针对通道维度归一化,适用于大batch_size和CV场景,而LayerNorm则针对样本或词向量尺度归一化,更适合NLP任务,如RNN和Transformer。

一、BatchNorm

1. 概念

对一个batch内的数据在通道尺度上计算均值和方差,将同批次同通道的数据归一化为均值为0、方差为1的正态分布,最后用对归一化后的数据进行缩放和平移来还原数据本身的分布。

上图展示了大小为[3,4,2,2]的tensor(批次大小为3,通道数为4,高为2,宽为2)的BatchNorm过程,该过程是针对训练数据的且无缩放和平移。可以看出,BatchNorm是对同一批次内同一通道的所有数据进行归一化。

训练过程中,其计算过程如下:

其中, μ B \mu_{\mathcal{B}} μB σ B 2 \sigma^2_{\mathcal{B}} σB2分别为当前批次下同一通道所有数据的均值和有偏方差, ϵ \epsilon ϵ用来防止分母为0, γ \gamma γ β \beta β是可学习的参数(通道数为 C C C时,两个参数在当前特征层的总量为 2 × C 2\times C 2×C),用来进行仿射变换,即通过缩放和平移使数据处于更好的分布上。

由于测试过程需要稳定的输出,所以并不是按照批次计算均值和方差,而是使用整个训练样本的均值和方差(通常由滑动平均法计算),如下:

其中, V a r [ x ] Var[x] Var[x]指的是无偏方差,根据下式可以看出,将之前的有偏方差转为无偏方差乘上 m m − 1 \frac{m}{m-1} <

### 不同归一化方法的核心区别 #### Batch Normalization (BatchNorm) Batch Normalization 是一种通过减少内部协变量偏移来加速深度网络训练的技术。它通过对每一批次的小批量数据计算均值和方差来进行归一化操作[^1]。具体来说,BatchNorm 对每一层的激活值进行标准化处理,使得其具有零均值和单位方差。为了保持模型表达能力不受影响,BatchNorm 还引入了可学习参数 γ 和 β 来执行 scale 和 shift 操作[^2]。 优点: - 减少梯度消失/爆炸问题。 - 加速收敛速度。 缺点: - 需要较大的 mini-batch 大小才能获得稳定的统计量。 - 推理阶段需要额外存储训练期间的全局均值和方差。 适用场景: - 主要用于卷积神经网络(CNNs),尤其是当 batch size 较大时效果最佳。 --- #### Layer Normalization (LayerNorm) Layer Normalization 的设计目标是在序列建模任务中克服 BatchNorm 的局限性。与 BatchNorm 不同的是,LayerNorm 计算同一样本内所有特征维度上的均值和标准差,并对其进行归一化[^4]。这意味着即使在一个批次只有一个样例的情况下也能正常工作。 优点: - 不依赖于 mini-batch 统计信息,在较小或单一实例上表现良好。 - 更适合 RNN/LSTM 等结构中的长期依赖关系捕捉。 缺点: - 可能会忽略跨样本之间的交互特性。 适用场景: - 自然语言处理领域内的 Transformer 架构通常采用 LayerNorm,特别是在长上下文中。 --- #### Instance Normalization (InstanceNorm) Instance Normalization 类似于 LayerNorm,但它针对图像风格迁移等领域进行了优化。InstanceNorm 将每个通道视为独立单元并分别对其应用归一化过程;即对于单张图片而言,它是逐像素地基于该位置处各颜色分量单独调整尺度因子[^3]。 优点: - 特别适用于生成对抗网络(GAN),有助于分离内容与样式特征。 - 能够更好地保留局部纹理细节。 缺点: - 如果输入包含过多噪声,则可能导致不稳定的结果。 适用场景: - 图像生成任务如 CycleGAN 或 Style Transfer 中广泛运用。 --- #### Group Normalization (GroupNorm) Group Normalization 提出了另一种折衷方案——将通道划分为若干组并对每组实施类似于 InstanceNorm 的方式完成规范化流程[^5]。这种方法既避免了对大批量的需求又兼顾到了多维空间的信息融合需求。 优点: - 在极端情况下(比如非常低分辨率或者高宽比失衡的数据集),仍然可以维持较好的性能. - 结合了其他三种技术的优点. 缺点: - 参数调节较为复杂,需手动设定合适的分组数目. 适用场景: - COCO 数据集中的人体姿态估计等视觉识别挑战赛项目里表现出色. ```python import torch.nn as nn # Example usage of different normalization layers in PyTorch batch_norm = nn.BatchNorm2d(num_features=64) # For CNN with large batches layer_norm = nn.LayerNorm(normalized_shape=(64)) # Suitable for NLP tasks like Transformers instance_norm = nn.InstanceNorm2d(num_features=64)# Ideal for GAN-based image generation models group_norm = nn.GroupNorm(num_groups=8, num_channels=64) # Balanced approach combining benefits across domains ```
评论 2
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Fulin_Gao

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值