如何提升神经网络训练稳定性 从损失震荡到收敛

神经网络训练稳定性由梯度分布、学习率策略、初始化方式和网络结构共同决定。损失震荡的直接诱因是梯度范数失衡,梯度裁剪阈值建议Transformer设为1.0、CNN设为0.5到2.0。余弦退火配合线性预热可将后期收敛更平滑,批归一化使达到相同精度所需迭代减少约14倍。排查应遵循先可视化损失曲线再检查梯度范数的路径。

阅读全文