LSTM 长短期记忆递归神经网络

1、神经网络简介

1.1 神经网络起源

人工���经网络（Aritificial Neural Networks, ANN）是一种仿生的网络结构，起源于对人类大脑的研究。人工神经网络（Aritificial Neural Networks）也常被简称为神经网络（Neural Networks, NN），基本思想是通过大量简单的神经元之间的相互连接来构造复杂的网络结构，信号（数据）可以在这些神经元之间传递，通过激活不同的神经元和对传递的信号进行加权来使得信号被放大或衰减，经过多次的传递来改变信号的强度和表现形式。

1.2 传统神经网络原理

传统神经网络结构示意图：

其中：

• 输入层：可以包含多个神经元，可以接收多维的信号输入（特征信息）；
• 输出层：可以包含多个神经元，可以输出多维信号；
• 隐含层：可以包含多个神经网络层，每一层包含多个神经元。

  每层的神经元与上一层神经元和下一层神经元连接（类似生物神经元的突触），这些连接通路用于信号传递。每个神经元接收来自上一层的信号输入，使用一定的加和规则将所有的信号输入汇聚到一起，并使用激活函数将输入信号激活为输出信号，再将信号传递到下一层。

  影响神经网络表现能力的主要因素有神经网络的层数、神经元的个数、神经元之间的连接方式以及神经元所采用的激活函数。神经元之间以不同的连接方式（全连接、部分连接）组合，可以构成不同神经网络，对于不同的信号处理效果也不一样。

  神经网络优点很明显，给我们提供了构建模型的便利，你大可不用顾及模型本身是如何作用的，只需要按照规则构建网络，然后使用训练数据集不断调整参数，在许多问题上都能得到一个比较“能接受”的结果，然而我们对其中发生了什么是未可知的。在深度学习领域，许多问题都可以通过构建深层的神经网络模型来解决。

  1.3 传统神经网络结构的缺陷

  从传统的神经网络结构我们可以看出，信号流从输入层到输出层依次流过，同一层级的神经元之间，信号是不会相互传递的。这样就会导致一个问题，输出信号只与输入信号有关，而与输入信号的先后顺序无关。并且神经元本身也不具有存储信息的能力，整个网络也就没有“记忆”能力，当输入信号是一个跟时间相关的信号时，如果我们想要通过这段信号的“上下文”信息来理解一段时间序列的意思，传统的神经网络结构就显得无力了。与我们人类的理解过程类似，我们听到一句话时往往需要通过这句话中词语出现的顺序以及我们之前所学的关于这些词语的意思来理解整段话的意思，而不是简单的通过其中的几个词语来理解。

  例如，在自然语言处理领域，我们要让神经网络理解这样一句话：“地球上最高的山是珠穆朗玛峰”，按照传统的神经网络结构，它可能会将这句话拆分为几个单独的词（地球、上、最高的、山、是、珠穆朗玛峰），分别输入到模型之中，而不管这几个词之间的顺序。然而，直观上我们可以看到，这几个词出现的顺序是与最终这句话要表达的意思是密切相关的，但传统的神经网络结构无法处理这种情况。

  因此，我们需要构建具有“记忆”能力的神经网络模型，用来处理需要理解上下文意思的信号，也就是时间序列数据。循环神经网络（RNN）就是用来处理这类信号的，RNN之所以能够有效的处理时间序列数据，主要是基于它比较特殊的运行原理。

  2 循环神经网络RNN

  2.1 RNN 的构造过程

  RNN是一种特殊的神经网路结构，其本身是包含循环的网络，允许信息在神经元之间传递：

  

  图示是一个 RNN 结构示意图，图中的 A A A 表示神经网络模型， X t X_t Xt​ 表示模型的输入信号， h t h_t ht​ 表示模型的输出信号。如果去掉 A 的输出信号传递到 A 的箭头，这个网络模型与普通的神经网络结构相同。 A 的输出信号传递到 A 的箭头，允许 A 将信息传递给 A，神经网络将自己的输出作为输入。

  在输入信号是一个时间序列信号的时候，输入信号和时间 t 相关。在 t 时刻，输入信号 X t X_t Xt​ 作为神经网络 A 的输入， A 的输出分流成两部分，一部分输出给 h t h_t ht​ ，一部分作为一个隐藏信号流被输入到 A 中，在下一时刻输入信号 X t + 1 X_{t+1} Xt+1​ 时，这部分隐藏分信号流也被作为输入信号输入到了 A 中。此时神经网络 A 就同时接受了 t 时刻和 t+1 时刻的信号输入了，此时的输出信号又将被传递到下一时刻的 A 中，把上面那个图根据时间 t 展开：

  

  如上图所示， t = 0 t=0 t=0 时刻的输出给 t = 1 t=1 t=1 时刻的模型 A 作为输入， t = 1 t=1 t=1 时刻的输出给 t = 2 t=2 t=2 时刻的模型 A 作为 输入……。相当于 RNN 在时间序列上把自己复制了很多遍，每个模型都对应一个时刻的输入，并且当前时刻的输出还作为下一时刻的模型的输入信号。

  这样链式的结构揭示了RNN本质上是与序列相关的，是对于时间序列数据最自然的神经网络架构。并且理论上，RNN可以保留以前任意时刻的信息。

  2.2 RNN 的局限性

  RNN利用了神经网络的“内部循环”来保留时间序列的上下文信息，可以使用过去的信号数据来推测对当前信号的理解，这是非常重要的进步，并且理论上RNN可以保留过去任意时刻的信息。但实际使用RNN时往往遇到问题。

  假如我们构造了一个语言模型，可以通过当前这一句话的意思来预测下一个词语。现在有这样一句话：“我是一个中国人，出生在普通家庭，我最常说汉语，也喜欢写汉字。我喜欢妈妈做的菜”。我们的语言模型在预测“我最常说汉语”的“汉语”这个词时，它要预测“我最长说”这后面可能跟的是一个语言，可能是英语，也可能是汉语，那么它需要用到第一句话的“我是中国人”这段话的意思来推测我最常说汉语，而不是英语、法语等。而在预测“我喜欢妈妈做的菜”的最后的词“菜”时并不需要“我是中国人”这个信息以及其他的信息，它跟我是不是一个中国人没有必然的关系。

  在这个例子中可以看出，想要精确地处理时间序列，有时候我们只需要用到最近的时刻的信息。例如预测“我喜欢妈妈做的菜”最后这个词“菜”，此时信息传递是这样的：

  

  “菜”这个词与“我”、“喜欢”、“妈妈”、“做”、“的”这几个词关联性比较大，距离也比较近，所以可以直接利用这几个词进行最后那个词语的推测。

  有时候我们又需要用到很早以前时刻的信息，例如预测“我最常说汉语”最后的这个词“汉语”。此时信息传递是这样的：

  

  此时，我们要预测“汉语”这个词，仅仅依靠“我”、“最”、“常”、“说”这几个词还不能得出我说的是汉语，必须要追溯到更早的句子“我是一个中国人”，由“中国人”这个词语来推测我最常说的是汉语。因此，这种情况下，我们想要推测“汉语”这个词的时候就比前面那个预测“菜”这个词所用到的信息就处于更早的时刻。

  RNN虽然在理论上可以保留所有历史时刻的信息，但在实际使用时，信息的传递往往会因为时间间隔太长而逐渐衰减，传递一段时刻以后其信息的作用效果就大大降低了。因此，普通RNN对于信息的长期依赖问题没有很好的处理办法。

  为了克服这个问题，Hochreiter等人在1997年改进了RNN，提出了一种特殊的RNN模型——LSTM网络，可以学习长期依赖信息，在后面的20多年被改良和得到了广泛的应用，并且取得了极大的成功。

  3 长短时间记忆网络（LSTM）

  3.1 LSTM 与 RNN 的关系

  长短期记忆（Long Short Term Memory，LSTM）网络是一种特殊的RNN模型，其特殊的结构设计使得它可以避免长期依赖问题，记住很早时刻的信息是LSTM的默认行为，而不需要专门为此付出很大代价。

  普通的RNN模型中，其重复神经网络模块的链式模型如下图所示，这个重复的模块只有一个非常简单的结构，一个单一的神经网络层（例如tanh层），这样就会导致信息的处理能力比较低。

  

  LSTM在此基础上将这个结构改进了，不再是单一的神经网络层，而是4个，并且以一种特殊的方式进行交互。

  

  图中的模块分为以下几种：

  

  • 黄色方块：表示一个神经网络层（Neural Network Layer）；
  • 粉色圆圈：表示按位操作或逐点操作（pointwise operation），例如向量加和、向量乘积等；
  • 单箭头：表示信号传递（向量传递）；
  • 合流箭头：表示两个信号的连接（向量拼接）；
  • 分流箭头：表示信号被复制后传递到2个不同的地方。

    3.2 LSTM 基本思想

    LSTM的关键是细胞状态（直译：cell state），表示为 C t C_t Ct​，用来保存当前LSTM的状态信息并传递到下一时刻的LSTM中，也就是RNN中那根“自循环”的箭头。当前的LSTM接收来自上一个时刻的细胞状态 C t − 1 C_{t-1} Ct−1​，并与当前LSTM接收的信号输入 x t x_t xt​，共同作用产生当前LSTM的细胞状态 C t C_t Ct​。

    

    在LSTM中，采用专门设计的“门”来引入或者去除细胞状态 C t C_t Ct​ 中的信息。门是一种让信息选择性通过的方法。有的门跟信号处理中的滤波器有点类似，允许信号部分通过或者通过时被门加工了；有的门也跟数字电路中的逻辑门类似，允许信号通过或者不通过。这里所采用的门包含一个 sigmoid 神经网络层和一个按位的乘法操作，如下图所示：

    

    其中黄色方块表示sigmoid神经网络层，粉色圆圈表示按位乘法操作。sigmoid神经网络层可以将输入信号转换为 0 到 1 之间的数值，用来描述有多少量的输入信号可以通过。0 表示“不允许任何量通过”，1 表示“允许所有量通过”。sigmoid神经网络层起到类似下图的sigmoid函数所示的作用：

    

    其中，横轴表示输入信号，纵轴表示经过sigmoid函数以后的输出信号。

    LSTM主要包括三个不同的门结构：遗忘门、记忆门和输出门。这三个门用来控制LSTM的信息保留和传递，最终反映到细胞状态 C t C_t Ct​ 和输出信号 h t h_t ht​，如下图所示：

    

    图中标示了LSTM中各个门的构成情况和相互之间的关系，其中：

    • 遗忘门由一个sigmoid神经网络层和一个按位乘操作构成；
    • 记忆门由输入门（input gate）与tanh神经网络层和一个按位乘操作构成；
    • 输出门（output gate）与 tanh 函数（注意：这里不是 tanh 神经网络层）以及按位乘操作共同作用将细胞状态和输入信号传递到输出端。

      3.3 遗忘门

      顾名思义，遗忘门的作用就是用来“忘记”信息的。在LSTM的使用过程中，有一些信息不是必要的，因此遗忘门的作用就是用来选择这些信息并“忘记”它们。遗忘门决定了细胞状态 C t − 1 C_{t-1} Ct−1​ 中的哪些信息将被遗忘。

      

      • 左边高亮的结构就是遗忘门，包含一个sigmoid神经网络层（黄色方框，神经网络参数为 W f , b f W_f, b_f Wf​,bf​）
      • 接收 t 时刻的输入信号 x t x_t xt​ 和 t − 1 t-1 t−1 时刻LSTM的上一个输出信号 h t − 1 h_{t-1} ht−1​，这两个信号进行拼接以后共同输入到sigmoid神经网络层中，然后输出信号 f t f_t ft​， f t f_t ft​ 是一个 0 到1之间的数值，并与 C t − 1 C_{t-1} Ct−1​ 相乘来决定 C t − 1 C_{t-1} Ct−1​ 中的哪些信息将被保留，哪些信息将被舍弃。

        假设 C t − 1 = [ 0.5 , 0.6 , 0.4 ] , h t − 1 = [ 0.3 , 0.8 , 0.69 ] , x t = [ 0.2 , 1.3 , 0.7 ] C_{t-1} = [0.5, 0.6, 0.4], h_{t-1} = [0.3, 0.8, 0.69], x_t = [0.2, 1.3, 0.7] Ct−1​=[0.5,0.6,0.4],ht−1​=[0.3,0.8,0.69],xt​=[0.2,1.3,0.7]，遗忘门的输入信号就是 h t − 1 h_{t-1} ht−1​ 和 x t x_t xt​ 的组合，即 [ t t − 1 , x t ] = [ 0.3 , 0.8 , 0.69 , 0.2 , 1.3 , 0.7 ] [t_{t-1}, x_t] = [0.3, 0.8, 0.69, 0.2, 1.3, 0.7] [tt−1​,xt​]=[0.3,0.8,0.69,0.2,1.3,0.7]，然后通过 sigmoid 神经网络层输出每一个元素都处于0到1之间的向量 f t = [ 0.5 , 0.1 , 0.8 ] f_t = [0.5, 0.1, 0.8] ft​=[0.5,0.1,0.8]。此时 f t f_t ft​ 是一个与 C t − 1 C_{t-1} Ct−1​ 维数相同的向量，此处为3维。

        这里需要注意的是，这里的 simgmoid 是 sigmoid 神经网络层 ，而不是 sigmoid 激活函数，因此输入信号是6维的向量，而 f t f_t ft​ 可以是3维的。

        3.4 记忆门

        记忆门的作用与遗忘门相反，它将决定新输入的信息 x t x_t xt​ 和 h t − 1 h_{t-1} ht−1​ 中哪些信息将被保留。

        

        如上图所示，记忆门包含2个部分。第一个是包含sigmoid神经网络层（输入门，神经网络网络参数为 W i , b i W_i, b_i Wi​,bi​）和一个 tanh 神经网络层（神经网络参数为 W c , b c W_c, b_c Wc​,bc​）。

        • sigmoid 神经网络层的作用很明显，跟遗忘门一样，它接收 x t x_t xt​ 和 h t − 1 h_{t-1} ht−1​ 作为输入，然后输出一个 0 到 1 之间的数值 i t i_t it​ 来决定哪些信息需要被更新；
        • Tanh神经网络层的作用是将输入的 x t x_t xt​ 和 h t − 1 h_{t-1} ht−1​ 整合，然后通过一个tanh神经网络层来创建一个新的状态候选向量 C t ~ \tilde{C_t} Ct​~​， C t ~ \tilde{C_t} Ct​~​ 的值范围在 -1 到 1 之间。

          记忆门的输出由上述两个神经网络层的输出决定， i t i_t it​ 与 C t ~ \tilde{C_t} Ct​~​ 相乘来选择哪些信息将被新加入到 t 时刻的细胞状态 C t C_t Ct​ 中。

          3.5 更新细胞状态

          有了遗忘门和记忆门，就可以更新细胞状态 C t C_t Ct​ 了。

          

          这里将遗忘门的输出 f t f_t ft​ 与上一时刻的细胞状态 C t − 1 C_{t-1} Ct−1​ 相乘来选择遗忘和保留一些信息，将记忆门的输出与从遗忘门选择后的信息加和得到新的细胞状态 C t C_t Ct​。这表示 t 时刻的细胞状态 C t C_t Ct​ 包含了此时需要丢弃的 t-1 时刻传递的信息和 t 时刻从输入信号获取的需要新加入的信息 i t ⋅ C t ~ i_t \cdot \tilde{C_t} it​⋅Ct​~​。 C t C_t Ct​ 将继续传递到 t+1 时刻的LSTM网络中，作为新的细胞状态传递下去。

          3.6 输出门

          

          如上面左图所示，输出门就是将t-1时刻传递过来并经过了前面遗忘门与记忆门选择后的细胞状态 C t − 1 C_{t-1} Ct−1​，与 t-1 时刻的输出信号 h t − 1 h_{t-1} ht−1​ 和 t 时刻的输入信号 x t x_t xt​ 整合到一起作为当前时刻的输出信号。

          整合的过程如上图所示， x t x_t xt​ 和 h t − 1 h_{t-1} ht−1​ 经过一个sigmoid神经网络层（神经网络参数为 W o , b o W_o, b_o Wo​,bo​）输出一个 0 到 1 之间的数值 o t o_t ot​。 C t C_t Ct​ 经过一个tanh函数（注意：这里不是 tanh 神经网络层）得到一个在 -1 到 1 之间的数值，并与 o t o_t ot​ 相乘得到输出信号 h t h_t ht​，同时 h t h_t ht​ 也作为下一个时刻的输入信号传递到下一阶段。

          其中，tanh 函数是激活函数的一种，函数图像为：

          

          3.7 LSTM 的一些变体

          3.7.1 在门上增加窥视孔

          

          这是2000年Gers和Schemidhuber教授提出的一种LSTM变体。图中，在传统的LSTM结构基础上，每个门（遗忘门、记忆门和输出门）增加了一个“窥视孔”（Peephole），也可以在使用时选择只对部分门加入窥视孔。

          3.7.2 整合遗忘门和输入门

          

          与传统的LSTM不同的是，这个变体不需要分开来确定要被遗忘和记住的信息，采用一个结构搞定。在遗忘门的输出信号值（0 到 1之间）上，用 1 减去该数值来作为记忆门的状态选择，表示只更新需要被遗忘的那些信息的状态。

          3.7.3 GRU

          改进比较大的一个LSTM变体叫Gated Recurrent Unit (GRU)，目前应用较多。结构图如下

          

          GRU主要包含2个门：重置门和更新门。GRU混合了细胞状态 C t C_t Ct​ 和隐藏状态 h t − 1 h_{t-1} ht−1​ 为一个新的状态，使用 h t h_t ht​ 来表示。 该模型比传统的标准LSTM模型简单。

          4 基于Pytorch的LSTM代码实现

          使用正弦函数和余弦函数来构造时间序列，而正余弦函数之间是成导数关系，所以我们可以构造模型来学习正弦函数与余弦函数之间的映射关系，通过输入正弦函数的值来预测对应的余弦函数的值。

          每一个正弦函数的值都对应一个余弦函数值。如果只关心正弦函数的值本身而不考虑当前值所在的时间，那么正弦函数值和余弦函数值不是一一对应关系。例如，当 t=2.5 和 t=6.8 时，sin(t)=0.5 ，但在这两个不同的时刻，cos(t) 的值却不一样，也就是说如果不考虑时间，同一个正弦函数值可能对应了不同的几个余弦函数值。对于传统的神经网络来说，它仅仅基于当前的输入来预测输出，对于这种同一个输入可能对应多个输出的情况不再适用。

          取正弦函数的值作为LSTM的输入，来预测余弦函数的值。基于Pytorch来构建LSTM模型，采用1个输入神经元，1个输出神经元，16个隐藏神经元作为LSTM网络的构成参数，平均绝对误差（LMSE）作为损失误差，使用Adam优化算法来训练LSTM神经网络。

          # -*- coding:UTF-8 -*-
          import numpy as np
          import torch
          from torch import nn
          import matplotlib.pyplot as plt
          # Define LSTM Neural Networks
          class LstmRNN(nn.Module):
              """
                  Parameters：
                  - input_size: feature size
                  - hidden_size: number of hidden units
                  - output_size: number of output
                  - num_layers: layers of LSTM to stack
              """
              def __init__(self, input_size, hidden_size=1, output_size=1, num_layers=1):
                  super().__init__()
           
                  self.lstm = nn.LSTM(input_size, hidden_size, num_layers) # utilize the LSTM model in torch.nn 
                  self.forwardCalculation = nn.Linear(hidden_size, output_size)
           
              def forward(self, _x):
                  x, _ = self.lstm(_x)  # _x is input, size (seq_len, batch, input_size)
                  s, b, h = x.shape  # x is output, size (seq_len, batch, hidden_size)
                  x = x.view(s*b, h)
                  x = self.forwardCalculation(x)
                  x = x.view(s, b, -1)
                  return x
          if __name__ == '__main__':
              # create database
              data_len = 200
              t = np.linspace(0, 12*np.pi, data_len)
              sin_t = np.sin(t)
              cos_t = np.cos(t)
              dataset = np.zeros((data_len, 2))
              dataset[:,0] = sin_t
              dataset[:,1] = cos_t
              dataset = dataset.astype('float32')
              # plot part of the original dataset
              plt.figure()
              plt.plot(t[0:60], dataset[0:60,0], label='sin(t)')
              plt.plot(t[0:60], dataset[0:60,1], label = 'cos(t)')
              plt.plot([2.5, 2.5], [-1.3, 0.55], 'r--', label='t = 2.5') # t = 2.5
              plt.plot([6.8, 6.8], [-1.3, 0.85], 'm--', label='t = 6.8') # t = 6.8
              plt.xlabel('t')
              plt.ylim(-1.2, 1.2)
              plt.ylabel('sin(t) and cos(t)')
              plt.legend(loc='upper right')
              # choose dataset for training and testing
              train_data_ratio = 0.5 # Choose 80% of the data for testing
              train_data_len = int(data_len*train_data_ratio)
              train_x = dataset[:train_data_len, 0]
              train_y = dataset[:train_data_len, 1]
              INPUT_FEATURES_NUM = 1
              OUTPUT_FEATURES_NUM = 1
              t_for_training = t[:train_data_len]
              # test_x = train_x
              # test_y = train_y
              test_x = dataset[train_data_len:, 0]
              test_y = dataset[train_data_len:, 1]
              t_for_testing = t[train_data_len:]
              # ----------------- train -------------------
              train_x_tensor = train_x.reshape(-1, 5, INPUT_FEATURES_NUM) # set batch size to 5
              train_y_tensor = train_y.reshape(-1, 5, OUTPUT_FEATURES_NUM) # set batch size to 5
           
              # transfer data to pytorch tensor
              train_x_tensor = torch.from_numpy(train_x_tensor)
              train_y_tensor = torch.from_numpy(train_y_tensor)
              # test_x_tensor = torch.from_numpy(test_x)
           
              lstm_model = LstmRNN(INPUT_FEATURES_NUM, 16, output_size=OUTPUT_FEATURES_NUM, num_layers=1) # 16 hidden units
              print('LSTM model:', lstm_model)
              print('model.parameters:', lstm_model.parameters)
           
              loss_function = nn.MSELoss()
              optimizer = torch.optim.Adam(lstm_model.parameters(), lr=1e-2)
           
              max_epochs = 10000
              for epoch in range(max_epochs):
                  output = lstm_model(train_x_tensor)
                  loss = loss_function(output, train_y_tensor)
           
                  loss.backward()
                  optimizer.step()
                  optimizer.zero_grad()
           
                  if loss.item()  
          该模型在训练集和测试集上的结果如下：
           
           
          图中，红色虚线的左边表示该模型在训练数据集上的表现，右边表示该模型在测试数据集上的表现。可以看到，使用LSTM构建训练模型，可以仅仅使用正弦函数在 t 时刻的值作为输入来准确预测 t 时刻的余弦函数值，不用额外添加当前的时间信息、速度信息等。