NLP Course documentation

Transformers 是如何工作的?

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Transformers 是如何工作的?

Ask a Question

在本节中,我们简要了解 Transformer 模型的架构。

一点 Transformers 的发展历史

以下是 Transformer 模型(简要)历史中的一些关键节点:

A brief chronology of Transformers models.

Transformer 架构 于 2017 年 6 月提出。原本研究的重点是翻译任务。随后推出了几个有影响力的模型,包括

  • 2018 年 6 月GPT ,第一个预训练的 Transformer 模型,用于各种 NLP 任务并获得极好的结果

  • 2018 年 10 月BERT ,另一个大型预训练模型,该模型旨在生成更好的句子摘要(下一章将详细介绍!)

  • 2019 年 2 月GPT-2 ,GPT 的改进(并且更大)版本,由于道德问题没有立即公开发布

  • 2019 年 10 月DistilBERT ,BERT 的提炼版本,速度提高 60%,内存减轻 40%,但仍保留 BERT 97% 的性能

  • 2019 年 10 月BARTT5 ,两个使用与原始 Transformer 模型原始架构的大型预训练模型(第一个这样做)

  • 2020 年 5 月GPT-3 ,GPT-2 的更大版本,无需微调即可在各种任务上表现良好(称为零样本学习)

此列表远非详尽无遗,只是为了突出一些不同类型的 Transformer 模型。大体上,它们可以分为三类:

  • GPT-like (也被称作自回归 Transformer 模型)
  • BERT-like (也被称作自动编码 Transformer 模型)
  • BART/T5-like (也被称作序列到序列的 Transformer 模型)

稍后我们将更深入地探讨这些分类。

Transformers 是语言模型

上面提到的所有 Transformer 模型(GPT、BERT、BART、T5 等)都被训练为语言模型。这意味着他们已经以无监督学习的方式接受了大量原始文本的训练。无监督学习是一种训练类型,其中目标是根据模型的输入自动计算的。这意味着不需要人工来标记数据!

这种类型的模型可以对其训练过的语言有统计学的理解,但对于特定的实际任务的效果并不是很好。因此,一般的预训练模型会经历一个称为迁移学习(transfer learning)的过程。在此过程中,模型在给定任务上以监督方式(即使用人工注释标签)进行微调。

任务的一个例子是阅读 n 个单词的句子,预测下一个单词。这被称为因果语言建模(causal language modeling),因为输出取决于过去和现在的输入,而不依赖于未来的输入。

Example of causal language modeling in which the next word from a sentence is predicted.

另一个例子是掩码语言建模(masked language modeling),俗称完形填空,该模型预测句子中的遮住的词。

Example of masked language modeling in which a masked word from a sentence is predicted.

Transformer 是大模型

除了一些特例(如 DistilBERT)外,实现更好性能的一般策略是增加模型的大小以及预训练的数据量。

Number of parameters of recent Transformers models

不幸的是,训练模型,尤其是大型模型,需要大量的数据,时间和计算资源。它甚至会对环境产生影响,如下图所示。

The carbon footprint of a large language model.

这只是显示了一支团队领导的(非常大的)模型项目,该团队试图减少预训练对环境的影响。如果为了获得最佳超参数而进行大量试验,所造成的碳排放当量会更高。

想象一下,如果每次一个研究团队、一个学生组织或一家公司想要训练一个模型,都从头开始训练的。这将导致巨大的、不必要的浪费!

这就是为什么共享语言模型至关重要:共享经过训练的权重,当遇见新的需求时在预训练的权重之上进行微调,可以降低训练模型训练的算力和时间消耗,降低全球的总体计算成本和碳排放。

顺便说一句,你可以通过多种工具评估模型训练的碳排放。例如 ML CO2 Impact 或集成在 🤗 Transformers 中的 Code Carbon 。要了解更多相关信息,你可以阅读这篇 博客文章 ,其中将向你展示如何生成 emissions.csv 文件估计训练的碳排放,这里还有 🤗 Transformers 关于碳排放的 文档

迁移学习

预训练(Pretraining)是是指从头开始训练模型:随机初始化权重,在没有任何先验知识的情况下开始训练。

The pretraining of a language model is costly in both time and money.

这种预训练通常是在非常大量的数据上进行的。因此,它需要大量的数据,而且训练可能需要几周的时间。

另一方面,微调(Fine-tuning)是在模型经过预训练后完成的训练。要执行微调,首先需要获取一个经过预训练的语言模型,然后使用特定于任务的数据集进行额外的训练。等等,为什么不直接为实际的最终任务而训练模型呢?有几个原因:

  • 预训练模型已经在与微调数据集有一些相似之处的数据集上进行了训练。因此,微调过程能够利用模型在预训练期间获得的知识(例如,对于 NLP 问题,预训练模型将对你在任务中使用的语言有某种统计规律上的理解)。
  • 由于预训练模型已经在大量数据上进行了训练,因此微调需要更少的数据来获得不错的结果。
  • 出于同样的原因,获得好结果所需的时间和资源要少得多

例如,可以利用英语的预训练过的模型,然后在 arXiv 语料库上对其进行微调,从而形成一个基于科学研究的模型。微调只需要有限的数据量:预训练模型获得的知识可以“迁移”到目标任务上,因此被称为迁移学习(transfer learning)。

