Gated Recurrent Unit(GRU)

整理并翻译自吴恩达深度学习系列视频:序列模型第一周,有所详略。

Gated Recurrent Unit(GRU)

Gated Recurrent Unit(GRU), which is a modification to the RNN hidden layer that makes it much better at capturing long range connections and helps a lot with the vanishing gradient problem.

Vanishing gradients AND Exploding gradients with RNNs

在这里插入图片描述
标准的RNN结构面临着2个很严重的问题。一是梯度消失和梯度爆炸;二是计算某个 y ^ < t > \hat{y}^{<t>} y^<t>时,只用到了前面的输入,尤其是是靠近 y ^ < t > \hat{y}^{<t>} y^<t>的输入,非常靠前的输入和后面的输入对结果产生的影响微乎其微。

第一个问题非常好理解,与CNN相同,当网络很深时,反向传播就很难从后向前对网络的前几层在计算序列上产生影响,越往前计算出的导数越小直至0,这是很直观的梯度消失问题。偶尔也会遇到梯度爆炸的问题,即计算出的导数越来越大。这是基本RNN结构的缺点。

第二个问题,在判断"Cat eat a lot of things,…, which was full."和"Cats eat a lot of things,…,which were full."两个句子的which后面到底应该跟单数还是复数形式时,显然需要考虑到第一个单词是cat还是cats,但是由于输出which后的单词时,普通的RNN结构很难保留第一个单词产生的影响在此时发挥作用,因此引入了GRU单元LTSM单元来解决这个问题。

在对"He said, “Teddy bears are on sale!” "和"He said, “Teddy Roosevelt was a great President!”"做命名实体判别时,需要知道Teddy后一个单词是什么,普通的RNN结构显然没有这种能力,因此引入了双向循环网络解决这一问题。

这篇文章仅介绍GRU,LTSM和双向神经网络参见:
Long Short term memory unit(LSTM)

GRU unit

在这里插入图片描述
普通的RNN unit如上图所示,一个简化了的GRU unit如下图所示:
在这里插入图片描述
GRU里首先引入了一个新的变量C作为memory cell使用,即保留一些前面的层中的某些有价值的信息。

GRU里的Gated是指我们引入了门控,即使用 Γ u \Gamma_u Γu来决定我们是否使用当前层计算出的 C ~ < t > \tilde{C}^{<t>} C~<t>来更新C。

C < t > = a < t > C^{<t>}=a^{<t>} C<t>=a<t>

C ~ < t > = t a n h ( W c [ C < t − 1 > , x < t > ] + b c ) \tilde{C}^{<t>}=tanh(W_c[C^{<t-1>}, x^{<t>}]+b_c) C~<t>=tanh(Wc[C<t1>,x<t>]+bc)
Γ u = σ ( W u [ C < t − 1 > , x < t > ] + b u ) \Gamma_u=\sigma(W_u[C^{<t-1>}, x^{<t>}]+b_u) Γu=σ(Wu[C<t1>,x<t>]+bu)
C < t > = Γ u ∗ C ~ < t > + ( 1 − Γ u ) ∗ C < t − 1 > C{<t>}=\Gamma_u*\tilde{C}^{<t>}+(1-\Gamma_u)*C^{<t-1>} C<t>=ΓuC~<t>+(1Γu)C<t1>
门控值 Γ u = 1 \Gamma_u=1 Γu=1时,用的是新的值 C ~ \tilde{C} C~;门控值 Γ u = 0 \Gamma_u=0 Γu=0时,用的是前一层的值 C < t − 1 > C^{<t-1>} C<t1>

完整的GRU计算公式如下:
在这里插入图片描述

C ~ < t > = t a n h ( Γ r ∗ W c [ C < t − 1 > , x < t > ] + b c ) \tilde{C}^{<t>}=tanh(\Gamma_r*W_c[C^{<t-1>}, x^{<t>}]+b_c) C~<t>=tanh(ΓrWc[C<t1>,x<t>]+bc)
Γ u = σ ( W u [ C < t − 1 > , x < t > ] + b u ) \Gamma_u=\sigma(W_u[C^{<t-1>}, x^{<t>}]+b_u) Γu=σ(Wu[C<t1>,x<t>]+bu)
Γ r = σ ( W r [ C < t − 1 > , x < t > ] + b r ) \Gamma_r=\sigma(W_r[C^{<t-1>}, x^{<t>}]+b_r) Γr=σ(Wr[C<t1>,x<t>]+br)
C < t > = Γ u C ~ < t > + ( 1 − Γ u ) C < t − 1 > C{<t>}=\Gamma_u\tilde{C}^{<t>}+(1-\Gamma_u)C^{<t-1>} C<t>=ΓuC~<t>+(1Γu)C<t1>

注意,这里引入了第二个门控值 Γ r \Gamma_r Γr来衡量 C ~ < t > \tilde{C}^{<t>} C~<t> C < t − 1 > {C}^{<t-1>} C<t1>之间的相关程度,计算该值使用了与 Γ u \Gamma_u Γu不同的参数矩阵。除此之外,在计算 C ~ < t > \tilde{C}^{<t>} C~<t>也使用了 Γ r \Gamma_r Γr

posted @ 2019-01-16 21:53  从流域到海域  阅读(188)  评论(0编辑  收藏  举报