乐趣区

关于asr:微调用于多语言-ASR-的-MMS-适配器模型

新内容 (06/2023): 这篇博文受到“在多语言 ASR 上微调 XLS-R”的强烈启发,能够看作是它的改良版本。

Wav2Vec2 是主动语音辨认 (ASR) 的预训练模型,由 Alexei Baevski、Michael AuliAlex Conneau 于 2020 年 9 月 公布。其在最风行的 ASR 英语数据集之一 LibriSpeech 上展现了 Wav2Vec2 的弱小性能后不久,Facebook AI 就推出了 Wav2Vec2 的两个多语言版本,称为 XLSR 和 XLM-R,可能辨认多达 128 种语言的语音。XLSR 代表 跨语言语音示意,指的是模型学习跨多种语言有用的语音示意的能力。

Meta AI 的最新版本,大规模多语言语音 (MMS),由 Vineel Pratap、Andros Tjandra、Bowen Shi 等人编写。将多语言语音示意晋升到一个新的程度。通过公布的各种 语言辨认、语音辨认和文本转语音检查点,能够辨认、转录和生成超过 1,100 多种书面语。

在这篇博文中,咱们展现了 MMS 的适配器训练如何在短短 10-20 分钟的微调后实现惊人的低单词错误率。

对于资源匮乏的语言,咱们 强烈 倡议应用 MMS 的适配器训练,而不是像“在多语言 ASR 上微调 XLS-R”中那样微调整个模型。

在咱们的试验中,MMS 的适配器训练不仅内存效率更高、更持重,而且对于低资源语言也能产生更好的性能。对于中到高资源语言,微调整个检查点而不是应用适配器层依然是无利的。

爱护世界语言多样性

依据 https://www.ethnologue.com/ 的数据,大概 3000 种语言 (即所有“现存”语言的 40%) 因为母语人士越来越少而濒临灭绝。这种趋势只会在日益全球化的世界中继续上来。

MMS 可能转录许多濒临灭绝的语言,例如 AriKaivi。将来,MMS 能够通过帮忙残余的使用者创立书面记录并用母语进行交换,这在放弃语言生机方面施展至关重要的作用。

为了适应 1000 多个不同的词汇表,MMS 应用适配器 (Adapters) – 一种仅训练一小部分模型权重的训练方法。

适配器层就像语言桥梁一样,使模型可能在解读另一种语言时利用一种语言的常识。

微调 MMS

MMS 无监督检查点应用 1,400 多种语言的超过 50 万 小时的音频进行了预训练,参数范畴从 3 亿到 10 亿不等。

你能够在 🤗 Hub 上找到 3 亿个参数 (300M) 和 10 亿个参数 (1B) 模型大小的仅预训练检查点:

  • mms-300m
  • mms-1b

留神 : 如果你想微调根本模型,能够依照“在多语言 ASR 上微调 XLS-R”中所示的完全相同的形式进行操作。

与 BERT 的掩码语言建模指标 相似,MMS 通过随机遮蔽特征向量来学习上下文语音示意,而后在自监督预训练期间将其传递到 Transformer 网络。

对于 ASR,预训练 MMS-1B 检查点 通过联结词汇输入层以监督形式对 1000 多种语言进行了进一步微调。最初一步,联结词汇输入层被抛弃,并保留特定于语言的适配器层。每个适配器层 蕴含约 2.5M 权重,由每个注意力块的小型线性投影层以及特定于语言的词汇输入层组成。

已公布针对语音辨认 (ASR) 进行微调的三个 MMS 检查点。它们别离包含 102、1107 和 1162 个适配器权重 (每种语言一个):

  • mms-1b-fl102
  • mms-1b-l1107
  • mms-1b-all

你能够看到根本模型 (像平常一样) 保留为文件 model.safetensors,但此外这些存储库还存储了许多适配器权重,例如 针对法国的 adapter.fra.safetensors

Hugging Face 文档很好地 解释了如何应用此类检查点进行推理,因而在这篇博文中,咱们将重点学习如何基于任何已公布的 ASR 检查点无效地训练高性能适配器模型。

训练自适应权重

在机器学习中,适配器是一种用于微调预训练模型同时放弃原始模型参数不变的办法。他们通过在模型的现有层之间插入小型可训练模块 (称为 适配器层) 来实现此目标,而后使模型适应特定工作,而无需进行大量的从新训练。

适配器在语音辨认,尤其是 谈话人辨认 方面有着悠久的历史。在谈话人辨认中,适配器已被无效地用于调整事后存在的模型,以辨认单个谈话人的特质,正如 Gales 和 Woodland (1996) 以及 Miao 等人 (2014) 的工作中所强调的那样。与训练残缺模型相比,这种办法不仅大大降低了计算要求,而且使得特定于谈话者的调整更好、更灵便。

MMS 中实现的工作利用了跨不同语言的语音辨认适配器的想法。对大量适配器权重进行了微调,以把握每种目标语言独特的语音和语法特色。因而,MMS 使单个大型根底模型 (_例如_ mms-1b-all 模型检查点) 和 1000 多个小型适配器层 (每个 2.5M 权重 mms-1b-all) 可能了解和转录多种语言。这极大地缩小了为每种语言开发不同模型的计算需要。

