[深刻浅出] LSTM神经网络

由来git

人类并非每时每刻都从一片空白的大脑开始他们的思考。在你阅读这篇文章时候,你都是基于本身已经拥有的对先前所见词的理解来推断当前词的真实含义。咱们不会将全部的东西都所有丢弃,而后用空白的大脑进行思考。咱们的思想拥有持久性。
github


传统的神经网络并不能作到这点,看起来也像是一种巨大的弊端。例如,假设你但愿对电影中的每一个时间点的时间类型进行分类。传统的神经网络应该很难来处理这个问题——使用电影中先前的事件推断后续的事件。算法


RNN 解决了这个问题。RNN 是包含循环的网络,容许信息的持久化。微信


RNN 的关键点之一就是他们能够用来链接先前的信息到当前的任务上,例如使用过去的视频段来推测对当前段的理解。不幸的是,当时间间隔不断增大时,RNN 会丧失学习到链接如此远的信息的能力。网络


幸运的是,LSTM 并无这个问题!dom

——摘自简书<[译]理解LSTM>机器学习

LSTM思路函数

长短时记忆网络(Long Short Term Memory Network, LSTM),是一种改进以后的循环神经网络,能够解决RNN没法处理长距离的依赖的问题,目前比较流行。工具

 

长短时记忆网络的思路:性能

原始 RNN 的隐藏层只有一个状态,即h,它对于短时间的输入很是敏感。

再增长一个状态,即c,让它来保存长期的状态,称为单元状态(cell state)

640?wx_fmt=png&wxfrom=5&wx_lazy=1

把上图按照时间维度展开:

0?wx_fmt=png

在 t 时刻,LSTM 的输入有三个:当前时刻网络的输入值 x_t、上一时刻 LSTM 的输出值 h_t-1、以及上一时刻的单元状态 c_t-1
LSTM 的输出有两个:当前时刻 LSTM 输出值 h_t、和当前时刻的单元状态 c_t.


关键问题是:怎样控制长期状态 c ?

方法是:使用三个控制开关

0?wx_fmt=png

第一个开关,负责控制继续保存长期状态c;
第二个开关,负责控制把即时状态输入到长期状态c;
第三个开关,负责控制是否把长期状态c做为当前的LSTM的输出。


如何在算法中实现这三个开关?

方法:用 门(gate)

定义:gate 实际上就是一层全链接层,输入是一个向量,输出是一个 0到1 之间的实数向量。
公式为:

0?wx_fmt=png

回忆一下它的样子:

0?wx_fmt=png



gate 如何进行控制?

方法:用门的输出向量按元素乘以咱们须要控制的那个向量
原理:门的输出是 0到1 之间的实数向量,
当门输出为 0 时,任何向量与之相乘都会获得 0 向量,这就至关于什么都不能经过;
输出为 1 时,任何向量与之相乘都不会有任何改变,这就至关于什么均可以经过。


LSTM前向计算


在 LSTM-1 中提到了,模型是经过使用三个控制开关来控制长期状态 c 的:

0?wx_fmt=png

这些开关就是用门(gate)来实现:

0?wx_fmt=png

接下来具体看这三重门

LSTM 的前向计算(一共有 6 个公式)

 
 

遗忘门(forget gate)
它决定了上一时刻的单元状态 c_t-1 有多少保留到当前时刻 c_t

输入门(input gate)
它决定了当前时刻网络的输入 x_t 有多少保存到单元状态 c_t

输出门(output gate)
控制单元状态 c_t 有多少输出到 LSTM 的当前输出值 h_t

遗忘门的计算为

0?wx_fmt=png


遗忘门的计算公式中:
W_f 是遗忘门的权重矩阵,[h_t-1, x_t] 表示把两个向量链接成一个更长的向量,b_f 是遗忘门的偏置项,σ 是 sigmoid 函数。



输入门的计算:

0?wx_fmt=png

根据上一次的输出和本次输入来计算当前输入的单元状态:

0?wx_fmt=png

当前输入的单元状态c_t的计算:由上一次的单元状态 c_t-1 按元素乘以遗忘门 f_t,再用当前输入的单元状态 c_t 按元素乘以输入门 i_t,再将两个积加和:

这样,就能够把当前的记忆 c_t 和长期的记忆 c_t-1 组合在一块儿,造成了新的单元状态 c_t

因为遗忘门的控制,它能够保存好久好久以前的信息,因为输入门的控制,它又能够避免当前可有可无的内容进入记忆。

0?wx_fmt=png



输出门的计算:

0?wx_fmt=png



LSTM反向传播

主要有三步:

1. 前向计算每一个神经元的输出值,一共有 5 个变量,计算方法就是前一部分:

0?wx_fmt=png


2. 反向计算每一个神经元的偏差项值。与 RNN 同样,LSTM 偏差项的反向传播也是包括两个方向:
一个是沿时间的反向传播,即从当前 t 时刻开始,计算每一个时刻的偏差项;
一个是将偏差项向上一层传播。

3. 根据相应的偏差项,计算每一个权重的梯度。



gate 的激活函数定义为 sigmoid 函数,输出的激活函数为 tanh 函数,导数分别为:

0?wx_fmt=png

具体推导公式为:

0?wx_fmt=png

0?wx_fmt=png

具体推导公式为:

0?wx_fmt=png


调参

目标是要学习 8 组参数,以下图所示:

0?wx_fmt=png

又权重矩阵 W 都是由两个矩阵拼接而成,这两部分在反向传播中使用不一样的公式,所以在后续的推导中,权重矩阵也要被写为分开的两个矩阵。

接着就来求两个方向的偏差,和一个梯度计算。


1. 偏差项沿时间的反向传递:

定义 t 时刻的偏差项:

0?wx_fmt=png

目的是要计算出 t-1 时刻的偏差项:

0?wx_fmt=png

利用 h_t c_t 的定义,和全导数公式,能够获得 将偏差项向前传递到任意k时刻的公式:

0?wx_fmt=png


2. 将偏差项传递到上一层的公式:

0?wx_fmt=png


3. 权重梯度的计算:

0?wx_fmt=png

以上就是 LSTM 的训练算法的所有公式。

 
  

来自 LSTM Hyperparameter Tuning:
https://deeplearning4j.org/lstm

还有一个用 LSTM 作 text_generation 的例子

https://github.com/fchollet/keras/blob/master/examples/lstm_text_generation.py

学习资料:
https://zybuluo.com/hanbingtao/note/581764
原文连接:
https://www.jianshu.com/p/dcec3f07d3b5


 近期热文

Tomas Mikolov论文简评:从Word2Vec到FastText

推荐 | 中文文本标注工具Chinese-Annotator

资源 | 2017年GitHub中Top 30开源机器学习项目

天然语言对话引擎(技术类)

AdaBoost元算法如何提升分类性能——机器学习实战

奇异值分解(SVD)原理

分享 | 由0到1走入Kaggle-入门指导 (长文、干货)

常见文本类似度量方法总结

干货|免费文本语料训练数据集

Experience is the mother of wisdom

经验是智慧之母


0?wx_fmt=jpeg

更多干货内容请关注微信公众号“AI 深刻浅出”

长按二维码关注