gpt4 book ai didi

GLM:通用语言模型

转载 作者:我是一只小鸟 更新时间:2023-03-30 14:31:34 25 4
gpt4 key购买 nike

ChatGPT已经火了一段时间了,国内也出现了一些平替,其中比较容易使用的是ChatGLM-6B: https://github.com/THUDM/ChatGLM-6B ,主要是能够让我们基于单卡自己部署。ChatGLM的基座是GLM: General Language Model Pretraining with Autoregressive Blank Infilling论文中提出的模型,接下来我们来看看.

论文名称:GLM: General Language Model Pretraining with Autoregressive Blank Infilling 。

论文地址: https://aclanthology.org/2022.acl-long.26.pdf 。

代码地址: https://github.com/THUDM/GLM 。

介绍

预训练语言吗模型大体可以分为三种:自回归(GPT系列)、自编码(BERT系列)、编码-解码(T5、BART),它们每一个都在各自的领域上表现不俗,但是,目前没有一个预训练模型能够很好地完成所有任务。GLM是一个通用的预训练语言模型,它在NLU(自然语言理解)、conditional(条件文本生成) and unconditional generation(非条件文本生成)上都有着不错的表现.

GLM的核心是:Autoregressive Blank Infilling,如下图1所示:

即,将文本中的一段或多段空白进行填充识别。具体细节如图2所示:

说明,对于一个文本: \(x_{1},x_{2},x_{3},x_{4},x_{5}\) ,空白长度会以 \(\lambda=3\) 的泊松分布进行采样。重复采样直到空白token的总数目占文本token数的15%。将文本分为两部分,A部分由原始token和[MASK]组成,B部分由空白token组成,最终将A部分和B部分进行拼接,同时B部分的每一个空白会被打乱,这样在自回归预测每个token的时候可以看到上下文的信息(具体通过注意力掩码来实现)。需要注意的是位置编码是2D的,位置编码1用于表示token在文本的位置,位置编码2用于表示原始文本和每一个空白中token的顺序.

多任务训练

为了能够兼顾NLU和文本生成,对于文档和句子采用不同的空白填充方式.

  • 文档:span的长度从原始长度的50%-100%的均匀分布中抽取。该目标旨在生成长文本。
  • 句子:限制masked span必须是完整的句子。多个span(句子)被取样,以覆盖15%的的原始标记。这个目标是针对seq2seq任务,其预测往往是完整的句子或段落。

模型架构

GLM使用单个Transformer,并对架构进行了修改

(1)调整layer normalization和residual connection的顺序.

(2)使用单一线性层进行输出token预测.

(3)将ReLU激活函数替换为GeLUs.

2D位置编码

两个位置id通过可学习嵌入表投影到两个向量,这两个向量都被添加到输入标记嵌入中。该编码方法确保模型在重建时不知道被屏蔽的跨度的长度。这种设计适合下游任务,因为通常生成的文本的长度是事先未知的.

微调

对于分类任务,在模板后面预测类别.

  • It’s a beautiful day, I’m in a great mood. it is [MASK]. [S] good
  • I failed in the exam today. I was very depressed. it is [MASK] [S] bad

对于文本生成任务,输入的文本视为A部分,在该部分后面添加[MASK],使用自回归来生成文本.

这样的好处是预训练和微调是保持一致的.

结果

最后此篇关于GLM:通用语言模型的文章就讲到这里了,如果你想了解更多关于GLM:通用语言模型的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com