棒极了!当初咱们理解其动机和实践,上面让咱们钻研一下 mms-1b-all 🔥的适配器权重微调

Notebook 设置

正如之前在“多语言 ASR 上微调 XLS-R”博客文章中所做的那样,咱们在 Common Voice 的低资源 ASR 数据集上微调模型,该数据集仅蕴含 ca. 4 小时通过验证的训练数据。

就像 Wav2Vec2 或 XLS-R 一样,MMS 应用连贯时序分类 (CTC) 进行微调,CTC 是一种用于训练神经网络解决序列到序列问题 (例如 ASR 和手写辨认) 的算法。

无关 CTC 算法的更多详细信息,我强烈建议浏览 Awni Hannun 的写得很好的一篇博客文章 Sequence Modeling with CTC (2017)

在咱们开始之前,让咱们装置 datasetstransformers。此外,咱们须要 torchaudio 来加载音频文件,以及应用 字错误率 (WER) 指标 ({}^1 ) 评估咱们微调后的模型,因而也须要装置 jiwer

%%capture
!pip install --upgrade pip
!pip install datasets
!pip install evaluate
!pip install git+https://github.com/huggingface/transformers.git
!pip install jiwer
!pip install accelerate

咱们强烈建议你在训练时将训练检查点间接上传到 🤗 Hub。Hub 存储库内置了版本控制,因而你能够确保在训练期间不会失落任何模型检查点。

为此,你必须存储来自 Hugging Face 网站的身份验证令牌 (如果你还没有注册,请在 此处 注册!)

from huggingface_hub import notebook_login

notebook_login()

筹备数据、分词器、特征提取器

ASR 模型将语音转录为文本,这意味着咱们须要一个将语音信号处理为模型输出格局 (例如特征向量) 的特征提取器,以及一个将模型输入格局解决为文本的分词器。

在🤗 Transformers 中,MMS 模型同时随同着一个名为 Wav2Vec2FeatureExtractor 的特征提取器和一个名为 Wav2Vec2CTCTokenizer 的分词器。

咱们首先创立标记生成器,将预测的输入类解码为输入转录。

创立 Wav2Vec2CTCTokenizer

微调的 MMS 模型,例如 mms-1b-all 曾经有一个伴随模型检查点的 分词器。然而,因为咱们想要在某种语言的特定低资源数据上微调模型,因而倡议齐全删除分词器和词汇输入层,并依据训练数据自身创立新的。

在 CTC 上微调的相似 Wav2Vec2 的模型通过一次前向传递来转录音频文件,首先将音频输出解决为一系列通过解决的上下文示意,而后应用最终的词汇输入层将每个上下文示意分类为示意该字符的字符转录。

该层的输入大小对应于词汇表中的标记数量,咱们将从用于微调的标记数据集中提取该词汇表。因而,第一步,咱们将查看所选的 Common Voice 数据集,并依据转录定义词汇表。

对于本 notebook,咱们将应用 Common Voice 的 6.1 土耳其语数据集。土耳其语对应于语言代码 "tr"

太好了,当初咱们能够应用 🤗 Datasets 的简略 API 来下载数据了。数据集名称是 "mozilla-foundation/common_voice_6_1",配置名称对应于语言代码,在咱们的例子中是 "tr"

留神: 在下载数据集之前,你必须登录你的 Hugging Face 帐户,进入 数据集存储库页 面并单击“批准并拜访存储库”来拜访它

Common Voice 有许多不同的宰割,其中包含 invalidated,它指的是未被评为“足够洁净”而被认为有用的数据。在此 notebook 中,咱们将仅应用拆分的 "train", "validation""test"

from datasets import load_dataset, load_metric, Audio

common_voice_train = load_dataset("mozilla-foundation/common_voice_6_1", "tr", split="train+validation", use_auth_token=True)
common_voice_test = load_dataset("mozilla-foundation/common_voice_6_1", "tr", split="test", use_auth_token=True)

许多 ASR 数据集仅提供每个音频数组 ('audio') 和文件 ('path') 的指标文本 ('sentence')。实际上,Common Voice 提供了对于每个音频文件的更多信息,例如 'accent' 等。为了使 notebook 尽可能通用,咱们仅思考用于微调的转录文本。

common_voice_train = common_voice_train.remove_columns(["accent", "age", "client_id", "down_votes", "gender", "locale", "segment", "up_votes"])
common_voice_test = common_voice_test.remove_columns(["accent", "age", "client_id", "down_votes", "gender", "locale", "segment", "up_votes"])

让咱们编写一个简短的函数来显示数据集的一些随机样本,并运行它几次以理解转录的感觉。

from datasets import ClassLabel
import random
import pandas as pd
from IPython.display import display, HTML

def show_random_elements(dataset, num_examples=10):
    assert num_examples <= len(dataset), "Can't pick more elements than there are in the dataset."
    picks = []
    for _ in range(num_examples):
        pick = random.randint(0, len(dataset)-1)
        while pick in picks:
            pick = random.randint(0, len(dataset)-1)
        picks.append(pick)

    df = pd.DataFrame(dataset[picks])
    display(HTML(df.to_html()))
