LSTM模型与前向反向传播算法

简介:

1. 从RNN到LSTM

    在RNN模型里,我们讲到了RNN具有如下的结构,每个序列索引位置t都有一个隐藏状态h(t)h(t)

    如果我们略去每层都有的o(t),L(t),y(t)o(t),L(t),y(t),则RNN的模型可以简化成如下图的形式:

    图中可以很清晰看出在隐藏状态h(t)h(t)x(t)x(t)h(t1)h(t−1)得到。得到h(t)h(t)后一方面用于当前层的模型损失计算,另一方面用于计算下一层的h(t+1)h(t+1)

    由于RNN梯度消失的问题,大牛们对于序列索引位置t的隐藏结构做了改进,可以说通过一些技巧让隐藏结构复杂了起来,来避免梯度消失的问题,这样的特殊RNN就是我们的LSTM。由于LSTM有很多的变种,这里我们以最常见的LSTM为例讲述。LSTM的结构如下图:

    可以看到LSTM的结构要比RNN的复杂的多,真佩服牛人们怎么想出来这样的结构,然后这样居然就可以解决RNN梯度消失的问题?由于LSTM怎么可以解决梯度消失是一个比较难讲的问题,我也不是很熟悉,这里就不多说,重点回到LSTM的模型本身。

 2. LSTM模型结构剖析

    上面我们给出了LSTM的模型结构,下面我们就一点点的剖析LSTM模型在每个序列索引位置t时刻的内部结构。

    从上图中可以看出,在每个序列索引位置t时刻向前传播的除了和RNN一样的隐藏状态h(t)h(t),还多了另一个隐藏状态,如图中上面的长横线。这个隐藏状态我们一般称为细胞状态(Cell State),记为C(t)C(t)。如下图所示:

    除了细胞状态,LSTM图中还有了很多奇怪的结构,这些结构一般称之为门控结构(Gate)。LSTM在在每个序列索引位置t的门一般包括遗忘门,输入门和输出门三种。下面我们就来研究上图中LSTM的遗忘门,输入门和输出门以及细胞状态。

 2.1 LSTM之遗忘门

    遗忘门(forget gate)顾名思义,是控制是否遗忘的,在LSTM中即以一定的概率控制是否遗忘上一层的隐藏细胞状态。遗忘门子结构如下图所示:

    图中输入的有上一序列的隐藏状态h(t1)h(t−1)和本序列数据x(t)x(t),通过一个激活函数,一般是sigmoid,得到遗忘门的输出f(t)f(t)。由于sigmoid的输出f(t)f(t)在[0,1]之间,因此这里的输出f^{(t)}代表了遗忘上一层隐藏细胞状态的概率。用数学表达式即为:

f(t)=σ(Wfh(t1)+Ufx(t)+bf)f(t)=σ(Wfh(t−1)+Ufx(t)+bf)

    其中Wf,Uf,bfWf,Uf,bf为线性关系的系数和偏倚,和RNN中的类似。σσ为sigmoid激活函数。

2.2 LSTM之输入门

    输入门(input gate)负责处理当前序列位置的输入,它的子结构如下图:

    从图中可以看到输入门由两部分组成,第一部分使用了sigmoid激活函数,输出为i(t)i(t),第二部分使用了tanh激活函数,输出为a(t)a(t), 两者的结果后面会相乘再去更新细胞状态。用数学表达式即为:

i(t)=σ(Wih(t1)+Uix(t)+bi)i(t)=σ(Wih(t−1)+Uix(t)+bi)
a(t)=tanh(Wah(t1)+Uax(t)+ba)a(t)=tanh(Wah(t−1)+Uax(t)+ba)

    其中Wi,Ui,bi,Wa,Ua,ba,Wi,Ui,bi,Wa,Ua,ba,为线性关系的系数和偏倚,和RNN中的类似。σσ为sigmoid激活函数。

2.3 LSTM之细胞状态更新

    在研究LSTM输出门之前,我们要先看看LSTM之细胞状态。前面的遗忘门和输入门的结果都会作用于细胞状态C(t)C(t)。我们来看看从细胞状态C(t1)C(t−1)如何得到C(t)C(t)。如下图所示:

    细胞状态C(t)C(t)由两部分组成,第一部分是C(t1)C(t−1)和遗忘门输出f(t)f(t)的乘积,第二部分是输入门的i(t)i(t)a(t)a(t)的乘积,即:

