深度学习面试题38:LSTM如何解决梯度消失问题

目录

  回顾简单RNN的梯度消失问题

  LSTM如何解决梯度消失

  遗忘门对梯度消失的影响

  遗忘门的初始化技巧

  参考资料


 

回顾简单RNN的梯度消失问题

在简单RNN的前向传播过程中,输入的数据循环地与隐藏层里的权重W(都是很小的数)做乘法运算,那么损失函数对较长时间步前的W的梯度就会很小(因为W会累乘,激活函数大多也是小数),详细内容见【深度学习面试题35:RNN梯度消失问题(vanishing gradient)】。

 返回目录

 

LSTM如何解决梯度消失

 

在【深度学习面试题35:RNN梯度消失问题(vanishing gradient)】中有一个结论,即简单RNN越早的时间步梯度消失的问题越严重;而这一现象在LSTM不太会发生(注意这里的用词,是不太会)。下面来分析原理。

以上图为例,我们假设有三个时间步,最后一个时间步产生输出,并且遗忘门、输入门和输出门的输出都是1,那么整个前馈过程的计算如图所示。

下面以最复杂的输入门为例,演示为什么梯度不太会消失。反向求导结果如下

可以看到,损失函数对每个时间步的梯度都是一样的,然后根据链式求导发展,每个时间步继续对输入门里的参数求梯度,结果可想而知,每个时间步下的输入门的梯度都比较大,所以就没有简单RNN梯度消失的问题。

遗忘门和输出门同理。

 返回目录

 

遗忘门对梯度消失的影响

在上一节我们假设了遗忘门的输出为1。

如果遗忘门的输出为1,也就是不对记忆细胞C进行遗忘。

如果遗忘门的输出小于1,那么,越早的时间步的梯度就会越小,可能出现梯度消失。

 返回目录

 

遗忘门的初始化技巧

一般而言为了避免训练LSTM因梯度消失而训练缓慢的问题,可以将输入门的偏置初始化大一些,这样就可以保证遗忘门的输出接近于1,就可以高效地训练网络了。

 返回目录

 

参考资料

https://www.bilibili.com/video/BV15b411g7Wd?p=44

 返回目录

 

posted @ 2020-07-16 00:22  黎明程序员  阅读(2098)  评论(0编辑  收藏  举报