show_random_elements(common_voice_train.remove_columns(["path", "audio"]), num_examples=10)
Oylar teker teker elle sayılacak.
Son olaylar endişe seviyesini yükseltti.
Tek bir kart hepsinin kapılarını açıyor.
Blogcular da tam bundan bahsetmek istiyor.
Bu Aralık iki bin onda oldu.
Fiyatın altmış altı milyon avro olduğu bildirildi.
Ardından da silahlı çatışmalar çıktı.
"Romanya'da kurumlar gelir vergisi oranı yüzde on altı."
Bu konuda neden bu kadar az şey söylendiğini açıklayabilir misiniz?

好吧!转录看起来相当洁净。翻译完转录的句子后,这种语言仿佛更多地对应于书面文本,而不是嘈杂的对话。思考到 Common Voice 是一个众包浏览语音语料库,这也解释的通。

咱们能够看到,转录文本中蕴含一些特殊字符,如 ,.?!;:。没有语言模型,要将语音块分类为这些特殊字符就更难了,因为它们并不真正对应于一个特征性的声音单元。例如,字母 "s" 有一个或多或少清晰的声音,而特殊字符 "." 则没有。此外,为了了解语音信号的含意,通常不须要在转录中蕴含特殊字符。

让咱们简略地删除所有对单词的含意没有奉献并且不能真正用声音示意的字符,并对文本进行规范化。

import re
chars_to_remove_regex = '[\,\?\.\!\-\;\:\"\“\%\‘\”\�\']'

def remove_special_characters(batch):
    batch["sentence"] = re.sub(chars_to_remove_regex, '', batch["sentence"]).lower()
    return batch
common_voice_train = common_voice_train.map(remove_special_characters)
common_voice_test = common_voice_test.map(remove_special_characters)

咱们再看看解决后的文本标签。

show_random_elements(common_voice_train.remove_columns(["path","audio"]))
i̇kinci tur müzakereler eylül ayında başlayacak
jani ve babası bu düşüncelerinde yalnız değil
onurun gözlerindeki büyü
bandiç oyların yüzde kırk sekiz virgül elli dördünü topladı
bu imkansız
bu konu açık değildir
cinayet kamuoyunu şiddetle sarstı
kentin sokakları iki metre su altında kaldı
muhalefet partileri hükümete karşı ciddi bir mücadele ortaya koyabiliyorlar mı
festivale tüm dünyadan elli film katılıyor

好!这看起来更好了。咱们曾经从转录中删除了大多数特殊字符,并将它们规范化为仅小写。

在实现预处理之前,征询目标语言的母语人士总是无益的,以查看文本是否能够进一步简化。
对于这篇博客文章,Merve 很敌对地疾速查看了一下,并指出带帽子的字符 (如 â) 在土耳其语中曾经不再应用,能够用它们的无帽子等效物 (例如 a) 替换。

这意味着咱们应该将像 "yargı sistemi hâlâ sağlıksız" 这样的句子替换为 "yargı sistemi hala sağlıksız"

让咱们再写一个简短的映射函数来进一步简化文本标签。记住 – 文本标签越简略,模型学习预测这些标签就越容易。

def replace_hatted_characters(batch):
    batch["sentence"] = re.sub('[â]', 'a', batch["sentence"])
    batch["sentence"] = re.sub('[î]', 'i', batch["sentence"])
    batch["sentence"] = re.sub('[ô]', 'o', batch["sentence"])
    batch["sentence"] = re.sub('[û]', 'u', batch["sentence"])
    return batch
common_voice_train = common_voice_train.map(replace_hatted_characters)
common_voice_test = common_voice_test.map(replace_hatted_characters)

在 CTC 中,将语音块分类为字母是很常见的,所以咱们在这里也做同样的事件。让咱们提取训练和测试数据中所有不同的字母,并从这组字母中构建咱们的词汇表。

咱们编写一个映射函数,将所有转录连接成一个长转录,而后将字符串转换为一组字符。将参数传递 batched=Truemap(...) 函数十分重要,以便映射函数能够立刻拜访所有转录。

def extract_all_chars(batch):
  all_text = "".join(batch["sentence"])
  vocab = list(set(all_text))
  return {"vocab": [vocab], "all_text": [all_text]}
vocab_train = common_voice_train.map(extract_all_chars, batched=True, batch_size=-1, keep_in_memory=True, remove_columns=common_voice_train.column_names)
vocab_test = common_voice_test.map(extract_all_chars, batched=True, batch_size=-1, keep_in_memory=True, remove_columns=common_voice_test.column_names)

当初,咱们创立训练数据集和测试数据集中所有不同字母的并集,并将后果列表转换为枚举字典。

vocab_list = list(set(vocab_train["vocab"][0]) | set(vocab_test["vocab"][0]))
vocab_dict = {v: k for k, v in enumerate(sorted(vocab_list))}
vocab_dict
    {'': 0,'a': 1,'b': 2,'c': 3,'d': 4,'e': 5,'f': 6,'g': 7,'h': 8,'i': 9,'j': 10,'k': 11,'l': 12,'m': 13,'n': 14,'o': 15,'p': 16,'q': 17,'r': 18,'s': 19,'t': 20,'u': 21,'v': 22,'w': 23,'x': 24,'y': 25,'z': 26,'ç': 27,'ë': 28,'ö': 29,'ü': 30,'ğ': 31,'ı': 32,'ş': 33,'̇': 34}

