如何缓解梯度消失和梯度膨胀(微调、梯度截断、改良激活函数等)
蓝海大脑深度学习液冷工作站研究人员表示:
简言之,引起梯度消失和梯度爆炸的原因分别是:梯度消失:根据链式法则,如果每一层神经元对上一层的输出的偏导乘上权重结果都小于1的话,那么即使这个结果是0.99,在经过足够多层传播之后,误差对输入层的偏导会趋于0
可以采用ReLU激活函数有效的解决梯度消失的情况,也可以用Batch Normalization解决这个问题。关于深度学习中 Batch Normalization为什么效果好?参见:深度学习中 Batch Normalization为什么效果好?
梯度膨胀:根据链式法则,如果每一层神经元对上一层的输出的偏导乘上权重结果都大于1的话,在经过足够多层传播之后,误差对输入层的偏导会趋于无穷大
可以通过激活函数来解决,或用Batch Normalization解决这个问题。
蓝海大脑 京ICP备18017748号-1