一个 batch 的输入是 4 个样本 × 3 个特征。请分别说明 BatchNorm 和 LayerNorm 各在哪个维度上计算均值和方差——各有几个均值、每个均值由几个数算出——并解释为什么训练时用当前 batch 的统计量、推理时却要用滑动平均。
面试官在考什么
判断候选人是否能准确画出归一化维度,而不是靠「BN 按 batch、LN 按层」这种模糊口诀。
回答框架 · 前两步
- BN:对每个特征(通道)在 batch 维度上统计,4×3 的输入得到 3 个均值,每个由 4 个数算出;LN:对每个样本在自己所有特征上统计,得到 4 个均值,每个由 3 个数算出。
- BN 在 CNN 中通常对 (N, H, W) 求统计量、每个通道一组 γ/β;LN 与 batch 无关,对每个样本单独归一化。
还有 3 条答案要点、 3 层追问、 5 个常见坑, 以及针对你简历的追问路径,在定制报告里。
延伸阅读
想看这个方向的整体准备路线? 进入「深度学习基础」方向页
同一能力域的其他题
- 请说明 BatchNorm 和 LayerNorm 的计算方式有什么不同?在同一个 batch 里,BN 的统计量是怎么算出来的、推理时又用的是什么?
- 为什么深层网络需要残差连接?它解决了什么问题?换个角度问:残差结构为什么能让梯度更好地传回去?
- Dropout 在训练和推理阶段的行为分别是什么?为什么推理时要乘一个系数?它和 BatchNorm 一起用时有什么坑?
- 你手写了一个两层全连接网络(输入 784 → 隐层 256 + ReLU → 10 类 softmax 交叉熵)和完整的前向/反向传播,但训练完全不收敛。你打算用数值梯度检验(有限差分)来确认反向传播写对了。请具体说明:你用什么误差度量判断"通过"、epsilon 取多大、用 float32 还是 float64、用多少参数和多少条样本做检验、要不要关掉随机性;以及如果解析梯度和数值梯度在 1e-2 量级对不上,你如何定位到是哪一层、哪个公式写错了。
- 你在训练一个40层Transformer,PyTorch DDP多机多卡,混合精度。固定数据子集和随机种子后,loss前100步正常下降,之后部分卡突然出现NaN,其他卡继续下降。团队有人建议全局学习率减半,有人建议加梯度裁剪,有人建议用gradient checkpointing。作为训练负责人,你会怎样切分定位问题、决定短期止血和长期改动?请给出具体排查顺序、验证实验和取舍标准。
- 一个两层网络:z = W1·x,a = ReLU(z),y = W2·a,损失 L = ½‖y − t‖²。请写出 ∂L/∂W2 和 ∂L/∂W1 的表达式(写清每个因子的形状,以及要不要转置),并回答:如果某个样本的 z 的每个分量都小于 0,这个样本对 W1 的梯度是多少?为什么?