试谈RNN中门的变迁

终于发完proposal 的邮件 深吸一口气

希望明天不要被怼的太惨

已经连续 高强度(hua shui) 看paper n天了

一切 索然无味

随着看到的paper层次越来越高

就越发羡慕搞NLP

昨天还在跟室友说 一开始觉得写SMNWuWei dalao指不定是个中年油腻大叔

结果人家研究生还没毕业

哇 满脸的羡慕

言归正传 打算用两三篇blog 讲一下最近学习的多轮检索式对话这个领域

第一篇就来谈一谈 在检索式对话中用到最多的 RNN 模型家族 (之所以 称之为 家族 因为变种太多了)

Naïve RNN

RNN = Recureent Neural Network

翻译成中文就是循环神经网络(注意不是递归,虽然它的过程很递归)

和传统的卷积神经网络CNN 全连接神经网络DNN不同的是其包含时序信息

顺带说一下另外两者的特点

DNN: n层与n-1层每个都有关, 参数数量级巨大;
CNN: 卷积+pool,至于什么是卷积?加权叠加

这一点十分有利于用于训练和时间相关的Dataset 尤其是NLP方面

有没有觉得很像马尔科夫链(en 不是过程 就是链)

事实上在有CNN之前 确实一般都做成隐马尔科夫链

NN起源于多层感知机MLP

感知机之所以能战胜同时期的元胞自动机异军突起 主要是其拥有反向传播算法

但NN随着训练层数的增大 会出现梯度消失现象 但层度深 确实效果好呀

于是就有一堆学者提出了 各种办法 使得 NN的层数能够扩展

比如说预处理 高速公路网络(highway network)和深度残差学习(deep residual learning)

时序 不仅仅是 训练结果和之前的转态有关 还是得输出结果长度可变的效果 这点和其他NN尤为不同

图片.png | center | 556x500

CNN 可以在下一个时间点 把输出作用在节点自身

如果按时间展开 就变成那张经典的图 作用在t+1时刻输出o(t+1)是该时刻输入和所有历史共同作用的结果

图片.png | center | 556x500

可以看出st+1,ot=f(st,xt,U,V,W)s_{t+1}, o_t = f(s_t, x_t, U, V, W)

和别的NN不同的是RNN所有步骤共享相同的参数U,V,WU, V, W

有正向的RNN 很容易想到是不是还有双向的(Bidirectional RNN) 深度(Deep Bidirectional RNN)

但对于上述RNN都不可避免的会出现前面说的梯度消失的现象

只不过在这里对的是时间维度上的消失(即 时序信息传播不过k间隔)

所以就有了一系列改进版RNN

LSTM

比如说最著名的LSTM[4]

LSTM = LONG SHORT-TERM MEMORY

其通过的设置来实现长时期的记忆能力

LSTM每个时刻的hidden state包含了多个memory blocks

每个block包含了多个memory cell

每个memory cell包含一个Cell和三个Gate: 输入门,输出门,遗忘门

图片.png | center | 556x500

Forward Pass

  • InputGateInput Gate
    alt=i=1Iwilxit+h=1Hwhlbht1+c=1Cwclsct1a_l^t=\sum\limits_{i=1}^Iw_{il}x_i^t+\sum\limits_{h=1}^Hw_{hl}b_h^{t-1}+\sum\limits_{c=1}^Cw_{cl}s_c^{t-1}

    blt=f(alt)b_l^t=f(a_l^t)

  • ForgetGateForget Gate
    aϕt=i=1Iwiϕxit+h=1Hwhϕbht1+c=1Cwcϕsct1a_\phi^t=\sum\limits_{i=1}^Iw_{i\phi}x_i^t+\sum\limits_{h=1}^Hw_{h\phi}b_h^{t-1}+\sum\limits_{c=1}^Cw_{c\phi}s_c^{t-1}

    bϕt=f(aϕt)b_\phi^t=f(a_\phi^t)

  • CellCell
    act=i=1Iwicxit+h=1Hwhcbht1a_c^t=\sum\limits_{i=1}^Iw_{ic}x_i^t+\sum\limits_{h=1}^Hw_{hc}b_h^{t-1}

    sct=bϕtsct1+bltg(act)s_c^t=b_\phi ^ts_c^{t-1}+b_l^tg(a_c^t)

  • OutputGateOutput Gate
    aωt=i=1Iwiωxit+h=1Hwhωbht1+c=1Cwcωscta_\omega^t=\sum\limits_{i=1}^Iw_{i\omega}x_i^t+\sum\limits_{h=1}^Hw_{h\omega}b_h^{t-1}+\sum\limits_{c=1}^Cw_{c\omega}s_c^t

    bωt=f(aωt)b_\omega^t=f(a_\omega^t)

  • CellOutputsCell Outputs

    bct=bωth(sct)b_c^t=b_\omega^th(s_c^t)