很酷,咱们看到字母表中的所有字母都呈现在数据集中 (这并不令人诧异),咱们还提取了特殊字符 ""'。请留神,咱们没有排除这些特殊字符,因为模型必须学会预测单词何时完结,否则预测将始终是一系列字母,这将使得不可能将单词彼此离开。

人们应该始终记住,在训练模型之前,预处理是一个十分重要的步骤。例如,咱们不心愿咱们的模型仅仅因为咱们遗记规范化数据而辨别 aAaA 之间的区别基本不取决于字母的“声音”,而更多地取决于语法规定 – 例如,在句子结尾应用大写字母。因而,删除大写字母和非大写字母之间的差别是理智的,这样模型在学习转录语音时就更容易了。

为了更分明地表明 " " 具备本人的标记类别,咱们给它一个更显著的字符 |。此外,咱们还增加了一个“未知”标记,以便模型当前可能解决 Common Voice 训练集中未遇到的字符。

vocab_dict["|"] = vocab_dict[" "]
del vocab_dict[" "]

最初,咱们还增加了一个对应于 CTC 的“空白标记”的填充标记。“空白标记”是 CTC 算法的外围组成部分。欲了解更多信息,请查看 此处 的“对齐”局部。

vocab_dict["[UNK]"] = len(vocab_dict)
vocab_dict["[PAD]"] = len(vocab_dict)
len(vocab_dict)
    37

很酷,当初咱们的词汇表曾经实现,蕴含 37 个标记,这意味着咱们将作为适配器权重的一部分增加在预训练的 MMS 检查点顶部的线性层将具备 37 的输入维度。

因为单个 MMS 检查点能够为多种语言提供定制权重,因而分词器也能够蕴含多个词汇表。因而,咱们须要嵌套咱们的 vocab_dict,以便未来可能向词汇表中增加更多语言。字典应该嵌套应用适配器权重的名称,并在分词器配置中以 target_lang 的名称保留。

让咱们像原始的 mms-1b-all 检查点一样应用 ISO-639-3 语言代码。

target_lang = "tur"

让咱们定义一个空字典,咱们能够在其中增加刚刚创立的词汇表

new_vocab_dict = {target_lang: vocab_dict}

留神 : 如果你想应用此 notebook 将新的适配器层增加到 现有模型仓库 ,请确保 不要 创立一个空的新词汇表,而是重用曾经存在的词汇表。为此,你应该勾销正文以下单元格,并将 "patrickvonplaten/wav2vec2-large-mms-1b-turkish-colab" 替换为你要增加适配器权重的模型仓库 ID。

# from transformers import Wav2Vec2CTCTokenizer

# mms_adapter_repo = "patrickvonplaten/wav2vec2-large-mms-1b-turkish-colab" # make sure to replace this path with a repo to which you want to add your new adapter weights

# tokenizer = Wav2Vec2CTCTokenizer.from_pretrained(mms_adapter_repo)
# new_vocab = tokenizer.vocab

# new_vocab[target_lang] = vocab_dict

当初让咱们将词汇表保留为 json 文件。

import json
with open('vocab.json', 'w') as vocab_file:
    json.dump(new_vocab_dict, vocab_file)

最初一步,咱们应用 json 文件将词汇表加载到类的实例中 Wav2Vec2CTCTokenizer

from transformers import Wav2Vec2CTCTokenizer

tokenizer = Wav2Vec2CTCTokenizer.from_pretrained("./", unk_token="[UNK]", pad_token="[PAD]", word_delimiter_token="|", target_lang=target_lang)

如果想要在本 notebook 的微调模型中重用刚刚创立的分词器,强烈建议将 tokenizer 上传到 🤗 Hub。让咱们将上传文件的仓库命名为 "wav2vec2-large-mms-1b-turkish-colab":

repo_name = "wav2vec2-large-mms-1b-turkish-colab"

并将分词器上传到 🤗 Hub。

tokenizer.push_to_hub(repo_name)
    CommitInfo(commit_url='https://huggingface.co/patrickvonplaten/wav2vec2-large-mms-1b-turkish-colab/commit/48cccbfd6059aa6ce655e9d94b8358ba39536cb7', commit_message='Upload tokenizer', commit_description='', oid='48cccbfd6059aa6ce655e9d94b8358ba39536cb7', pr_url=None, pr_revision=None, pr_num=None)

太好了,你能够在上面看到刚刚创立的存储库 https://huggingface.co/<your-username>/wav2vec2-large-mms-1b-tr-colab

创立 Wav2Vec2FeatureExtractor

语音是一个间断的信号,要被计算机解决,首先必须离散化,这通常被称为 采样 。采样率在这里起着重要的作用,它定义了每秒测量语音信号的数据点数。因而,采纳更高的采样率采样会更好地近似 实在 语音信号,但也须要每秒更多的值。