C(t)=C(t1)f(t)+i(t)a(t)C(t)=C(t−1)⊙f(t)+i(t)⊙a(t)

    其中,为Hadamard积,在DNN中也用到过。

2.4 LSTM之输出门

    有了新的隐藏细胞状态C(t)C(t),我们就可以来看输出门了,子结构如下:

    从图中可以看出,隐藏状态h(t)h(t)的更新由两部分组成,第一部分是o(t)o(t), 它由上一序列的隐藏状态h(t1)h(t−1)和本序列数据x(t)x(t),以及激活函数sigmoid得到,第二部分由隐藏状态C(t)C(t)和tanh激活函数组成, 即:

o(t)=σ(Woh(t1)+Uox(t)+bo)o(t)=σ(Woh(t−1)+Uox(t)+bo)
h(t)=o(t)tanh(C(t))h(t)=o(t)⊙tanh(C(t))

    通过本节的剖析,相信大家对于LSTM的模型结构已经有了解了。当然,有些LSTM的结构和上面的LSTM图稍有不同,但是原理是完全一样的。

3. LSTM前向传播算法

    现在我们来总结下LSTM前向传播算法。LSTM模型有两个隐藏状态h(t),C(t)h(t),C(t),模型参数几乎是RNN的4倍,因为现在多了Wf,Uf,bf,Wa,Ua,ba,Wi,Ui,bi,Wo,Uo,boWf,Uf,bf,Wa,Ua,ba,Wi,Ui,bi,Wo,Uo,bo这些参数。

    前向传播过程在每个序列索引位置的过程为:

    1)更新遗忘门输出:

f(t)=σ(Wfh(t1)+Ufx(t)+bf)f(t)=σ(Wfh(t−1)+Ufx(t)+bf)

    2)更新输入门两部分输出:

i(t)=σ(Wih(t1)+Uix(t)+bi)i(t)=σ(Wih(t−1)+Uix(t)+bi)
a(t)=tanh(Wah(t1)+Uax(t)+ba)a(t)=tanh(Wah(t−1)+Uax(t)+ba)

    3)更新细胞状态:

C(t)=C(t1)f(t)+i(t)a(t)C(t)=C(t−1)⊙f(t)+i(t)⊙a(t)

    4)更新输出门输出:

o(t)=σ(Woh(t1)+Uox(t)+bo)o(t)=σ(Woh(t−1)+Uox(t)+bo)
h(t)=o(t)tanh(C(t))h(t)=o(t)⊙tanh(C(t))

    5)更新当前序列索引预测输出:

y^(t)=σ(Vh(t)+c)y^(t)=σ(Vh(t)+c)

4.  LSTM反向传播算法推导关键点

    有了LSTM前向传播算法,推导反向传播算法就很容易了, 思路和RNN的反向传播算法思路一致,也是通过梯度下降法迭代更新我们所有的参数,关键点在于计算所有参数基于损失函数的偏导数。

    在RNN中,为了反向传播误差,我们通过隐藏状态h(t)h(t)的梯度δ(t)δ(t)一步步向前传播。在LSTM这里也类似。只不过我们这里有两个隐藏状态h(t)h(t)C(t)C(t)。这里我们定义两个δδ,即:

δ(t)h=Lh(t)δh(t)=∂L∂h(t)
δ(t)C=LC(t)δC(t)=∂L∂C(t)

    反向传播时只使用了δ(t)CδC(t),变量δ(t)hδh(t)仅为帮助我们在某一层计算用,并没有参与反向传播,这里要注意。如下图所示:

 

    而在最后的序列索引位置ττδ(τ)hδh(τ)和 δ(τ)CδC(τ)为:

δ(τ)h=LO(τ)O(τ)h(τ)=VT(y^(τ)y(τ))δh(τ)=∂L∂O(τ)∂O(τ)∂h(τ)=VT(y^(τ)−y(τ))
δ(τ)C=Lh(τ)h(τ)C(τ)=δ(τ)ho(τ)(1tanh2(C(τ)))δC(τ)=∂L∂h(τ)∂h(τ)∂C(τ)=δh(τ)⊙o(τ)⊙(1−tanh2(C(τ)))

    接着我们由δ(t+1)CδC(t+1)反向推导δ(t)CδC(t)

    δ(t)hδh(t)的梯度由本层的输出梯度误差决定,即:

δ(t)h=Lh(t)=VT(y^(t)y(t))δh(t)=∂L∂h(t)=VT(y^(t)−y(t))

    而δ(t)CδC(t)的反向梯度误差由前一层δ(t+1)CδC(t+1)的梯度误差和本层的从h(t)h(t)传回来的梯度误差两部分组成,即:

δ(t)C=LC(t+1)C(t+1)C(t)+Lh(t)h(t)C(t)=δ(t+1)Cf(t+1)+δ(t)ho(t)(1tanh2(C(t)))δC(t)=∂L∂C(t+1)∂C(t+1)∂C(t)+∂L∂h(t)∂h(t)∂C(t)=δC(t+1)⊙f(t+1)+δh(t)⊙o(t)⊙(1−tanh2(C(t)))

    有了δ(t)hδh(t)δ(t)CδC(t), 计算这一大堆参数的梯度就很容易了,这里只给出WfWf的梯度计算过程,其他的Uf,bf,Wa,Ua,ba,Wi,Ui,bi,Wo,Uo,boV,cUf,bf,Wa,Ua,ba,Wi,Ui,bi,Wo,Uo,bo,V,c的梯度大家只要照搬就可以了。

LWf=t=1τLC(t)C(t)f(t)f(t)Wf=t=1τδ(t)CC(t1)f(t)(1f(t))(h(t1))T∂L∂Wf=∑t=1τ∂L∂C(t)∂C(t)∂f(t)∂f(t)∂Wf=∑t=1τδC(t)⊙C(t−1)⊙f(t)(1−f(t))(h(t−1))T

5. LSTM小结

    LSTM虽然结构复杂,但是只要理顺了里面的各个部分和之间的关系,进而理解前向反向传播算法是不难的。当然实际应用中LSTM的难点不在前向反向传播算法,这些有算法库帮你搞定,模型结构和一大堆参数的调参才是让人头痛的问题。不过,理解LSTM模型结构仍然是高效使用的前提。


本文转自刘建平Pinard博客园博客,原文链接:http://www.cnblogs.com/pinard/p/6519110.html,如需转载请自行联系原作者


