Causal decoder、Prefix decoder和encoder-decoder

如下图:
蓝色:the attention between prefix tokens
绿色:the attention between prefix and target tokens
黄色:the attention betweetn target tokens and masked attention
请添加图片描述

  • 编码器-解码器(encoder-decoder):传统 Transformer 模型是建立在编码器-解码器架构上的 ,由两个 Transformer 块分别作为编码器和解码器。编码器采用堆叠的多头自注意层对输入序列进行编码以生成其潜在表示,而解码器对这些表示进行交叉注意并自回归地生成目标序列。目前,只有少数大语言模型是基于编码器-解码器架构构建的例如 Flan-T5。
  • 因果解码器(causal decoder):因果解码器架构采用单向注意力掩码,以确保每个输入标记只能关注过去的标记和它本身。输入和输出标记通过解码器以相同的方式进行处理。
    • chatGPT等
    • 因果编码器:OPT,BLOOM,Gopher等。
  • 前缀解码器(prefix decoder):前缀解码器结构修正了因果编码器的掩码机制,以使其能可对前缀标记执行双向注意力,并仅对生成的标记执行单向注意力。这样,与encoder-decoder类似,可以双向编码前缀序列并自回归低逐个预测输出标记,其中在编码和解码阶段共享相同的参数。现在前缀编码器的大模型包括U-PaLM、GLM-130B等。

猜你喜欢

转载自blog.csdn.net/qq_35812205/article/details/130734021