预训练检查点冀望其输出数据与其训练数据的散布大致相同。两个不同采样率采样的雷同语音信号具备十分不同的散布,例如,将采样率加倍会导致数据点数量加倍。因而,在微调 ASR 模型的预训练检查点之前,必须验证用于预训练模型的数据的采样率与用于微调模型的数据集的采样率是否匹配。Wav2Vec2FeatureExtractor 对象须要以下参数能力实例化:

  • feature_size: 语音模型以特征向量序列作为输出。尽管这个序列的长度显然会变动,但特色大小不应该变动。在 Wav2Vec2 的状况下,特色大小为 1,因为该模型是在原始语音信号上训练的 ({}^2 )。
  • sampling_rate: 模型训练时应用的采样率。
  • padding_value: 对于批量推理,较短的输出须要用特定值填充
  • do_normalize: 输出是否应该进行 零均值单位方差 归一化。通常,语音模型在归一化输出时体现更好
  • return_attention_mask: 模型是否应该应用 attention_mask 进行批量推理。通常状况下,XLS-R 模型检查点应该 始终 应用 attention_mask
from transformers import Wav2Vec2FeatureExtractor

feature_extractor = Wav2Vec2FeatureExtractor(feature_size=1, sampling_rate=16000, padding_value=0.0, do_normalize=True, return_attention_mask=True)

太好了,MMS 的特征提取管道曾经齐全定义!

为了进步用户敌对性,特征提取器和分词器被 封装 到一个 Wav2Vec2Processor 类中,这样只须要一个 modelprocessor 对象。

from transformers import Wav2Vec2Processor

processor = Wav2Vec2Processor(feature_extractor=feature_extractor, tokenizer=tokenizer)

接下来,咱们能够筹备数据集。

预处理数据

到目前为止,咱们还没有看过语音信号的理论值,只看过转录。除了 sentence,咱们的数据集还包含另外两个列名 pathaudiopath 示意音频文件的绝对路径,audio 示意曾经加载的音频数据。MMS 冀望输出格局为 16kHz 的一维数组。这意味着音频文件必须加载并从新采样。

值得庆幸的是,当列名为 audio 时,datasets 会主动实现这一操作。让咱们试试。

common_voice_train[0]["audio"]
    {'path': '/root/.cache/huggingface/datasets/downloads/extracted/71ba9bd154da9d8c769b736301417178729d2b87b9e00cda59f6450f742ed778/cv-corpus-6.1-2020-12-11/tr/clips/common_voice_tr_17346025.mp3',
     'array': array([ 0.00000000e+00, -2.98378618e-13, -1.59835903e-13, ...,
            -2.01663317e-12, -1.87991593e-12, -1.17969588e-12]),
     'sampling_rate': 48000}

在下面的示例中,咱们能够看到音频数据以 48kHz 的采样率加载,而模型冀望的是 16kHz,正如咱们所见。咱们能够通过应用 cast_column 将音频特色设置为正确的采样率:

common_voice_train = common_voice_train.cast_column("audio", Audio(sampling_rate=16_000))
common_voice_test = common_voice_test.cast_column("audio", Audio(sampling_rate=16_000))

咱们再来看一下 "audio"

common_voice_train[0]["audio"]
{'path': '/root/.cache/huggingface/datasets/downloads/extracted/71ba9bd154da9d8c769b736301417178729d2b87b9e00cda59f6450f742ed778/cv-corpus-6.1-2020-12-11/tr/clips/common_voice_tr_17346025.mp3',
 'array': array([ 9.09494702e-13, -6.13908924e-12, -1.09139364e-11, ...,
         1.81898940e-12, 4.54747351e-13, 3.63797881e-12]),
 'sampling_rate': 16000}

这仿佛见效了!让咱们通过打印语音输入的形态、转录内容和相应的采样率来最初检查数据是否筹备正确。

rand_int = random.randint(0, len(common_voice_train)-1)

print("Target text:", common_voice_train[rand_int]["sentence"])
print("Input array shape:", common_voice_train[rand_int]["audio"]["array"].shape)
print("Sampling rate:", common_voice_train[rand_int]["audio"]["sampling_rate"])
    Target text: bağış anlaşması bir ağustosta imzalandı
    Input array shape:(70656,)
    Sampling rate: 16000

很好!所有看起来都很棒 – 数据是一维数组,采样率始终对应于 16kHz,并且指标文本已标准化。

最初,咱们能够利用 Wav2Vec2Processor 将数据处理成 Wav2Vec2ForCTC 训练所需的格局。为此,让咱们利用 Dataset 的 map(...) 函数。

首先,咱们通过调用 batch["audio"] 来加载并从新采样音频数据。
其次,咱们从加载的音频文件中提取 input_values。在咱们的状况下,Wav2Vec2Processor 只规范化数据。然而,对于其余语音模型,这一步可能包含更简单的特征提取,例如 Log-Mel 特征提取。
第三,咱们将转录编码为标签 id。

留神 : 这个映射函数是一个很好的例子,阐明了如何应用 Wav2Vec2Processor 类。在“失常”状况下,调用 processor(...) 会重定向到 Wav2Vec2FeatureExtractor 的调用办法。然而,当将处理器封装到 as_target_processor 上下文中时,同一个办法会重定向到 Wav2Vec2CTCTokenizer 的调用办法。
欲了解更多信息,请查看 文档。

def prepare_dataset(batch):
    audio = batch["audio"]

    # batched output is "un-batched"
    batch["input_values"] = processor(audio["array"], sampling_rate=audio["sampling_rate"]).input_values[0]
    batch["input_length"] = len(batch["input_values"])

    batch["labels"] = processor(text=batch["sentence"]).input_ids
    return batch