相关文章
|
13天前
|
机器学习/深度学习 人工智能 算法
深度学习入门:理解神经网络与反向传播算法
【9月更文挑战第20天】本文将深入浅出地介绍深度学习中的基石—神经网络,以及背后的魔法—反向传播算法。我们将通过直观的例子和简单的数学公式,带你领略这一技术的魅力。无论你是编程新手,还是有一定基础的开发者,这篇文章都将为你打开深度学习的大门,让你对神经网络的工作原理有一个清晰的认识。
|
13天前
|
机器学习/深度学习 人工智能 算法
鸟类识别系统Python+卷积神经网络算法+深度学习+人工智能+TensorFlow+ResNet50算法模型+图像识别
鸟类识别系统。本系统采用Python作为主要开发语言,通过使用加利福利亚大学开源的200种鸟类图像作为数据集。使用TensorFlow搭建ResNet50卷积神经网络算法模型,然后进行模型的迭代训练,得到一个识别精度较高的模型,然后在保存为本地的H5格式文件。在使用Django开发Web网页端操作界面,实现用户上传一张鸟类图像,识别其名称。
60 12
鸟类识别系统Python+卷积神经网络算法+深度学习+人工智能+TensorFlow+ResNet50算法模型+图像识别
|
10天前
|
机器学习/深度学习 人工智能 自然语言处理
深度学习还不如浅层网络?RL教父Sutton持续反向传播算法登Nature
【9月更文挑战第24天】近年来,深度学习在人工智能领域取得巨大成功,但在连续学习任务中面临“损失可塑性”问题,尤其在深度强化学习中更为突出。加拿大阿尔伯塔大学的研究人员提出了一种名为“持续反向传播”的算法,通过选择性地重新初始化网络中的低效用单元,保持模型的可塑性。该算法通过评估每个连接和权重的贡献效用来决定是否重新初始化隐藏单元,并引入成熟度阈值保护新单元。实验表明,该算法能显著提升连续学习任务的表现,尤其在深度强化学习领域效果明显。然而,算法也存在计算复杂性和成熟度阈值设置等问题。
22 2
WK
|
20天前
|
机器学习/深度学习 监控 算法
反向传播算法是如何工作的
反向传播算法通过最小化损失函数优化神经网络。首先,输入数据经由前向传播得到预测结果,并计算损失;接着,反向传播计算各参数的梯度,并利用梯度下降法更新权重和偏置。这一过程反复进行,直至满足停止条件。算法具备高效性、灵活性及可扩展性,能处理复杂模式识别与预测任务,适用于不同类型与规模的神经网络,显著提升了模型的预测准确性和泛化能力。
WK
16 3
|
29天前
|
存储 自然语言处理 算法
【算法精讲系列】MGTE系列模型,RAG实施中的重要模型
检索增强生成(RAG)结合检索与生成技术,利用外部知识库提升大模型的回答准确性与丰富性。RAG的关键组件包括文本表示模型和排序模型,前者计算文本向量表示,后者进行精细排序。阿里巴巴通义实验室推出的GTE-Multilingual系列模型,具备高性能、长文档支持、多语言处理及弹性向量表示等特性,显著提升了RAG系统的检索与排序效果。该系列模型已在多个数据集上展示出优越性能,并支持多语言和长文本处理,适用于各种复杂应用场景。
|
1月前
|
自然语言处理 监控 算法
【算法精讲系列】通义模型Prompt调优的实用技巧与经验分享
本文详细阐述了Prompt的设计要素,包括引导语、上下文信息等,还介绍了多种Prompt编写策略,如复杂规则拆分、关键信息冗余、使用分隔符等,旨在提高模型输出的质量和准确性。通过不断尝试、调整和优化,可逐步实现更优的Prompt设计。
|
1月前
|
算法
基于SIR模型的疫情发展趋势预测算法matlab仿真
该程序基于SIR模型预测疫情发展趋势,通过MATLAB 2022a版实现病例增长拟合分析,比较疫情防控力度。使用SIR微分方程模型拟合疫情发展过程,优化参数并求解微分方程组以预测易感者(S)、感染者(I)和移除者(R)的数量变化。![]该模型将总人群分为S、I、R三部分,通过解析或数值求解微分方程组预测疫情趋势。
|
1月前
|
机器学习/深度学习 数据采集 存储
一文读懂蒙特卡洛算法:从概率模拟到机器学习模型优化的全方位解析
蒙特卡洛方法起源于1945年科学家斯坦尼斯劳·乌拉姆对纸牌游戏中概率问题的思考,与约翰·冯·诺依曼共同奠定了该方法的理论基础。该方法通过模拟大量随机场景来近似复杂问题的解,因命名灵感源自蒙特卡洛赌场。如今,蒙特卡洛方法广泛应用于机器学习领域,尤其在超参数调优、贝叶斯滤波等方面表现出色。通过随机采样超参数空间,蒙特卡洛方法能够高效地找到优质组合,适用于处理高维度、非线性问题。本文通过实例展示了蒙特卡洛方法在估算圆周率π和优化机器学习模型中的应用,并对比了其与网格搜索方法的性能。
163 1
|
2月前
|
机器学习/深度学习 API 异构计算
7.1.3.2、使用飞桨实现基于LSTM的情感分析模型的网络定义
该文章详细介绍了如何使用飞桨框架实现基于LSTM的情感分析模型,包括网络定义、模型训练、评估和预测的完整流程,并提供了相应的代码实现。
|
2月前
|
机器学习/深度学习
【机器学习】面试题:LSTM长短期记忆网络的理解?LSTM是怎么解决梯度消失的问题的?还有哪些其它的解决梯度消失或梯度爆炸的方法?
长短时记忆网络(LSTM)的基本概念、解决梯度消失问题的机制,以及介绍了包括梯度裁剪、改变激活函数、残差结构和Batch Normalization在内的其他方法来解决梯度消失或梯度爆炸问题。
53 2
下一篇
无影云桌面