注意OutPut Gate中最后一项是scts_c^t, 而不是sct1s_c^{t-1} 因为此时Cell结果已经产生了

Backward Pass

定义 ϵct=Γbct\epsilon_c^t=\dfrac{\partial \Gamma}{\partial b_c^t}ϵst=Γsct\epsilon_s^t=\dfrac{\partial \Gamma}{\partial s_c^t}

  • CellOutputsCell Outputs

    ϵct=k=1Kwckδkt+g=1Gwcgδgt+1\epsilon_c^t=\sum\limits_{k=1}^Kw_{ck}\delta_k^t+\sum\limits_{g=1}^Gw_{cg}\delta_g^{t+1}

  • OutputGatesOutput Gates

    ϵωt=f(aωt)c=1Ch(sct)ϵct\epsilon_\omega^t=f'(a_\omega^t)\sum\limits_{c=1}^Ch(s_{c}^t)\epsilon_c^t

  • StateState

    ϵst=bwth(sct)+bϕt+1ϵst+1+wcϕδϕt+1+wcwδwt\epsilon_s^t=b_w^th'(s_c^t)+b_\phi^{t+1}\epsilon_s^{t+1}+w_{c\phi}\delta_\phi^{t+1}+w_{cw}\delta_w^t

  • CellCell

    δct=bltg(act)ϵst\delta_c^t=b_l^tg'(a_c^t)\epsilon_s^{t}

  • ForgetGatesForget Gates

    ϵϕt=f(aϕt)c=1Csct1ϵst\epsilon_\phi^t=f'(a_\phi^t)\sum\limits_{c=1}^Cs_{c}^{t-1}\epsilon_s^t

  • InputGatesInput Gates

    ϵlt=f(alt)c=1Cg(act)ϵst\epsilon_l^t=f'(a_l^t)\sum\limits_{c=1}^Cg(a_{c}^{t})\epsilon_s^t

可以看出Forget Gates和其他两个Gates在指数上略有差别

嗯 我放这么些公式就是想要恶心大家的

我已经料想到没什么人 可以看到这里了

实际上你可以把LSTM想象成一个传送带,从过去一直拉到未来

而门则是管控上下这根传送带尽职的保安大叔

Forget Gate规定着什么时候必须下车放心 这是去幼儿园的车

Input Gate负责到点把东西放入传送带

Output Gate负责到点把东西从传送带输出(Forget是无用的 Output是有用的)

结合下图再理解下

图片.png | center | 556x500

GRU

注意到LSTM有三个门

在计算时这三个门都需要进行迭代 在计算时耗时较大 并行操作空间较小

故提出了GRU模型[9]

其通过Update Gates 替代Output Gates + Forget Gates

Cell State 和 隐状态hih_i 合并

  • LSTM转态转移方程(这才是需要记得公式)

    it=σ(Wist1+Uixt+bi)i_t=\sigma(W_is_{t-1}+U_ix_t+b_i)

    ot=σ(Wost1+Uoxt+bo)o_t=\sigma(W_os_{t-1}+U_ox_t+b_o)

    ft=σ(Wfst1+Ufxt+bf)f_t=\sigma(W_fs_{t-1}+U_fx_t+b_f)

    st~=ϕ(W(otst1))+Uxt+b)\tilde{s_t}=\phi(W(o_t\bigodot s_{t-1}))+Ux_t+b)

    st=ftst1+itst~s_t=f_t\bigodot s_{t-1}+i_t\bigodot \tilde{s_t}

    其中i, o, f分别代表input, output, forget gates

  • GRU转态转移方程

    rt=σ(Wrst1+Urxt+br)r_t=\sigma(W_rs_{t-1}+U_rx_t+b_r)

    zt=σ(Wzst1+Uzxt+bz)z_t=\sigma(W_zs_{t-1}+U_zx_t+b_z)

    st~=ϕ(W(rtst1))+Uxt+b)\tilde{s_t}=\phi(W(r_t\bigodot s_{t-1}))+Ux_t+b)

    st=ztst1+(1zt)st~s_t=z_t\bigodot s_{t-1}+(1-z_t)\bigodot \tilde{s_t}

    其中r, z分别代表reset, update