让咱们将数据筹备性能利用到所有示例中。

common_voice_train = common_voice_train.map(prepare_dataset, remove_columns=common_voice_train.column_names)
common_voice_test = common_voice_test.map(prepare_dataset, remove_columns=common_voice_test.column_names)

留神: datasets 主动解决音频加载和从新采样。如果你心愿实现本人的定制数据加载 / 采样,请随便应用该 "path" 列并疏忽该 "audio" 列。

太棒了,当初咱们筹备开始训练了!

训练

数据曾经解决好,咱们筹备开始设置训练流程。咱们将应用 🤗 的 Trainer,为此咱们基本上须要做以下几件事:

  • 定义一个数据整顿器。与大多数 NLP 模型不同,MMS 的输出长度比输入长度大得多。例如,输出长度为 50000 的样本的输入长度不超过 100。鉴于输出大小较大,动静填充训练批次更为高效,这意味着所有训练样本只应填充到其批次中最长的样本,而不是整体最长的样本。因而,微调 MMS 须要一个非凡的填充数据整顿器,咱们将在上面定义它
  • 评估指标。在训练过程中,模型应该依据字错误率进行评估。咱们应该相应地定义一个 compute_metrics 函数
  • 加载预训练检查点。咱们须要加载预训练检查点并正确配置它进行训练。
  • 定义训练配置。

在微调模型之后,咱们将正确地在测试数据上评估它,并验证它是否的确学会了正确转录语音。

设置 Trainer

让咱们从定义数据整顿器开始。数据整顿器的代码是从 这个示例 中复制的。

不具体讲述,与常见的数据整顿器不同,这个数据整顿器别离看待 input_valueslabels,因而对它们利用两个独自的填充函数 (再次利用 MMS 处理器的上下文管理器)。这是必要的,因为在语音辨认中,输出和输入属于不同的模态,因而它们不应该被雷同的填充函数解决。
与常见的数据整顿器相似,标签中的填充标记用 -100 填充,以便在计算损失时 思考这些标记。

import torch

from dataclasses import dataclass, field
from typing import Any, Dict, List, Optional, Union

@dataclass
class DataCollatorCTCWithPadding:
    """
    Data collator that will dynamically pad the inputs received.
    Args:
        processor (:class:`~transformers.Wav2Vec2Processor`)
            The processor used for proccessing the data.
        padding (:obj:`bool`, :obj:`str` or :class:`~transformers.tokenization_utils_base.PaddingStrategy`, `optional`, defaults to :obj:`True`):
            Select a strategy to pad the returned sequences (according to the model's padding side and padding index)
            among:
            *:obj:`True` or :obj:`'longest'`: Pad to the longest sequence in the batch (or no padding if only a single
              sequence if provided).
            *:obj:`'max_length'`: Pad to a maximum length specified with the argument :obj:`max_length` or to the
              maximum acceptable input length for the model if that argument is not provided.
            *:obj:`False` or :obj:`'do_not_pad'` (default): No padding (i.e., can output a batch with sequences of
              different lengths).
    """

    processor: Wav2Vec2Processor
    padding: Union[bool, str] = True

    def __call__(self, features: List[Dict[str, Union[List[int], torch.Tensor]]]) -> Dict[str, torch.Tensor]:
        # split inputs and labels since they have to be of different lenghts and need
        # different padding methods
        input_features = [{"input_values": feature["input_values"]} for feature in features]
        label_features = [{"input_ids": feature["labels"]} for feature in features]

        batch = self.processor.pad(
            input_features,
            padding=self.padding,
            return_tensors="pt",
        )

        labels_batch = self.processor.pad(
            labels=label_features,
            padding=self.padding,
            return_tensors="pt",
        )

        # replace padding with -100 to ignore loss correctly
        labels = labels_batch["input_ids"].masked_fill(labels_batch.attention_mask.ne(1), -100)

        batch["labels"] = labels

        return batch
data_collator = DataCollatorCTCWithPadding(processor=processor, padding=True)

接下来,定义评估指标。如前所述,ASR 中的次要指标是单词错误率 (WER),因而咱们也将在本 notebook 中应用它。

from evaluate import load

wer_metric = load("wer")

模型将返回一系列 logit 向量:
(\mathbf{y}_1, \ldots, \mathbf{y}_m ) 其中 (\mathbf{y} _1 = f_{\theta}(x_1, \ldots, x_n)[0] ) 且 (n >> m)。

logit 向量 (\mathbf{y}_1 ) 蕴含咱们后面定义的词汇表中每个单词的对数几率,因而 (\text{len}(\mathbf{y}_i) = ) config.vocab_size。咱们对模型最可能的预测感兴趣,因而取 logits 的 argmax(...)。此外,咱们通过将 -100 替换为 pad_token_id 并解码 id,同时确保间断标记 以 CTC 格调分组到同一标记 ({}^1 ),将编码后的标签转换回原始字符串。

def compute_metrics(pred):
    pred_logits = pred.predictions
    pred_ids = np.argmax(pred_logits, axis=-1)

    pred.label_ids[pred.label_ids == -100] = processor.tokenizer.pad_token_id

    pred_str = processor.batch_decode(pred_ids)
    # we do not want to group tokens when computing the metrics
    label_str = processor.batch_decode(pred.label_ids, group_tokens=False)

    wer = wer_metric.compute(predictions=pred_str, references=label_str)

    return {"wer": wer}

