NLP Course documentation

模块化构建 tokenizer

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

模块化构建 tokenizer

Ask a Question Open In Colab Open In Studio Lab

正如我们在前几节中看到的,tokenization 包括几个步骤:

  • 标准化(任何认为必要的文本清理,例如删除空格或重音符号、Unicode 规范化等)
  • 预分词(将输入拆分为单词)
  • 通过模型处理输入(使用预先拆分的词来生成一系列 tokens )
  • 后处理(添加 tokenizer 的特殊 tokens 生成注意力掩码和 token 类型 ID)

作为复习,这里再看一遍整个过程:

The tokenization pipeline.

🤗 Tokenizers 库旨在为每个步骤提供多个选项,你可以任意搭配这些选项。在这一节中,我们将看到如何从零开始构建 tokenizer,而不是像我们在 第二节 中那样从旧的 tokenizer 训练新的 tokenizer 然后,你将能够构建任何你能想到的类型的 tokenizer

更精确地说,这个库围绕一个中心的 Tokenizer 类,实现了组成 Tokenizer 的各种子模块:

  • normalizers 包含所有可能使用的 Normalizer(标准化) 模块(完整列表 在这里 )。
  • pre_tokenizesr 包含所有可能使用的 PreTokenizer(预处理) 模块(完整列表 [在这里](https://huggi ngface.co/docs/tokenizers/api/pre-tokenizers) )。
  • models 包含了你可以使用的各种 Model(子词分词算法模型) 模块,如 BPEWordPieceUnigram (完整列表 在这里 )。
  • trainers 包含所有不同类型的 trainer ,你可以使用它们在语料库上训练你的模型(每种模型一个;完整列表 在这里 )。
  • post_processors 包含你可以使用的各种类型的 PostProcessor(后处理) 模块,(完整列表 在这里 )。
  • decoders 包含各种类型的 Decoder ,可以用来解码 tokenization 后的输出(完整列表 在这里 )。

你可以 在这里 找到完整的模块列表。

获取语​​料库

为了训练新的 tokenizer 我们将使用一小部分文本作为语料库(这样运行得更快)。获取语​​料库的步骤与我们在 在这章的开头 采取的步骤类似,但这次我们将使用 WikiText-2 数据集:

from datasets import load_dataset

dataset = load_dataset("wikitext", name="wikitext-2-raw-v1", split="train")


def get_training_corpus():
    for i in range(0, len(dataset), 1000):
        yield dataset[i : i + 1000]["text"]

get_training_corpus() 函数是一个生成器,每次调用的时候将产生 1,000 个文本,我们将用它来训练 tokenizer 。

🤗 Tokenizers 也可以直接在文本文件上进行训练。以下是我们生成一个包含 WikiText-2 所有文本的代码,这样我们就可以在本地离线使用:

with open("wikitext-2.txt", "w", encoding="utf-8") as f:
    for i in range(len(dataset)):
        f.write(dataset[i]["text"] + "\n")

接下来,我们将展示如何模块化地构建你自己的 BERT、GPT-2 和 XLNet tokenizer 这将包含主要的分词算法:WordPiece、BPE 和 Unigram 的例子。让我们从 BERT 开始吧!

从头开始构建 WordPiece tokenizer

要用🤗 Tokenizers 库构建一个 tokenizer 我们首先实例化一个带有 modelTokenizer 对象,然后将其 normalizerpre_tokenizerpost_processordecoder 属性设置为我们想要的值。

以这个例子来说,我们将创建一个使用 WordPiece 模型的 Tokenizer

from tokenizers import (
    decoders,
    models,
    normalizers,
    pre_tokenizers,
    processors,
    trainers,
    Tokenizer,
)

tokenizer = Tokenizer(models.WordPiece(unk_token="[UNK]"))

我们必须指定 unk_token ,这样当模型遇到它从未见过的字符时,它就会返回 unk_token。我们在这里可以设置的其他参数包括已有的 vocab(词汇表) (我们要重新训练模型,所以我们不需要设置这个)和 max_input_chars_per_word ,它指定了每个词的最大长度(比 max_input_chars_per_word 长的词将被拆分)。

tokenization 的第一步是标准化,所以我们从这里开始。由于 BERT 被广泛使用,所以我们可以使用 BertNormalizer ,我们可以为 BERT 设置经典参数: lowercase(小写)strip_accents(去除重音的字符)clean_text 用于删除所有控制字符并将重复的空格替换为一个空格;以及 handle_chinese_chars ,它将在中文字符周围添加空格。要复现 bert-base-uncased tokenizer 我们可以这样设置 normalizer

tokenizer.normalizer = normalizers.BertNormalizer(lowercase=True)

然而,通常来说,当你构建一个新的 tokenizer 时,也需要同步构建一个新的 normalizer —— 所以我们来看看如何手动创建 BERT normalizer 。🤗 Tokenizers 库提供了一个 Lowercase normalizer 和一个 StripAccents normalizer ,并且你可以使用 Sequence 来组合多个 normalizer。

tokenizer.normalizer = normalizers.Sequence(
    [normalizers.NFD(), normalizers.Lowercase(), normalizers.StripAccents()]
)

我们还使用了一个 NFD Unicode normalizer ,否则,否则 StripAccents normalizer 将因为无法正确识别带有重音的字符,从而没办法去除重音。

正如我们之前看到的,我们可以使用 normalizernormalize_str() 方法来对它进行测试:

print(tokenizer.normalizer.normalize_str("Héllò hôw are ü?"))
hello how are u?

更进一步如果你在包含 unicode 字符的字符串上测试先前 normalizers 的两个版本,你肯定会注意到这两个 normalizers 并不完全等效。

为了避免 normalizers.Sequence 过于复杂,我们的实现没有包含当 clean_text 参数设置为 TrueBertNormalizer 需要的正则表达式替换 —— 而这是 BertNormalizer 默认会实现的。但不要担心:通过在 normalizer 序列中添加两个 normalizers.Replace 可以在不使用方便的 BertNormalizer 的情况下获得完全相同的标准化。

下一步是预分词。同样,我们可以使用预构建的 BertPreTokenizer

tokenizer.pre_tokenizer = pre_tokenizers.BertPreTokenizer()

或者我们可以从头开始构建它:

tokenizer.pre_tokenizer = pre_tokenizers.Whitespace()

注意, Whitespace 会使用空格和所有不是字母、数字或下划线的字符进行分割,因此在本次的例子中上会根据空格和标点符号进行分割:

tokenizer.pre_tokenizer.pre_tokenize_str("Let's test my pre-tokenizer.")
[('Let', (0, 3)), ("'", (3, 4)), ('s', (4, 5)), ('test', (6, 10)), ('my', (11, 13)), ('pre', (14, 17)),
 ('-', (17, 18)), ('tokenizer', (18, 27)), ('.', (27, 28))]

如果你只想使用空格进行分割,则应该使用 WhitespaceSplit

pre_tokenizer = pre_tokenizers.WhitespaceSplit()
pre_tokenizer.pre_tokenize_str("Let's test my pre-tokenizer.")
[("Let's", (0, 5)), ('test', (6, 10)), ('my', (11, 13)), ('pre-tokenizer.', (14, 28))]

就像 normalizer 一样,你可以使用 Sequence 来组合几个预分词的步骤:

pre_tokenizer = pre_tokenizers.Sequence(
    [pre_tokenizers.WhitespaceSplit(), pre_tokenizers.Punctuation()]
)
pre_tokenizer.pre_tokenize_str("Let's test my pre-tokenizer.")
[('Let', (0, 3)), ("'", (3, 4)), ('s', (4, 5)), ('test', (6, 10)), ('my', (11, 13)), ('pre', (14, 17)),
 ('-', (17, 18)), ('tokenizer', (18, 27)), ('.', (27, 28))]

tokenization 流程的下一步是将输入数据传递给模型。我们已经在初始化时指定了我们的模型,但是我们还需要对其进行训练,这就需要一个 WordPieceTrainer 。在实例化一个🤗 Tokenizers 中的 Trainer 时,一件很重要的事情是,你需要将你打算使用的所有特殊 tokens 都传递给它——否则,由于它们不在训练语料库中,Trainer 就不会将它们添加到词汇表中:

special_tokens = ["[UNK]", "[PAD]", "[CLS]", "[SEP]", "[MASK]"]
trainer = trainers.WordPieceTrainer(vocab_size=25000, special_tokens=special_tokens)

除了指定 vocab_sizespecial_tokens ,我们还可以设置 min_frequency (一个 tokens 必须达到的最小的出现的次数才能被包含在词汇表中)或更改 continuing_subword_prefix (如果我们想使用其他的字符来替代 ## )。

我们只需要执行以下命令就可以使用我们之前定义的迭代器训练我们的模型:

tokenizer.train_from_iterator(get_training_corpus(), trainer=trainer)

我们还可以使用本地的文本文件来训练我们的 tokenizer 它看起来像这样(我们需要先使用 WordPiece 初始化一个空的模型):

tokenizer.model = models.WordPiece(unk_token="[UNK]")
tokenizer.train(["wikitext-2.txt"], trainer=trainer)

在这两种情况下,我们都可以通过调用 encode() 方法来测试 tokenizer

encoding = tokenizer.encode("Let's test this tokenizer.")
print(encoding.tokens)
['let', "'", 's', 'test', 'this', 'tok', '##eni', '##zer', '.']

所得到的 encoding 是一个 Encoding 对象,它包含 tokenizer 的所有必要属性: idstype_idstokensoffsetsattention_maskspecial_tokens_maskoverflowing

tokenizer 管道的最后一步是后处理。我们需要在开头添加 [CLS] token,然后在结束时(或在每句话后,如果我们有一对句子)添加 [SEP] token。我们将使用 TemplateProcessor 来完成这个任务,但首先我们需要知道词汇表中 [CLS][SEP] tokens 的 ID:

cls_token_id = tokenizer.token_to_id("[CLS]")
sep_token_id = tokenizer.token_to_id("[SEP]")
print(cls_token_id, sep_token_id)
(2, 3)

编写 TemplateProcessor 的模板时,我们必须指定如何处理单个句子和一对句子。对于这两者,我们写下我们想使用的特殊 tokens 第一句(或单句)用 $A 表示,而第二句(如果需要编码一对句子)用 $B 表示。对于这些(特殊 tokens 和句子),我们还需要在冒号后指定相应的 token 类型 ID。

因此,经典的 BERT 模板定义如下:

tokenizer.post_processor = processors.TemplateProcessing(
    single=f"[CLS]:0 $A:0 [SEP]:0",
    pair=f"[CLS]:0 $A:0 [SEP]:0 $B:1 [SEP]:1",
    special_tokens=[("[CLS]", cls_token_id), ("[SEP]", sep_token_id)],
)

请注意,我们需要传递特殊 tokens 的 ID,这样 tokenizer 才能正确地将它们转换为它们的 ID。

添加之后,我们再次对之前的例子进行 tokenization:

encoding = tokenizer.encode("Let's test this tokenizer.")
print(encoding.tokens)
['[CLS]', 'let', "'", 's', 'test', 'this', 'tok', '##eni', '##zer', '.', '[SEP]']

在一对句子中,我们也得到了正确的结果:

encoding = tokenizer.encode("Let's test this tokenizer...", "on a pair of sentences.")
print(encoding.tokens)
print(encoding.type_ids)
['[CLS]', 'let', "'", 's', 'test', 'this', 'tok', '##eni', '##zer', '...', '[SEP]', 'on', 'a', 'pair', 'of', 'sentences', '.', '[SEP]']
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]

我们几乎从头开始构建了这个 tokenizer ——但是还有最后一步:指定一个解码器:

tokenizer.decoder = decoders.WordPiece(prefix="##")

让我们在之前的 encoding 上测试一下它:

tokenizer.decode(encoding.ids)
"let's test this tokenizer... on a pair of sentences."

很好!我们可以将 tokenizer 保存在一个 JSON 文件中,如下所示:

tokenizer.save("tokenizer.json")

然后,我们可以在一个 Tokenizer 对象中使用 from_file() 方法重新加载该文件:

new_tokenizer = Tokenizer.from_file("tokenizer.json")

要在🤗 Transformers 中使用这个 tokenizer 我们需要将它封装在一个 PreTrainedTokenizerFast 类中。我们可以使用通用类(PreTrainedTokenizerFast),或者,如果我们的 tokenizer 对应于一个现有的模型,则可以使用该类(例如这里的 BertTokenizerFast )。如果你使用这个课程来构建一个全新的 tokenizer 并且没有一个现有的模型可以使用,就必须需要使用通用类。

要将构建的 tokenizer 封装在 PreTrainedTokenizerFast 类中,我们可以将我们构建的 tokenizer 作为 tokenizer_object 传入,或者将我们保存的 tokenizer 文件作为 tokenizer_file 传入。要记住的关键一点是,我们需要手动设置所有的特殊 tokens,因为这个类不能从 tokenizer 对象推断出哪个符号是掩码符号, [CLS] 符号等:

from transformers import PreTrainedTokenizerFast

wrapped_tokenizer = PreTrainedTokenizerFast(
    tokenizer_object=tokenizer,
    # tokenizer_file="tokenizer.json", # 也可以从tokenizer文件中加载
    unk_token="[UNK]",
    pad_token="[PAD]",
    cls_token="[CLS]",
    sep_token="[SEP]",
    mask_token="[MASK]",
)

如果你使用的是其他的 tokenizer 类(如 BertTokenizerFast ),你只需要指定那些与默认值不同的特殊符号(这里没有):

from transformers import BertTokenizerFast

wrapped_tokenizer = BertTokenizerFast(tokenizer_object=tokenizer)

然后,你就可以像使用其他的🤗 Transformers tokenizer 一样使用这个 tokenizer 了。你可以使用 save_pretrained() 方法来保存它,或者使用 push_to_hub() 方法将它上传到 Hub。

既然我们已经看到了如何构建一个 WordPiece tokenizer。那么让我们也尝试构建 BPE tokenizer。这次我们会快一些,因为你已经知道所有的步骤,我们主要强调其中的区别。

从头开始构建 BPE tokenizer

现在让我们构建一个 GPT-2 tokenizer,与 BERT tokenizer 一样,我们首先通过 BPE model 初始化一个 Tokenizer

tokenizer = Tokenizer(models.BPE())

同样,类似于 BERT,如果我们已经有一个词汇表,我们也可以使用这个词汇表来初始化 GPT 模型(在这种情况下,我们需要传入 vocabmerges 参数),但是因为我们将从头开始训练,所以我们不需要做这个。我们也不需要指定 unk_token ,因为 GPT-2 使用字节级 BPE,这不需要它。

GPT-2 不使用 normalizer ,因此我们跳过该步骤并直接进入预分词:

tokenizer.pre_tokenizer = pre_tokenizers.ByteLevel(add_prefix_space=False)

我们在这里给 ByteLevel 添加的选项的含义是不在句子的开始添加空格(默认为 ture)。我们可以看一下之前的示例文本经过预分词后的结果:

tokenizer.pre_tokenizer.pre_tokenize_str("Let's test pre-tokenization!")
[('Let', (0, 3)), ("'s", (3, 5)), ('Ġtest', (5, 10)), ('Ġpre', (10, 14)), ('-', (14, 15)),
 ('tokenization', (15, 27)), ('!', (27, 28))]

接下来是需要训练的模型。对于 GPT-2,唯一的特殊符号是文本结束符:

trainer = trainers.BpeTrainer(vocab_size=25000, special_tokens=["<|endoftext|>"])
tokenizer.train_from_iterator(get_training_corpus(), trainer=trainer)

就像 WordPieceTrainer 一样,除了 vocab_sizespecial_tokens ,我们也可以设置 min_frequency ,或者如果我们需要添加一个词尾后缀(如 </w> ),我们可以用 end_of_word_suffix 设置它。

这个 tokenizer 也可以在本地的文本文件上训练:

tokenizer.model = models.BPE()
tokenizer.train(["wikitext-2.txt"], trainer=trainer)

让我们看一下示例文本经过 tokenization 后的结果:

encoding = tokenizer.encode("Let's test this tokenizer.")
print(encoding.tokens)
['L', 'et', "'", 's', 'Ġtest', 'Ġthis', 'Ġto', 'ken', 'izer', '.']

我们对 GPT-2 tokenizer 添加字节级后处理,如下所示:

tokenizer.post_processor = processors.ByteLevel(trim_offsets=False)

trim_offsets = False 这个选项告诉 post-processor,我们应该让那些以‘Ġ’开始的 tokens 的偏移量保持不变:这样,偏移量起始的索引将指向单词前的空格,而不是单词的第一个字符(因为空格在技术上是 token 的一部分)。让我们看一下我们编码示例文本的结果,其中 'Ġtest' 是索引 4 的 token

sentence = "Let's test this tokenizer."
encoding = tokenizer.encode(sentence)
start, end = encoding.offsets[4]
sentence[start:end]
' test'

最后,我们添加一个字节级解码器:

tokenizer.decoder = decoders.ByteLevel()

我们可以再次检查它是否工作正常:

tokenizer.decode(encoding.ids)
"Let's test this tokenizer."

太好了!现在我们完成了,我们可以像之前一样保存 tokenizer,并且如果我们想在🤗 Transformers 中使用它,可以将它封装在 PreTrainedTokenizerFast 类或者 GPT2TokenizerFast 类中:

from transformers import PreTrainedTokenizerFast

wrapped_tokenizer = PreTrainedTokenizerFast(
    tokenizer_object=tokenizer,
    bos_token="<|endoftext|>",
    eos_token="<|endoftext|>",
)

或者:

from transformers import GPT2TokenizerFast

wrapped_tokenizer = GPT2TokenizerFast(tokenizer_object=tokenizer)

作为最后一个示例,我们将向你展示如何从零开始构建 Unigram tokenizer

从零开始构建 Unigram tokenizer

现在让我们构建一个 XLNet tokenizer 与之前的 tokenizer 一样,我们首先使用 Unigram model 初始化一个 Tokenizer

tokenizer = Tokenizer(models.Unigram())

同样,如果我们有词汇表,我们可以用词汇表初始化这个模型。

在标准化步骤,XLNet 进行了一些替换(来自 SentencePiece 算法):

from tokenizers import Regex

tokenizer.normalizer = normalizers.Sequence(
    [
        normalizers.Replace("``", '"'),
        normalizers.Replace("''", '"'),
        normalizers.NFKD(),
        normalizers.StripAccents(),
        normalizers.Replace(Regex(" {2,}"), " "),
    ]
)

这会将替换为,将任何连续两个或更多的空格替换为一个空格,同时还将去掉待分词文本中的重音符号。

任何 SentencePiece tokenizer 使用的预 tokenizer 是 Metaspace

tokenizer.pre_tokenizer = pre_tokenizers.Metaspace()

我们可以像以前一样查看示例文本的预分词:

tokenizer.pre_tokenizer.pre_tokenize_str("Let's test the pre-tokenizer!")
[("▁Let's", (0, 5)), ('▁test', (5, 10)), ('▁the', (10, 14)), ('▁pre-tokenizer!', (14, 29))]

接下来是需要训练的模型。XLNet 有不少特殊的 tokens

special_tokens = ["<cls>", "<sep>", "<unk>", "<pad>", "<mask>", "<s>", "</s>"]
trainer = trainers.UnigramTrainer(
    vocab_size=25000, special_tokens=special_tokens, unk_token="<unk>"
)
tokenizer.train_from_iterator(get_training_corpus(), trainer=trainer)

对于 UnigramTrainer 来说,一个非常重要的参数是 unk_token 。我们也可以传递一些 Unigram 算法独有的其他参数,例如我们可以设置每个删除 token 时的 shrinking_factor (默认为 0.75),或者指定 token 最大长度的 max_piece_length (默认为 16)。

这个 tokenizer 也可以在本地的文本文件上训练:

tokenizer.model = models.Unigram()
tokenizer.train(["wikitext-2.txt"], trainer=trainer)

让我们看一下示例文本的 tokenization 后的结果:

encoding = tokenizer.encode("Let's test this tokenizer.")
print(encoding.tokens)
['▁Let', "'", 's', '▁test', '▁this', '▁to', 'ken', 'izer', '.']

XLNet 的一个特点是它将 <cls> token 放在句子的末尾,token 类型 ID 为 2(以区别于其他 tokens)。因此,它在左边进行填充。我们可以像对待 BERT 一样,用模板处理所有特殊 tokens 和 tokens 类型 ID,但首先我们需要获取 <cls><sep> tokens 的 ID:

cls_token_id = tokenizer.token_to_id("<cls>")
sep_token_id = tokenizer.token_to_id("<sep>")
print(cls_token_id, sep_token_id)
0 1

模板如下所示:

tokenizer.post_processor = processors.TemplateProcessing(
    single="$A:0 <sep>:0 <cls>:2",
    pair="$A:0 <sep>:0 $B:1 <sep>:1 <cls>:2",
    special_tokens=[("<sep>", sep_token_id), ("<cls>", cls_token_id)],
)

我们可以通过编码一对句子来测试它是否有效:

encoding = tokenizer.encode("Let's test this tokenizer...", "on a pair of sentences!")
print(encoding.tokens)
print(encoding.type_ids)
['▁Let', "'", 's', '▁test', '▁this', '▁to', 'ken', 'izer', '.', '.', '.', '<sep>', '▁', 'on', '▁', 'a', '▁pair', 
  '▁of', '▁sentence', 's', '!', '<sep>', '<cls>']
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2]

最后,我们添加一个 Metaspace 解码器:

tokenizer.decoder = decoders.Metaspace()

我们完成了这个 tokenizer。我们可以像保存其他 tokenizer 一样保存它。如果我们想在 🤗 Transformers 中使用它,可以将它封装在 PreTrainedTokenizerFast 类或 XLNetTokenizerFast 类中。使用 PreTrainedTokenizerFast 类时需要注意的一点是,除了特殊 tokens 之外,我们还需要告诉🤗 Transformers 库在左边进行填充:

from transformers import PreTrainedTokenizerFast

wrapped_tokenizer = PreTrainedTokenizerFast(
    tokenizer_object=tokenizer,
    bos_token="<s>",
    eos_token="</s>",
    unk_token="<unk>",
    pad_token="<pad>",
    cls_token="<cls>",
    sep_token="<sep>",
    mask_token="<mask>",
    padding_side="left",
)

或者:

from transformers import XLNetTokenizerFast

wrapped_tokenizer = XLNetTokenizerFast(tokenizer_object=tokenizer)

现在你已经了解了如何使用各种模块来构建现有的 tokenizer,你应该能够使用 🤗 tokenizer 库编写你想要的任何 tokenizer 并能够在 🤗 Transformers 中使用它。

< > Update on GitHub