The fine-tuning of a language model is cheaper than pretraining in both time and money.

因此,微调模型具有较低的时间、数据、财务和环境成本。迭代不同的微调方案也更快、更容易,因为与完整的预训练相比,训练的约束更少。

这个过程也会比从头开始的训练(除非你有很多数据)取得更好的效果,这就是为什么你应该总是尝试利用一个预训练的模型——一个尽可能接近你手头的任务的模型——并对其进行微调。

Transformer 模型的通用架构

在本节中,我们将概述 Transformer 模型的通用架构。如果你不理解其中的一些概念,不用担心;后面的章节将详细介绍每个组件。

简介

该模型主要由两个块组成:

  • Encoder (左侧):编码器接收输入并构建其表示(特征)。这意味着模型的使命是从输入中获取理解。
  • Decoder (右侧):解码器使用编码器的表示(特征)以及其他输入来生成目标序列。这意味着模型的使命是生成输出。
Architecture of a Transformers models

这些部件中的每一个都可以独立使用,具体取决于任务:

  • Encoder-only 模型:适用于需要理解输入的任务,如句子分类和命名实体识别。
  • Decoder-only 模型:适用于生成任务,如文本生成。
  • Encoder-decoder 模型 或者 sequence-to-sequence 模型:适用于需要根据输入进行生成的任务,如翻译或摘要。

在后面的部分中,我们将单独地深入研究这些体系结构。

注意力层

Transformer 模型的一个关键特性是注意力层。事实上,提出 Transformer 架构的文章的标题是 “Attention Is All You Need” !我们将在课程的后面更加深入地探讨注意力层的细节;现在,你需要知道的是,这一层将告诉模型在处理每个单词的表示时,对不同单词的重视(忽略)程度。

可以考虑一个例子,假如我们要将英语翻译成法语。给定输入“You like this course”时,翻译模型需要注意相邻的词“You”,才能获得“like”的正确翻译,因为在法语中,动词“like”的变形依赖于主语。然而,对于那个词的翻译,句子的其余部分并没有用。同样,当翻译“this”时,模型也需要注意词“course”,因为“this”的翻译依赖于关联名词是单数还是复数。再次,句子中的其他词对于“course”的翻译并不重要。对于更复杂的句子(和更复杂的语法规则),模型可能需要特别注意可能在句子中更远处出现的词,以正确翻译每个词。

同样的概念也适用于与自然语言相关的任何任务:一个词本身有一个含义,但这个含义受语境的影响很大,语境可以是研究该词之前或之后的任何其他词(或多个词)。

现在你对注意力层是什么有了一些了解,让我们更详细地看一下 Transformer 架构。

Transformer 的原始结构

Transformer 架构最初是为翻译而设计的。在训练期间,编码器接收特定语言的输入(句子),而解码器需要输出对应语言的翻译。在编码器中,注意力层可以使用一个句子中的所有单词(正如我们刚才看到的,给定单词的翻译可以取决于它在句子中的其他单词)。然而,解码器是按顺序工作的,并且只能注意它已经翻译过的句子中的单词。例如,当我们预测了翻译目标的前三个单词时,我们将它们提供给解码器,然后解码器使用编码器的所有输出来尝试预测第四个单词。

为了在训练过程中加快速度(当模型可以访问目标句子时),会将整个目标输入解码器,但不允许获取到要翻译的单词(如果它在尝试预测位置 2 的单词时可以访问位置 2 的单词,解码器就会偷懒,直接输出那个单词,从而无法学习到正确的语言关系!)。例如,当试图预测第 4 个单词时,注意力层只能获取位置 1 到 3 的单词。

最初的 Transformer 架构如下所示,编码器位于左侧,解码器位于右侧:

Architecture of a Transformers models

注意,解码器块中的第一个注意力层关联到解码器的所有(过去的)输入,但是第二个注意力层只使用编码器的输出。因此,它在预测当前单词时,可以使用整个句子的信息。这是非常有用的,因因为不同的语言可以有把词放在不同顺序的语法规则,或者句子后面提供的一些上下文可能有助于确定给定单词的最佳翻译。

也可以在编码器/解码器中使用attention mask(注意力掩码层),以防止模型关注到某些特殊单词。例如,用于在批量处理句子时使所有输入长度一致的特殊填充词。

architecture(架构)与 checkpoints(权重参数又称检查点)

在本课程中,当我们深入探讨 Transformers 模型时,你将看到架构、参数和模型。这些术语的含义略有不同:

  • architecture(架构):这是模型的骨架 —— 每个层的定义以及模型中发生的每个操作。
  • Checkpoints(权重参数又称检查点):这些是将在给架构中结构中加载的权重参数,是一些具体的数值。
  • Model(模型):这是一个笼统的术语,没有“架构”或“参数”那么精确:它可以指两者。为了避免歧义,本课程使用将使用架构和参数。

例如,BERT 是一个架构,而 bert-base-cased ,这是谷歌团队为 BERT 的第一个版本训练的一组权重参数,是一个参数。我们可以说“BERT 模型”和“ bert-base-cased 模型。”

< > Update on GitHub