当初,咱们能够加载预训练的 mms-1b-all 检查点。分词器的 pad_token_id 必须定义模型的 pad_token_id,或者在 Wav2Vec2ForCTC 的状况下也是 CTC 的 空白标记 ({}^2 )。

因为咱们只训练一小部分权重,模型不容易过拟合。因而,咱们确保禁用所有 dropout 层。

留神: 当应用本笔记本在 Common Voice 的另一种语言上训练 MMS 时,这些超参数设置可能不会很好地工作。依据你的用例,随便调整这些设置。

from transformers import Wav2Vec2ForCTC

model = Wav2Vec2ForCTC.from_pretrained(
    "facebook/mms-1b-all",
    attention_dropout=0.0,
    hidden_dropout=0.0,
    feat_proj_dropout=0.0,
    layerdrop=0.0,
    ctc_loss_reduction="mean",
    pad_token_id=processor.tokenizer.pad_token_id,
    vocab_size=len(processor.tokenizer),
    ignore_mismatched_sizes=True,
)
    Some weights of Wav2Vec2ForCTC were not initialized from the model checkpoint at facebook/mms-1b-all and are newly initialized because the shapes did not match:
    - lm_head.bias: found shape torch.Size([154]) in the checkpoint and torch.Size([39]) in the model instantiated
    - lm_head.weight: found shape torch.Size([154, 1280]) in the checkpoint and torch.Size([39, 1280]) in the model instantiated
    You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.

留神: 预计一些权重将被从新初始化。这些权重对应于新初始化的词汇输入层。

咱们当初心愿确保只有适配器权重将被训练,而模型的其余部分放弃解冻。

首先,咱们从新初始化所有适配器权重,这能够通过不便的 init_adapter_layers 办法实现。也能够不从新初始化适配器权重并持续微调,但在这种状况下,在训练之前应该通过 load_adapter(...) 办法 加载适合的适配器权重。然而,词汇表通常依然不会很好地匹配自定义训练数据,因而通常更容易从新初始化所有适配器层,以便它们能够轻松地进行微调。

model.init_adapter_layers()

接下来,咱们解冻 适配器层之外的所有权重。

model.freeze_base_model()

adapter_weights = model._get_adapters()
for param in adapter_weights.values():
    param.requires_grad = True

最初一步,咱们定义与训练相干的所有参数。
对一些参数进行更多解释:

  • group_by_length 通过将输出长度类似的训练样本分组到一个批次中,使训练更加高效。这能够通过大大减少通过模型传递的无用填充标记的总数,从而显著放慢训练工夫
  • learning_rate 被抉择为 1e-3,这是应用 Adam 训练的罕用默认值。其余学习率可能同样无效。

无关其余参数的更多解释,能够查看 文档。为了节俭 GPU 内存,咱们启用 PyTorch 的 梯度检查点,并将损失缩小设置为“mean”。MMS 适配器微调十分快地收敛到十分好的性能,因而即便对于像 4 小时这样小的数据集,咱们也只会训练 4 个周期。在训练过程中,每 200 个训练步骤将异步上传一个检查点到 hub。它容许你在模型仍在训练时也能够应用演示小部件游玩。

留神: 如果不想将模型检查点上传到 hub,只需将 push_to_hub=False 即可。

from transformers import TrainingArguments

training_args = TrainingArguments(
  output_dir=repo_name,
  group_by_length=True,
  per_device_train_batch_size=32,
  evaluation_strategy="steps",
  num_train_epochs=4,
  gradient_checkpointing=True,
  fp16=True,
  save_steps=200,
  eval_steps=100,
  logging_steps=100,
  learning_rate=1e-3,
  warmup_steps=100,
  save_total_limit=2,
  push_to_hub=True,
)

当初,所有实例都能够传递给 Trainer,咱们筹备开始训练!

from transformers import Trainer

trainer = Trainer(
    model=model,
    data_collator=data_collator,
    args=training_args,
    compute_metrics=compute_metrics,
    train_dataset=common_voice_train,
    eval_dataset=common_voice_test,
    tokenizer=processor.feature_extractor,
)

({}^1 ) 为了使模型独立于谈话人速率,在 CTC 中,雷同的间断标记简略地分组为单个标记。然而,在解码时不应该对编码的标签进行分组,因为它们不对应于模型的预测标记,这就是为什么必须传递 group_tokens=False 参数。如果咱们不传递这个参数,像 "hello" 这样的单词会被谬误地编码,并解码为 "helo"

({}^2 ) 空白标记容许模型通过强制在两个 l 之间插入空白标记来预测一个词,例如 "hello"。咱们模型的 CTC 合乎预测 "hello" 将是 [PAD] [PAD]"h" "e" "e" "l" "l" [PAD]"l" "o" "o" [PAD]

训练

训练工夫应该少于 30 分钟,具体取决于所应用的 GPU。

trainer.train()
训练损失 训练步数 验证损失 Wer
4.905 100 0.215 0.280
0.290 200 0.167 0.232
0.2659 300 0.161 0.229
0.2398 400 0.156 0.223

训练损失和验证 WER 都很好地降落。

