码迷,mamicode.com
首页 > 其他好文 > 详细

真正的完全图解Seq2Seq Attention模型

时间:2020-11-26 14:57:13      阅读:5      评论:0      收藏:0      [点我收藏+]

标签:结构化   --   softmax   info   output   mic   权重   通过   表达   

五分钟看懂seq2seq attention模型。

本文通过图片,详细地画出了seq2seq+attention模型的全部流程,帮助小伙伴们无痛理解机器翻译等任务的重要模型。

seq2seq 是一个Encoder–Decoder 结构的网络,它的输入是一个序列,输出也是一个序列, Encoder 中将一个可变长度的信号序列变为固定长度的向量表达,Decoder 将这个固定长度的向量变成可变长度的目标的信号序列。--简书

好了别管了,接下来开始刷图吧。

大框架

技术图片
想象一下翻译任务,input是一段英文,output是一段中文。
公式(直接跳过看图最佳)
技术图片

详细图

技术图片
左侧为Encoder+输入,右侧为Decoder+输出。中间为Attention。
技术图片
技术图片
从左边Encoder开始,输入转换为word embedding, 进入LSTM。LSTM会在每一个时间点上输出hidden states。如图中的h1,h2,...,h8。
技术图片
技术图片

接下来进入右侧Decoder,输入为(1) 句首 <sos>符号,原始context vector(为0),以及从encoder最后一个hidden state: h8。LSTM的是输出是一个hidden state。(当然还有cell state,这里没用到,不提。)
技术图片

技术图片

Decoder的hidden state与Encoder所有的hidden states作为输入,放入Attention模块开始计算一个context vector。之后会介绍attention的计算方法。

下一个时间点
技术图片

来到时间点2,之前的context vector可以作为输入和目标的单词串起来作为lstm的输入。之后又回到一个hiddn state。以此循环。

技术图片
技术图片
另一方面,context vector和decoder的hidden state合起来通过一系列非线性转换以及softmax最后计算出概率。
在luong中提到了三种score的计算方法。这里图解前两种:
技术图片

Attention score function: dot
技术图片

输入是encoder的所有hidden states H: 大小为(hid dim, sequence length)。decoder在一个时间点上的hidden state, s: 大小为(hid dim, 1)。

第一步:旋转H为(sequence length, hid dim) 与s做点乘得到一个 大小为(sequence length, 1)的分数。

第二步:对分数做softmax得到一个合为1的权重。

第三步:将H与第二步得到的权重做点乘得到一个大小为(hid dim, 1)的context vector。

Attention score function: general
技术图片

输入是encoder的所有hidden states H: 大小为(hid dim1, sequence length)。decoder在一个时间点上的hidden state, s: 大小为(hid dim2, 1)。此处两个hidden state的纬度并不一样。

第一步:旋转H为(sequence length, hid dim1) 与 Wa [大小为 hid dim1, hid dim 2)] 做点乘, 再和s做点乘得到一个 大小为(sequence length, 1)的分数。

第二步:对分数做softmax得到一个合为1的权重。

第三步:将H与第二步得到的权重做点乘得到一个大小为(hid dim, 1)的context vector。

完结

看懂一个模型的最好办法就是在心里想一遍从输入到模型到输出每一个步骤里,tensor是如何流动的。
技术图片

推荐阅读:

闲聊结构化预测(structured learning)【这是一类问题】
我就不信看完这篇你还搞不懂信息熵
通俗易懂讲解感知机(三)--收敛性证明与对偶形式以及python代码讲解
技术图片

真正的完全图解Seq2Seq Attention模型

标签:结构化   --   softmax   info   output   mic   权重   通过   表达   

原文地址:https://blog.51cto.com/15009309/2553323

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!