注意力机制----Multi-Head Attention 和 transformer

multi-head attention:

  • 单个attention输入后,对应有三个权重矩阵

注意力机制----Multi-Head Attention 和 transformer_第1张图片

  • multi-head输入后,如果有l个头,就有l×3个参数矩阵

  • multi-head输出的向量会串联起来,变长

注意力机制----Multi-Head Attention 和 transformer_第2张图片

注意力机制----Multi-Head Attention 和 transformer_第3张图片

  • 不只是attention,self—attention也有多头

注意力机制----Multi-Head Attention 和 transformer_第4张图片

注意力机制----Multi-Head Attention 和 transformer_第5张图片 







 

transformer的encoder: 

  • transformer的encoder的一层结构:multi—head attention+dense+全连接层

注意力机制----Multi-Head Attention 和 transformer_第6张图片

  • 可以多累几层

注意力机制----Multi-Head Attention 和 transformer_第7张图片

  • transformer的encoder对于上述结构,一共使用了6层

注意力机制----Multi-Head Attention 和 transformer_第8张图片







transformer的decoder:

  • 在decoder底层先是一个multi-head

注意力机制----Multi-Head Attention 和 transformer_第9张图片

  • 然后,encoder,decoder合起来multi-head

注意力机制----Multi-Head Attention 和 transformer_第10张图片

  • 最后:+dense+全连接层

注意力机制----Multi-Head Attention 和 transformer_第11张图片

  • 输入和输出的大小是对等的:

注意力机制----Multi-Head Attention 和 transformer_第12张图片

  • 当然,以上结构也是decoder的一个block

注意力机制----Multi-Head Attention 和 transformer_第13张图片

  • 相应地,一共有6个block

注意力机制----Multi-Head Attention 和 transformer_第14张图片

  • 其实,RNN和transformer的模型输入和输出都是一样的,可以拿过来直接用

注意力机制----Multi-Head Attention 和 transformer_第15张图片




 


 


 


 


 

  • encoder的大小

注意力机制----Multi-Head Attention 和 transformer_第16张图片

  • decoder的大小

注意力机制----Multi-Head Attention 和 transformer_第17张图片

  • transformer是一个seq2seq模型

  • 它不是RNN,没有循环结构

  • 它只是基于attention和self-attention

  • 它在机器翻译领域完暴RNN

注意力机制----Multi-Head Attention 和 transformer_第18张图片

 

 

 

参考视频:https://www.bilibili.com/video/BV1Ap4y1Q7nT

 

你可能感兴趣的:(注意力机制,深度学习基础)