咱们看到,仅微调 mms-1b-all 的适配器层 100 步就大大超过了 这里 显示的微调整个 xls-r-300m 检查点。

从 官网论文 和这个疾速比拟中能够分明地看出,mms-1b-all 具备更高的将常识转移到低资源语言的能力,应该优先于 xls-r-300m。此外,训练也更节俭内存,因为只训练了一小部分层。

适配器权重将作为模型检查点的一部分上传,但咱们也心愿确保独自保留它们,以便它们能够轻松地上下线。

让咱们将所有适配器层保留到训练输入目录中,以便它可能正确上传到 Hub。

from safetensors.torch import save_file as safe_save_file
from transformers.models.wav2vec2.modeling_wav2vec2 import WAV2VEC2_ADAPTER_SAFE_FILE
import os

adapter_file = WAV2VEC2_ADAPTER_SAFE_FILE.format(target_lang)
adapter_file = os.path.join(training_args.output_dir, adapter_file)

safe_save_file(model._get_adapters(), adapter_file, metadata={"format": "pt"})

最初,你能够将训练后果上传到🤗 Hub。

trainer.push_to_hub()

适配器权重训练的次要长处之一是“根底”模型 (约占模型权重的 99%) 放弃不变,只需共享一个小的 2.5M 适配器检查点 即可应用训练好的检查点。

这使得训练额定的适配器层并将它们增加到你的仓库变得非常简单。

你能够通过简略地从新运行此脚本并将你想要训练的语言更改为另一种语言来轻松实现,例如 swe 示意瑞典语。此外,你应该确保词汇表不会被齐全笼罩,而是新语言词汇表应该像下面正文掉的单元格中所述那样 附加 到现有词汇表中。

为了演示如何加载不同的适配器层,我还训练并上传了一个瑞典语适配器层,其 iso 语言代码为 swe,如 此处 所示

你能够像平常一样应用 from_pretrained(...) 加载微调后的检查点,但应确保在办法中增加 target_lang="<your-lang-code>",以便加载正确的适配器。你还应该为分词器正确设置目标语言。

让咱们看看如何首先加载土耳其检查点。

model_id = "patrickvonplaten/wav2vec2-large-mms-1b-turkish-colab"

model = Wav2Vec2ForCTC.from_pretrained(model_id, target_lang="tur").to("cuda")
processor = Wav2Vec2Processor.from_pretrained(model_id)

processor.tokenizer.set_target_lang("tur")

让咱们查看模型是否能够正确转录土耳其语

from datasets import Audio

common_voice_test_tr = load_dataset("mozilla-foundation/common_voice_6_1", "tr", data_dir="./cv-corpus-6.1-2020-12-11", split="test", use_auth_token=True)
common_voice_test_tr = common_voice_test_tr.cast_column("audio", Audio(sampling_rate=16_000))

让咱们解决音频,运行前向传递并预测 ids

input_dict = processor(common_voice_test_tr[0]["audio"]["array"], sampling_rate=16_000, return_tensors="pt", padding=True)

logits = model(input_dict.input_values.to("cuda")).logits

pred_ids = torch.argmax(logits, dim=-1)[0]

最初,咱们能够解码该示例。

print("Prediction:")
print(processor.decode(pred_ids))

print("\nReference:")
print(common_voice_test_tr[0]["sentence"].lower())

输入:

    Prediction:
    pekçoğuda roman toplumundan geliyor

    Reference:
    pek çoğu da roman toplumundan geliyor.

这看起来简直完全正确,只是第一个单词中应该增加两个空格。
当初,通过调用 model.load_adapter(...) 并将分词器更改为瑞典语,能够非常简单地将适配器更改为瑞典语。

model.load_adapter("swe")
processor.tokenizer.set_target_lang("swe")

咱们再次从一般语音加载瑞典语测试集

common_voice_test_swe = load_dataset("mozilla-foundation/common_voice_6_1", "sv-SE", data_dir="./cv-corpus-6.1-2020-12-11", split="test", use_auth_token=True)
common_voice_test_swe = common_voice_test_swe.cast_column("audio", Audio(sampling_rate=16_000))

并转录一个样本:

input_dict = processor(common_voice_test_swe[0]["audio"]["array"], sampling_rate=16_000, return_tensors="pt", padding=True)

logits = model(input_dict.input_values.to("cuda")).logits

pred_ids = torch.argmax(logits, dim=-1)[0]

print("Prediction:")
print(processor.decode(pred_ids))

print("\nReference:")
print(common_voice_test_swe[0]["sentence"].lower())

输入:

    Prediction:
    jag lämnade grovjobbet åt honom

    Reference:
    jag lämnade grovjobbet åt honom.

太好了,这看起来像是一个完满的转录!

咱们在这篇博客文章中展现了 MMS 适配器权重微调不仅在低资源语言上提供了最先进的性能,而且还显著缩短了训练工夫,并容许轻松构建定制的适配器权重汇合。

相干帖子和附加链接列在这里:

  • 官网论文
  • 原始 cobebase
  • 官网演示
  • Transformers 文档
  • 相干 XLS-R 博客文章
  • Hub 上的模型

英文链接: https://hf.co/blog/mms_adapters

作者: Patrick von Platen

译者: innovation64

审校 / 排版: zhongdongy (阿东)

退出移动版