Transformers documentation

用于推理的多语言模型

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v4.46.3).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

用于推理的多语言模型

🤗 Transformers 中有多种多语言模型,它们的推理用法与单语言模型不同。但是,并非所有的多语言模型用法都不同。一些模型,例如 google-bert/bert-base-multilingual-uncased 就可以像单语言模型一样使用。本指南将向您展示如何使用不同用途的多语言模型进行推理。

XLM

XLM 有十个不同的检查点,其中只有一个是单语言的。剩下的九个检查点可以归为两类:使用语言嵌入的检查点和不使用语言嵌入的检查点。

带有语言嵌入的 XLM

以下 XLM 模型使用语言嵌入来指定推理中使用的语言:

  • FacebookAI/xlm-mlm-ende-1024 (掩码语言建模,英语-德语)
  • FacebookAI/xlm-mlm-enfr-1024 (掩码语言建模,英语-法语)
  • FacebookAI/xlm-mlm-enro-1024 (掩码语言建模,英语-罗马尼亚语)
  • FacebookAI/xlm-mlm-xnli15-1024 (掩码语言建模,XNLI 数据集语言)
  • FacebookAI/xlm-mlm-tlm-xnli15-1024 (掩码语言建模+翻译,XNLI 数据集语言)
  • FacebookAI/xlm-clm-enfr-1024 (因果语言建模,英语-法语)
  • FacebookAI/xlm-clm-ende-1024 (因果语言建模,英语-德语)

语言嵌入被表示一个张量,其形状与传递给模型的 input_ids 相同。这些张量中的值取决于所使用的语言,并由分词器的 lang2idid2lang 属性识别。

在此示例中,加载 FacebookAI/xlm-clm-enfr-1024 检查点(因果语言建模,英语-法语):

>>> import torch
>>> from transformers import XLMTokenizer, XLMWithLMHeadModel

>>> tokenizer = XLMTokenizer.from_pretrained("FacebookAI/xlm-clm-enfr-1024")
>>> model = XLMWithLMHeadModel.from_pretrained("FacebookAI/xlm-clm-enfr-1024")

分词器的 lang2id 属性显示了该模型的语言及其对应的id:

>>> print(tokenizer.lang2id)
{'en': 0, 'fr': 1}

接下来,创建一个示例输入:

>>> input_ids = torch.tensor([tokenizer.encode("Wikipedia was used to")])  # batch size 为 1

将语言 id 设置为 "en" 并用其定义语言嵌入。语言嵌入是一个用 0 填充的张量,这个张量应该与 input_ids 大小相同。

>>> language_id = tokenizer.lang2id["en"]  # 0
>>> langs = torch.tensor([language_id] * input_ids.shape[1])  # torch.tensor([0, 0, 0, ..., 0])

>>> # 我们将其 reshape 为 (batch_size, sequence_length) 大小
>>> langs = langs.view(1, -1)  # 现在的形状是 [1, sequence_length] (我们的 batch size 为 1)

现在,你可以将 input_ids 和语言嵌入传递给模型:

>>> outputs = model(input_ids, langs=langs)

run_generation.py 脚本可以使用 xlm-clm 检查点生成带有语言嵌入的文本。

不带语言嵌入的 XLM

以下 XLM 模型在推理时不需要语言嵌入:

  • FacebookAI/xlm-mlm-17-1280 (掩码语言建模,支持 17 种语言)
  • FacebookAI/xlm-mlm-100-1280 (掩码语言建模,支持 100 种语言)

与之前的 XLM 检查点不同,这些模型用于通用句子表示。

BERT

以下 BERT 模型可用于多语言任务:

  • google-bert/bert-base-multilingual-uncased (掩码语言建模 + 下一句预测,支持 102 种语言)
  • google-bert/bert-base-multilingual-cased (掩码语言建模 + 下一句预测,支持 104 种语言)

这些模型在推理时不需要语言嵌入。它们应该能够从上下文中识别语言并进行相应的推理。

XLM-RoBERTa

以下 XLM-RoBERTa 模型可用于多语言任务:

  • FacebookAI/xlm-roberta-base (掩码语言建模,支持 100 种语言)
  • FacebookAI/xlm-roberta-large (掩码语言建模,支持 100 种语言)

XLM-RoBERTa 使用 100 种语言的 2.5TB 新创建和清理的 CommonCrawl 数据进行了训练。与之前发布的 mBERT 或 XLM 等多语言模型相比,它在分类、序列标记和问答等下游任务上提供了更强大的优势。

M2M100

以下 M2M100 模型可用于多语言翻译:

  • facebook/m2m100_418M (翻译)
  • facebook/m2m100_1.2B (翻译)

在此示例中,加载 facebook/m2m100_418M 检查点以将中文翻译为英文。你可以在分词器中设置源语言:

>>> from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer

>>> en_text = "Do not meddle in the affairs of wizards, for they are subtle and quick to anger."
>>> chinese_text = "不要插手巫師的事務, 因為他們是微妙的, 很快就會發怒."

>>> tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M", src_lang="zh")
>>> model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")

对文本进行分词:

>>> encoded_zh = tokenizer(chinese_text, return_tensors="pt")

M2M100 强制将目标语言 id 作为第一个生成的标记,以进行到目标语言的翻译。在 generate 方法中将 forced_bos_token_id 设置为 en 以翻译成英语:

>>> generated_tokens = model.generate(**encoded_zh, forced_bos_token_id=tokenizer.get_lang_id("en"))
>>> tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
'Do not interfere with the matters of the witches, because they are delicate and will soon be angry.'

MBart

以下 MBart 模型可用于多语言翻译:

  • facebook/mbart-large-50-one-to-many-mmt (一对多多语言机器翻译,支持 50 种语言)
  • facebook/mbart-large-50-many-to-many-mmt (多对多多语言机器翻译,支持 50 种语言)
  • facebook/mbart-large-50-many-to-one-mmt (多对一多语言机器翻译,支持 50 种语言)
  • facebook/mbart-large-50 (多语言翻译,支持 50 种语言)
  • facebook/mbart-large-cc25

在此示例中,加载 facebook/mbart-large-50-many-to-many-mmt 检查点以将芬兰语翻译为英语。 你可以在分词器中设置源语言:

>>> from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

>>> en_text = "Do not meddle in the affairs of wizards, for they are subtle and quick to anger."
>>> fi_text = "Älä sekaannu velhojen asioihin, sillä ne ovat hienovaraisia ja nopeasti vihaisia."

>>> tokenizer = AutoTokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt", src_lang="fi_FI")
>>> model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")

对文本进行分词:

>>> encoded_en = tokenizer(en_text, return_tensors="pt")

MBart 强制将目标语言 id 作为第一个生成的标记,以进行到目标语言的翻译。在 generate 方法中将 forced_bos_token_id 设置为 en 以翻译成英语:

>>> generated_tokens = model.generate(**encoded_en, forced_bos_token_id=tokenizer.lang_code_to_id["en_XX"])
>>> tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
"Don't interfere with the wizard's affairs, because they are subtle, will soon get angry."

如果你使用的是 facebook/mbart-large-50-many-to-one-mmt 检查点,则无需强制目标语言 id 作为第一个生成的令牌,否则用法是相同的。

< > Update on GitHub