一文看懂从 RNN 到 LSTM_人工智能

作者 | 陈诚

一文看懂从 RNN 到 LSTM_人工智能_02

这是在看了台大李宏毅教授的深度学习视频之后的一点总结和感想。看完介绍的第一部分 RNN 尤其 LSTM 的介绍之后,整个人醍醐灌顶。本篇博客就是对视频的一些记录加上了一些个人的思考。

 

   0. 从 RNN 说起

循环神经网络(Recurrent Neural Network,RNN)是一种用于处理序列数据的神经网络。相比一般的神经网络来说,他能够处理序列变化的数据。比如某个单词的意思会因为上文提到的内容不同而有不同的含义,RNN 就能够很好地解决这类问题。

   1. 普通 RNN

先简单介绍一下一般的 RNN。其主要形式如下图所示(图片均来自台大李宏毅教授的PPT):

一文看懂从 RNN 到 LSTM_深度学习_03

这里:

一文看懂从 RNN 到 LSTM_人工智能_04 为当前状态下数据的输入, 一文看懂从 RNN 到 LSTM_人工智能_05 表示接收到的上一个节点的输入。

一文看懂从 RNN 到 LSTM_深度学习_06 为当前节点状态下的输出,而 一文看懂从 RNN 到 LSTM_深度学习_07 为传递到下一个节点的输出。

通过上图的公式可以看到,输出 h' 与 x 和 h 的值都相关。而 y 则常常使用 h' 投入到一个线性层(主要是进行维度映射)然后使用softmax进行分类得到需要的数据。

对这里的 y 如何通过 h' 计算得到往往看具体模型的使用方式。通过序列形式的输入,我们能够得到如下形式的 RNN。

一文看懂从 RNN 到 LSTM_深度学习_08

   2. LSTM

2.1 什么是 LSTM

长短期记忆(Long short-term memory, LSTM)是一种特殊的 RNN,主要是为了解决长序列训练过程中的梯度消失和梯度爆炸问题。简单来说,就是相比普通的 RNN,LSTM 能够在更长的序列中有更好的表现。

LSTM 结构(图右)和普通 RNN 的主要输入输出区别如下所示。

一文看懂从 RNN 到 LSTM_人工智能_09

相比RNN只有一个传递状态 一文看懂从 RNN 到 LSTM_深度学习_10 ,LSTM有两个传输状态,一个 一文看懂从 RNN 到 LSTM_深度学习_11 (cell state),和一个 一文看懂从 RNN 到 LSTM_深度学习_10(hidden state)。(Tips:RNN 中的 一文看懂从 RNN 到 LSTM_深度学习_10 对于 LSTM 中的 一文看懂从 RNN 到 LSTM_深度学习_11 )

其中对于传递下去的 一文看懂从 RNN 到 LSTM_深度学习_11 改变得很慢,通常输出的 一文看懂从 RNN 到 LSTM_深度学习_11 是上一个状态传过来的 一文看懂从 RNN 到 LSTM_深度学习_17 加上一些数值。

而 一文看懂从 RNN 到 LSTM_深度学习_10 则在不同节点下往往会有很大的区别。

2.2 深入 LSTM 结构

下面具体对 LSTM 的内部结构来进行剖析。

首先使用 LSTM 的当前输入 一文看懂从 RNN 到 LSTM_人工智能_19 和上一个状态传递下来的 一文看懂从 RNN 到 LSTM_深度学习_20 拼接训练得到四个状态。

一文看懂从 RNN 到 LSTM_深度学习_21一文看懂从 RNN 到 LSTM_深度学习_22

其中, 一文看懂从 RNN 到 LSTM_深度学习_23 , 一文看懂从 RNN 到 LSTM_深度学习_24 ,一文看懂从 RNN 到 LSTM_深度学习_25 是由拼接向量乘以权重矩阵之后,再通过一个 一文看懂从 RNN 到 LSTM_人工智能_26 激活函数转换成0到1之间的数值,来作为一种门控状态。而 一文看懂从 RNN 到 LSTM_深度学习_27 则是将结果通过一个 一文看懂从 RNN 到 LSTM_深度学习_28 激活函数将转换成-1到1之间的值(这里使用 一文看懂从 RNN 到 LSTM_深度学习_28 是因为这里是将其做为输入数据,而不是门控信号)。

下面开始进一步介绍这四个状态在 LSTM 内部的使用。(敲黑板)

一文看懂从 RNN 到 LSTM_深度学习_30

一文看懂从 RNN 到 LSTM_深度学习_31 是 Hadamard Product,也就是操作矩阵中对应的元素相乘,因此要求两个相乘矩阵是同型的。 一文看懂从 RNN 到 LSTM_人工智能_32 则代表进行矩阵加法。

LSTM 内部主要有三个阶段:

1. 忘记阶段。这个阶段主要是对上一个节点传进来的输入进行选择性忘记。简单来说就是会 “忘记不重要的,记住重要的”。

具体来说是通过计算得到的 一文看懂从 RNN 到 LSTM_深度学习_23 (f表示forget)来作为忘记门控,来控制上一个状态的 一文看懂从 RNN 到 LSTM_深度学习_17 哪些需要留哪些需要忘。

2. 选择记忆阶段。这个阶段将这个阶段的输入有选择性地进行“记忆”。主要是会对输入 一文看懂从 RNN 到 LSTM_人工智能_19 进行选择记忆。哪些重要则着重记录下来,哪些不重要,则少记一些。当前的输入内容由前面计算得到的 一文看懂从 RNN 到 LSTM_深度学习_27 表示。而选择的门控信号则是由 一文看懂从 RNN 到 LSTM_深度学习_24 (i 代表 information)来进行控制。

将上面两步得到的结果相加,即可得到传输给下一个状态的  一文看懂从 RNN 到 LSTM_深度学习_11  。也就是上图中的第一个公式。

3. 输出阶段。这个阶段将决定哪些将会被当成当前状态的输出。主要是通过 一文看懂从 RNN 到 LSTM_深度学习_25 来进行控制的。并且还对上一阶段得到的 一文看懂从 RNN 到 LSTM_深度学习_40 进行了放缩(通过一个 tanh 激活函数进行变化)。

与普通 RNN 类似,输出 一文看懂从 RNN 到 LSTM_人工智能_41 往往最终也是通过 一文看懂从 RNN 到 LSTM_深度学习_10 变化得到。

   3. 总结

以上,就是 LSTM 的内部结构。通过门控状态来控制传输状态,记住需要长时间记忆的,忘记不重要的信息;而不像普通的 RNN 那样只能够“呆萌”地仅有一种记忆叠加方式。对很多需要“长期记忆”的任务来说,尤其好用。

但也因为引入了很多内容,导致参数变多,也使得训练难度加大了很多。因此很多时候我们往往会使用效果和 LSTM 相当但参数更少的 GRU 来构建大训练量的模型。

 

一文看懂从 RNN 到 LSTM_深度学习_43