第六章 循环神经网络
循环神经网络(Recurrent Neural Network,RNN)是一类具有短期记忆能力的神经网络。循环神经网络的参数学习可以通过随时间反向传播算法来学习。随时间反向传播算法即按照时间的逆序将错误信息一步步地往前传递。当输入序列比较长时,会存在梯度爆炸和消失问题,也称为长程依赖问题。为了解决这个问题,人们对循环神经网络进行了很多的改进,其中最有效的改进方式引入门控机制。
循环神经网络可以很容易地扩展到两种更广义的记忆网络模型:递归神经网络和图网络。
给网络增加记忆能力
延时神经网络
一种简单的利用历史信息的方法是建立一个额外的延时单元,用来存储网络的历史信息(可以包括输入、输出、隐状态等)。比较有代表性的模型是延时神经网络
延时神经网络在时间维度上共享权值,以降低参数数量。因此对于序列输入来讲,延时神经网络就相当于卷积神经网络。
实现方式:延时神经网络是在前馈网络中的非输出层都添加一个延时器,记录最近几次神经元的输出。在第t 个时刻,第l + 1 层神经元和第l 层神经元的最近p 次输出相关,即:,通过延时器,前馈网络就具有了短期记忆的能力。
有外部输入的非线性自回归模型
自回归模型(AutoRegressive Model,AR)是统计学上常用的一类时间序列模型,用一个变量的历史信息来预测自己。
其中p 为超参数,为可学习参数,为第t 个时刻的噪声,方差和时间无关。
有外部输入的非线性自回归模型(NARX)是自回归模型的扩展,在每个时刻t 都有一个外部输入xt,产生一个输出yt。NARX通过一个延时器记录最近几次的外部输入和输出,第t 个时刻的输出yt 为:
其中f(·) 表示非线性函数,可以是一个前馈网络,p 和q 为超参数。
循环神经网络
循环神经网络(Recurrent Neural Network,RNN)通过使用带自反馈的神经元,能够处理任意长度的时序数据。
给定一个输入序列,循环神经网络通过公式更新带反馈边的隐藏层的活性值ht。其中h0 = 0,f(·) 为一个非线性函数,也可以是一个前馈网络。
理论上,循环神经网络可以近似任意的非线性动力系统。
简单循环网络
简单循环网络(Simple Recurrent Network,SRN) 是一个非常简单的循环神经网络,只有一个隐藏层的神经网络。
循环神经网络的计算能力
由于循环神经网络具有短期记忆能力,相当于存储装置,因此其计算能力十分强大。前馈神经网络可以模拟任何连续函数,而循环神经网络可以模拟任何程序。
我们先定义一个完全连接的循环神经网络,其输入为xt,输出为yt,
循环神经网络的通用近似定理
一个完全连接的循环网络是任何非线性动力系统的近似器。
图灵完备
是指一种数据操作规则,比如一种计算机编程语言,可以实现图灵机(Turing Machine)的所有功能,解决所有的可计算问题,目前主流的编程语言(比如C++、Java、Python 等)都是图灵完备的。
一个完全连接的循环神经网络可以近似解决所有的可计算问题。
应用到机器学习
序列到类别模式
序列到类别模式主要用于序列数据的分类问题:输入为序列,输出为类别。比如在文本分类中,输入数据为单词的序列,输出为该文本的类别。
同步的序列到序列模式
主要用于序列标注(Sequence Labeling)任务,即每一时刻都有输入和输出,输入序列和输出序列的长度相同。比如词性标注(Partof-Speech Tagging)中,每一个单词都需要标注其对应的词性标签。
异步的序列到序列模式
也称为编码器-解码器(Encoder-Decoder)模型,即输入序列和输出序列不需要有严格的对应关系,也不需要保持相同的长度。比如在机器翻译中,输入为源语言的单词序列,输出为目标语言的单词序列。
参数学习
可以通过梯度下降方法来进行学习。
以随机梯度下降为例,给定一个训练样本(x, y),其中为长度是T 的输入序列,是长度为T 的标签序列。即在每个时刻t,都有一个监督信息yt,我们定义时刻t 的损失函数为:,其中g(ht) 为第t 时刻的输出,为可微分的损失函数,比如交叉熵。那么整个序
列的损失函数为:,整个序列的损失函数关于参数U 的梯度为:,即每个时刻损失对参数U 的偏导数之和。
在循环神经网络中主要有两种计算梯度的方式:随时间反向传播(BPTT)算法和实时循环学习(RTRL)算法。
随时间反向传播算法
BPTT算法的主要思想是通过类似前馈神经网络的错误反向传播算法来计算梯度。
BPTT 算法将循环神经网络看作是一个展开的多层前馈网络,其中“每一层”对应循环网络中的“每个时刻”。这样,循环神经网络就可以按照前馈网络中的反向传播算法计算参数梯度。在“展开”的前馈网络中,所有层的参数是共享的,因此参数的真实梯度是所有“展开层”的参数梯度之和。
计算第t 时刻损失对参数U 的偏导数:
所以整个序列的损失函数关于参数U的梯度是:
同理可得,整个序列的损失函数关于权重W 和偏置b 的梯度为:和
实时循环学习算法
与反向传播的BPTT算法不同的是,实时循环学习(Real-Time Recurrent Learning,RTRL)是通过前向传播的方式来计算梯度。
假设循环神经网络中第t + 1时刻的状态ht+1为:
其关于参数的偏导数为:
两种算法比较:RTRL算法和BPTT算法都是基于梯度下降的算法,分别通过前向模式和反向模式应用链式法则来计算梯度。在循环神经网络中,一般网络输出维度远低于输入维度,因此BPTT算法的计算量会更小,但是BPTT算法需要保存所有时刻的中间梯度,空间复杂度较高。RTRL算法不需要梯度回传,因此非常适合用于需要在线学习或无限序列的任务中。
长程依赖问题
循环神经网络在学习过程中的主要问题是由于梯度消失或爆炸问题,很难建模长时间间隔(Long Range)的状态之间的依赖关系。
在BPTT算法中,将公式展开得到
如果定义,则
若γ > 1,当t − k → ∞时,γt−k → ∞。当间隔t − k 比较大时,梯度也变得很大,会造成系统不稳定,称为梯度爆炸问题。
相反,若γ < 1,当t − k → ∞时,γt−k → 0。当间隔t − k 比较大时,梯度也变得非常小,会出现和深层前馈神经网络类似的梯度消失问题,这就是长程依赖问题
由于循环神经网络经常使用非线性激活函数为Logistic函数或Tanh函数作为非线性激活函数,其导数值都小于1,并且权重矩阵∥U∥ 也不会太大,因此如果时间间隔t − k 过大,δt,k 会趋向于0,因而经常会出现梯度消失问题。
改进方案
为了避免梯度爆炸或消失问题,一种最直接的方式就是选取合适的参数,同时使用非饱和的激活函数,尽量使得 ≈ 1,这种方式需要足够的人工调参经验,限制了模型的广泛应用。比较有效的方式是通过改进模型或优化方法来缓解循环网络的梯度爆炸和梯度消失问题。
梯度爆炸一般而言,循环网络的梯度爆炸问题比较容易解决,一般通过权重衰减或梯度截断(是一种启发式的解
决梯度爆炸问题的有效方法)来避免。
梯度消失是循环网络的主要问题。除了使用一些优化技巧外,更有效的方式就是改变模型,比如让,同时令为单位矩阵,即:,其中是一个非线性函数,θ为参数。
可以看出是线性依赖关系,且权重系数为1,这样就不存在梯度爆炸或消失问题。但是,这种改变也丢失了神经元在反馈边上的非线性激活的性质,因此也降低了模型的表示能力。
为了避免这个缺点,我们可以采用一种更加有效的改进策略:
这样之间既有线性关系,也有非线性关系,并且可以缓解梯度消失问题。但这种改进依然存在两个问题:
- 梯度爆炸问题,因为其中存在非线性关系
- 记忆容量问题,:随着ht不断累积存储新的输入信息,会发生饱和现象。假设g(·) 为Logistic 函数,则随着时间t 的增长,ht 会变得越来越大,从而导致h变得饱和。也就是说,隐状态ht 可以存储的信息是有限的,随着记忆单元存储的内容越来越多,其丢失的信息也越来越多。
基于门控的循环神经网络
为了改善循环神经网络的长程依赖问题,一种非常好的解决方案是在上一个改进方案的的基础上引入门控机制来控制信息的累积速度,包括有选择地加入新的信息,并有选择地遗忘之前累积的信息。这一类网络可以称为基于门控的循环神经网络(Gated RNN)。本节中,主要介绍两种基于门控的循环神经网络:长短期记忆网络和门控循环单元网络。
长短期记忆网络(LSTM)
是循环神经网络的一个变体,可以有效地解决简单循环神经网络的梯度爆炸或消失问题。
LSTM的改进点在两个方面:
LSTM网络引入了一个新的内部状态ct专门进行线性的循环信息传递,同时(非线性地)输出信息给隐藏层的外部状态ht。
门控机制:LSTM网络中的三个“门”是一种“软”门,取值在(0, 1) 之间,表示以一定的比例运行信息通过。
遗忘门ft 控制上一个时刻的内部状态ct−1 需要遗忘多少信息。
输入门it 控制当前时刻的候选状态 有多少信息需要保存。
输出门ot 控制当前时刻的内部状态ct 有多少信息需要输出给外部状态ht。
其中σ(·) 为Logistic 函数,其输出区间为(0, 1)
上述公式可简洁的描述为:
记忆循环神经网络中的隐状态h存储了历史信息,可以看作是一种记忆(Memory)。在简单循环网络中,隐状态每个时刻都会被重写,因此可以看作是一种短期记忆(Short-Term Memory)。在神经网络中,长期记忆(Long-Term Memory)可以看作是网络参数,隐含了从训练数据中学到的经验,其更新周期要远远慢于短期记忆。而在LSTM 网络中,记忆单元c 可以在某个时刻捕捉到某个关键信息,并有能力将此关键信息保存一定的时间间隔。记忆单元c 中保存信息的生命周期要长于短期记忆h,但又远远短于长期记忆, 长短期记忆是指长的“短期记忆”。因此称为长短期记忆。
LSTM网络的各种变体
无遗忘门的LSTM网络:即最早提出的LSTM网络,其内部状态的更新为:
没有遗忘门,记忆单元c 会不断增大。当输入序列的长度非常大时,记忆单元的容量会饱和,从而大大降低LSTM模型的性能。
peephole 连接:让三个门不但依赖于输入xt 和上一时刻的隐状态ht−1,也依赖于上一个时刻的记忆单元ct−1。
耦合输入门和遗忘门:LSTM网络中的输入门和遗忘门有些互补关系,因此同时用两个门比较冗余。为了减少LSTM网络的计算复杂度,将这两门合并为一个门。令
这样,内部状态的更新方式为:
门控循环单元网络(GRU)
GRU是是一种比LSTM网络更加简单的循环神经网络。
GRU网络引入门控机制来控制信息更新的方式。和LSTM 不同,GRU不引入额外的记忆单元,GRU 网络也是在公式的基础上引入一个更新门(Update Gate)来控制当前状态需要从历史状态中保留多少信息(不经过非线性变换),以及需要从候选状态中接受多少新信息。
其中为更新门,
在LSTM网络中,输入门和遗忘门是互补关系,具有一定的冗余性。GRU网络直接使用一个门来控制输入和遗忘之间的平衡。当zt = 0 时,当前状态ht 和前一时刻的状态ht−1 之间为非线性函数关系;当zt = 1 时,ht 和ht−1 之间为线性函数关系。
在GRU网络中,函数g(xt,ht−1; θ) 的定义为:,(这里使用tanh 激活函数是
由于其导数有比较大的值域,能够缓解梯度消失问题。)
其中表示当前时刻的候选状态,为重置门,用来控制候选状态的计算是否依赖上一时刻的状态
当rt = 0 时,候选状态 只和当前输入xt 相关,和历史状态无关。当rt = 1 时,候选状态和当前输入xt 以及历史状态ht−1 相关,和简单循环网络一致。
综上,GRU网络的状态更新方式为:
可以看出,当zt = 0, r = 1 时,GRU网络退化为简单循环网络;若zt =0, r = 0 时,当前状态ht 只和当前输入xt相关,和历史状态ht-1 无关。当zt = 1时,当前状态ht = ht-1 等于上一时刻状态ht−1,和当前输入xt 无关。
深层循环神经网络
如果是长时间序列的神经网络,按时间片段展开,即的路径是很长的,可以说这是个很“深”的神经网络,但是不按时间片段展开,只看同一时刻网络输入到输出的路径,即,这又是一个很“浅”的网络。
深层循环神经网络是通过增加或者之间的路径,从而增强循环神经网络的能力。
堆叠循环神经网络(SRNN)
就是将多个循环网络堆叠起来。一个堆叠的简单循环网络(Stacked SRN)也称为循环多层感知器(RMLP)。
双向循环神经网络
假设第1 层按时间顺序,第2 层按时间逆序,在时刻t 时的隐状态定义为和,则:
扩展到图结构
因为同一层级的隐状态,在时间链中,可以构成一条链路,而链式结构是一种特殊的图结构,我们可以比较容易地将这种消息传递(Message Passing)的思想扩展到任意的图结构上。
递归神经网络(RecNN)
是循环神经网络在有向无循环图上的扩展。递归神经网络的一般结构为树状的层次结构。
以图a中的结构为例,有三个隐藏层h1、h2 和h3,其中h1 由两个输入x1和x2 计算得到,h2 由另外两个输入层x3 和x4 计算得到,h3 由两个隐藏层h1 和h2 计算得到。
对于一个节点hi,它可以接受来自父节点集合πi 中所有节点的消息,并更新自己的状态:
其中表示集合中所有节点状态的拼接,是一个和节点位置无关的非线性函数,可以为一个单层的前馈神经网络。比如图a所示的递归神经网络具体可以写为:
其中,表示非线性激活函数,W和b是可学习的参数。
同样,输出层y可以为一个分类器,比如:
图网络
图网络(Graph Network,GN)是将消息传递的思想扩展到图结构数据上的神经网络。