可以看出转态转移方程少了一个 计算量 势必会下降

很显然 GRU LSTM也都有对应的双向版本

SRU 及 类似模型

GRU的结果实际上已经比较好了

但 计算代价还是太大

于是在16年末到17年 逐渐由学者提出进一步缩减门运算的模型结构

门减少 势必会减小运算量 但之所以引入门 是因为 我们需要更好的传递性

当然在顶会上 发Paper的这几个模型 在实际效果上 都不错

我这里写SRU不太恰当 Quasi-RNN, MRU都是类似的思想 这里以SRU为例来进行分析

直接来看

  • SRU的转态转移方程[8]

    st~=Wst\tilde{s_t}=Ws_t

    ft=σ(Wfst+bf)f_t=\sigma(W_fs_t+b_f)

    rt=σ(Wrst+br)r_t=\sigma(W_rs_t+b_r)

    st~=ϕ(W(rtst1))+Uxt+b)\tilde{s_t}=\phi(W(r_t\bigodot s_{t-1}))+Ux_t+b)

    ct=ftct1+(1ft)st~c_t=f_t\bigodot c_{t-1}+(1-f_t)\bigodot \tilde{s_t}

    ht=rtg(ct)+(1rt)sth_t=r_t\bigodot g(c_t)+(1-r_t)\bigodot s_t

可以看出其相较于之前的模型最大的差别在于转态不再和之前转态有关

这意味着什么?

意味着 我们不再需要等着迭代

在预处理的时候 就可以把所有门状态值计算出来

!!!而且这些门的计算都是复杂度十分高的矩阵乘法

注意这里的是矩阵乘法 而下面隐层hth_t中的运算都是矩阵的Hadamard乘--对应i,ji,j直接相乘

这两者的复杂度差别十分大了

所以 SRU这类模型 最大的贡献就是 提升 RNN运算速度

当然 SRU 并没有不依赖前者转态

self-attention

当然 dalao也不会闲着

就在大家已经觉得时序就=RNN的时候

Google Brain的dalao 发了一篇题目就很拉风的paper

Attention is all you need[10]

简单来说 其一次性计算出带较长语句的word encodeing

通过positional matrix 来获得时序信息

这样的好处就是可以并行计算 在计算性能上较RNN更优

self-attention的另外一个优点就是寻找时序关系更优

尤其是适合在跳跃topic的语料中

举个例子聊天聊到一半你说你去收个衣服,在这里topic就中断了,直到你再次回来

position的方式更容易计算之间的关系 而不用担心梯度消失

  • 具体公式
    Attention(Q,K,V)=softmax(QKTdk)VAttention(Q,K,V) =softmax(\dfrac{QK^T}{\sqrt{d_k}})V

    QRn×dk,KRdk×m,VRm×dvQ\in R^{n\times d_k},K\in R^{d_k\times m},V\in R^{m\times d_v}

    Attention(qt,K,V)=s=1m1zexp(<qt,ks>dk)vsAttention(q_t,K,V) =\sum\limits_{s=1}^m\dfrac{1}{z}exp(\dfrac{<q_t,k_s>}{\sqrt{d_k}})v_s

    headi=Attention(QWiQ,KWiK,VWiV)head_i = Attention(QW_i^Q,KW_i^K,VW_i^V)

    MultiHead(Q,K,V)=concat(head1,head2,...,headh)MultiHead(Q,K,V) = concat(head_1, head_2,...,head_h)

图片.png | center | 556x500

Sliced Recurrent Neural Networks

从上文我们可以知道 RNN的结构是链式的

必须在前者进行完之后才能进行下一步

有人就对这样的链式结构进行优化[7]

通过类似二分的思想对RNN运行顺序进行优化 也得到了较好的结果

图片.png | center | 556x500

然后[6]中周志华dalao 利用FSA 对RNN 过程进行捕捉 从而进行可解释分析

Reference

  1. Understanding LSTM Networks
  2. The Unreasonable Effectiveness of Recurrent Neural Networks
  3. 如何评价新提出的RNN变种SRU?
  4. LONG SHORT-TERM MEMORY
  5. Supervised Sequence Labelling with Recurrent Neural Networks
  6. Learning with Interpretable Structure from RNN
  7. Sliced Recurrent Neural Networks
  8. Simple Recurrent Units for Highly Parallelizable Recurrence
  9. Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling
  10. Attention Is All You Need
累计访问量: | 昨日访问量: | 昨日爬虫数:
Gunjianpan © 2017 - 2020 Power by VuePress & iofu728/blog