一个用 BatchNorm 的 CNN 分类模型,batch size = 256 时正常收敛;现在因为显存限制把 batch 降到 8,训练 loss 抖动明显、验证精度掉 3~4 个点。请说明小 batch 下 BatchNorm 具体坏在哪(至少两点),并给出两个可行的替代或缓解方案,分别说清代价。
面试官在考什么
是否理解 BN 依赖 batch 维统计量这一前提,能否在真实资源约束下做出带代价意识的选型,而不是简单说"换个 Norm 就行"。
回答框架 · 前两步
- 小 batch 时 batch 内的均值/方差本身是高噪声估计(方差估计的方差约 O(1/b)),归一化相当于往每层输入注入随机扰动,训练抖动、收敛变差,BN 原本的正则化效果也随之减弱。
- 训练用 batch 统计量、推理用 running stats,两者不一致;小 batch 下 running stats 更新噪声大且滞后,训练/推理差距被进一步放大,验证精度掉点。
还有 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 的梯度是多少?为什么?