Fast-Bert

新功能 - 文本分类训练的学习率查找器（感谢借鉴自https://github.com/davidtvs/pytorch-lr-finder）

支持LAMB优化器以加快训练速度。 请参阅https://arxiv.org/abs/1904.00962了解LAMB优化器的相关论文。

支持BERT和XLNet进行多类和多标签文本分类。

Fast-Bert是一个深度学习库，允许开发人员和数据科学家训练和部署基于BERT和XLNet的模型，用于自然语言处理任务，从文本分类开始。

Fast-Bert的工作建立在优秀的Hugging Face BERT PyTorch库提供的坚实基础之上，并受到fast.ai的启发，致力于使最先进的深度学习技术为广大机器学习从业者所用。

使用Fast-Bert，您将能够：

1. 在自定义数据集上训练（更准确地说是微调）BERT、RoBERTa和XLNet文本分类模型。

2. 调整模型超参数，如轮数、学习率、批量大小、优化器调度等。

3. 保存和部署训练好的模型进行推理（包括在AWS Sagemaker上）。

Fast-Bert将支持以下多类和多标签文本分类，并且将来还会支持其他NLU任务，如命名实体识别、问答和自定义语料库微调。

1. BERT（来自Google），随论文BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding发布，作者为Jacob Devlin、Ming-Wei Chang、Kenton Lee和Kristina Toutanova。

2. XLNet（来自Google/CMU），随论文​XLNet: Generalized Autoregressive Pretraining for Language Understanding发布，作者为Zhilin Yang*、Zihang Dai*、Yiming Yang、Jaime Carbonell、Ruslan Salakhutdinov、Quoc V. Le。

3. RoBERTa（来自Facebook），一种经过稳健优化的BERT预训练方法，作者为Yinhan Liu、Myle Ott、Naman Goyal、Jingfei Du等。

4. DistilBERT（来自HuggingFace），随博客文章Smaller, faster, cheaper, lighter: Introducing DistilBERT, a distilled version of BERT发布，作者为Victor Sanh、Lysandre Debut和Thomas Wolf。

安装

本仓库在Python 3.6+上测试通过。

使用pip安装

可以通过pip安装PyTorch-Transformers，如下所示：

pip install fast-bert

从源代码安装

克隆仓库并运行：

pip install [--editable] .

或

pip install git+https://github.com/kaushaltrivedi/fast-bert.git

您还需要安装NVIDIA Apex。

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

使用方法

文本分类

1. 创建DataBunch对象

DataBunch对象接收训练、验证和测试CSV文件，并将数据转换为BERT、RoBERTa、DistilBERT或XLNet的内部表示。该对象还会根据设备配置、批量大小和最大序列长度实例化正确的数据加载器。

from fast_bert.data_cls import BertDataBunch

databunch = BertDataBunch(DATA_PATH, LABEL_PATH,
                          tokenizer='bert-base-uncased',
                          train_file='train.csv',
                          val_file='val.csv',
                          label_file='labels.csv',
                          text_col='text',
                          label_col='label',
                          batch_size_per_gpu=16,
                          max_seq_length=512,
                          multi_gpu=True,
                          multi_label=False,
                          model_type='bert')

train.csv和val.csv的文件格式

如果列名与通常的text和labels不同,你需要在databunch的text_col和label_col参数中提供这些名称。

labels.csv将包含所有唯一标签的列表。在这种情况下,文件将包含:

正面
负面

对于多标签分类,labels.csv将包含所有可能的标签:

有毒
严重有毒
淫秽
威胁
侮辱
仇恨身份

文件train.csv将为每个标签包含一列,每列的值为0或1。别忘了在BertDataBunch中将multi_label=True用于多标签分类。

label_col将是标签列名的列表。在这种情况下,它将是:

['有毒','严重有毒','淫秽','威胁','侮辱','仇恨身份']

分词器

你可以创建一个分词器对象并将其传递给DataBunch,或者将模型名称作为分词器传递,DataBunch将自动下载并实例化适当的分词器对象。

例如,要使用XLNet base cased模型,将分词器参数设置为'xlnet-base-cased'。DataBunch将自动下载并实例化XLNetTokenizer,其词汇表用于xlnet-base-cased模型。

模型类型

Fast-Bert支持基于XLNet、RoBERTa和BERT的分类模型。将模型类型参数值设置为**'bert'、roberta或'xlnet'**,以初始化适当的databunch对象。

2. 创建学习器对象

BertLearner是将所有内容整合在一起的"学习器"对象。它封装了模型生命周期的关键逻辑,如训练、验证和推理。

学习器对象将使用前面创建的databunch作为输入,同时还有一些其他参数,如预训练模型的位置、FP16训练、多GPU和多标签选项。

学习器类包含训练循环、验证循环、优化器策略和关键指标计算的逻辑。这有助于开发人员专注于自定义用例,而不必担心这些重复性活动。

同时,学习器对象具有足够的灵活性,可以通过使用灵活的参数或通过创建BertLearner的子类并重新定义相关方法来进行自定义。

from fast_bert.learner_cls import BertLearner
from fast_bert.metrics import accuracy
import logging

logger = logging.getLogger()
device_cuda = torch.device("cuda")
metrics = [{'name': 'accuracy', 'function': accuracy}]

learner = BertLearner.from_pretrained_model(
                        databunch,
                        pretrained_path='bert-base-uncased',
                        metrics=metrics,
                        device=device_cuda,
                        logger=logger,
                        output_dir=OUTPUT_DIR,
                        finetuned_wgts_path=None,
                        warmup_steps=500,
                        multi_gpu=True,
                        is_fp16=True,
                        multi_label=False,
                        logging_steps=50)

3. 找到最佳学习率

学习率是模型训练最重要的超参数之一。我们已经整合了Leslie Smith提出并后来被fastai库采用的学习率查找器。

learner.lr_find(start_lr=1e-5,optimizer_type='lamb')

这段代码大量借鉴自David Silva的pytorch-lr-finder库。

4. 训练模型

learner.fit(epochs=6,
			lr=6e-5,
			validate=True, 	# 每个epoch后评估模型
			schedule_type="warmup_cosine",
			optimizer_type="lamb")

Fast-Bert现在支持LAMB优化器。由于训练速度的原因，我们将LAMB设置为默认优化器。您可以通过将optimizer_type设置为'adamw'来切换回AdamW。

5. 保存训练好的模型产物

learner.save_model()

模型产物将被保存在提供给learner对象的output_dir/'model_out'路径中。以下文件将被保存：

由于所有模型产物都存储在同一文件夹中，您可以通过将pretrained_path指向此位置来实例化learner对象以运行推理。

6. 模型推理

如果您已经有一个带有训练好模型的Learner对象实例，只需在learner对象上调用predict_batch方法，传入文本数据列表：

texts = ['我真的很喜欢Netflix的原创电影',
		 '这部电影不值得一看']
predictions = learner.predict_batch(texts)

如果您有持久化的训练模型，只想在该训练模型上运行推理逻辑，请使用第二种方法，即predictor对象。

from fast_bert.prediction import BertClassificationPredictor

MODEL_PATH = OUTPUT_DIR/'model_out'

predictor = BertClassificationPredictor(
				model_path=MODEL_PATH,
				label_path=LABEL_PATH, # labels.csv文件的位置
				multi_label=False,
				model_type='xlnet',
				do_lower_case=False,
				device=None) # 设置自定义torch.device，如果可用则默认为cuda

# 单个预测
single_prediction = predictor.predict("只需为这段文本给我结果")

# 批量预测
texts = [
	"这是第一段文本",
	"这是第二段文本"
	]

multiple_predictions = predictor.predict_batch(texts)

语言模型微调

在自定义数据集上使用基于BERT的模型的一个有用方法是首先对自定义数据集微调语言模型任务，这是fast.ai的ULMFit所采用的方法。这个想法是从一个预训练模型开始，然后在自定义数据集的原始文本上进一步训练模型。我们将使用掩码语言模型任务来微调语言模型。

本节将描述如何使用FastBert来微调语言模型。

1. 导入必要的库

必要的对象存储在带有'_lm'后缀的文件中。

# 语言模型Databunch
from fast_bert.data_lm import BertLMDataBunch
# 语言模型learner
from fast_bert.learner_lm import BertLMLearner

from pathlib import Path
from box import Box

2. 定义参数并设置数据路径

# Box是一个很好的包装器，用于从JSON字典创建对象
args = Box({
    "seed": 42,
    "task_name": 'imdb_reviews_lm',
    "model_name": 'roberta-base',
    "model_type": 'roberta',
    "train_batch_size": 16,
    "learning_rate": 4e-5,
    "num_train_epochs": 20,
    "fp16": True,
    "fp16_opt_level": "O2",
    "warmup_steps": 1000,
    "logging_steps": 0,
    "max_seq_length": 512,
    "multi_gpu": True if torch.cuda.device_count() > 1 else False
})

DATA_PATH = Path('../lm_data/')
LOG_PATH = Path('../logs')
MODEL_PATH = Path('../lm_model_{}/'.format(args.model_type))

DATA_PATH.mkdir(exist_ok=True)
MODEL_PATH.mkdir(exist_ok=True)
LOG_PATH.mkdir(exist_ok=True)

3. 创建DataBunch对象

BertLMDataBunch类包含一个静态方法'from_raw_corpus'，该方法将接收原始文本列表并为语言模型学习器创建DataBunch。

该方法首先会通过删除HTML标签、多余空格等来预处理文本列表，然后创建lm_train.txt和lm_val.txt文件。这些文件将用于训练和评估语言模型微调任务。

下一步是对文本进行特征化处理。文本将被分词、数字化，并分割成512个token的块（包括特殊token）。

databunch_lm = BertLMDataBunch.from_raw_corpus(
					data_dir=DATA_PATH,
					text_list=texts,
					tokenizer=args.model_name,
					batch_size_per_gpu=args.train_batch_size,
					max_seq_length=args.max_seq_length,
                    multi_gpu=args.multi_gpu,
                    model_type=args.model_type,
                    logger=logger)

由于这一步可能会根据自定义数据集文本的大小而耗费一些时间，特征化后的数据将被缓存在data_dir/lm_cache文件夹中的pickle文件里。

下次，您可能想直接实例化DataBunch对象，而不是使用from_raw_corpus方法，如下所示：

databunch_lm = BertLMDataBunch(
						data_dir=DATA_PATH,
						tokenizer=args.model_name,
                        batch_size_per_gpu=args.train_batch_size,
                        max_seq_length=args.max_seq_length,
                        multi_gpu=args.multi_gpu,
                        model_type=args.model_type,
                        logger=logger)

4. 创建LM Learner对象

BertLearner是将所有内容整合在一起的"学习器"对象。它封装了模型生命周期的关键逻辑，如训练、验证和推理。

学习器对象将接收之前创建的databunch作为输入，以及其他一些参数，如预训练模型的位置、FP16训练、多GPU和多标签选项。

学习器类包含训练循环、验证循环和优化器策略的逻辑。这有助于开发人员专注于他们的自定义用例，而不必担心这些重复性活动。

同时，学习器对象足够灵活，可以通过使用灵活的参数或通过创建BertLearner的子类并重新定义相关方法来进行自定义。

learner = BertLMLearner.from_pretrained_model(
							dataBunch=databunch_lm,
							pretrained_path=args.model_name,
							output_dir=MODEL_PATH,
							metrics=[],
							device=device,
							logger=logger,
							multi_gpu=args.multi_gpu,
							logging_steps=args.logging_steps,
							fp16_opt_level=args.fp16_opt_level)

5. 训练模型

learner.fit(epochs=6,
			lr=6e-5,
			validate=True, 	# 在每个epoch后评估模型
			schedule_type="warmup_cosine",
			optimizer_type="lamb")

Fast-Bert现在支持LAMB优化器。由于训练速度的原因，我们将LAMB设置为默认优化器。您可以通过将optimizer_type设置为'adamw'来切换回AdamW。

6. 保存训练好的模型

learner.save_model()

模型文件将被保存在提供给学习器对象的output_dir/'model_out'路径中。以下文件将被保存：

pytorch_model.bin包含了微调后的权重，您可以通过finetuned_wgts_path参数将分类任务学习器对象指向这个文件。

Amazon Sagemaker支持

这个库的目的是让您能够训练和部署生产级别的模型。由于transformer模型需要昂贵的GPU来训练，我已经添加了在AWS SageMaker上训练和部署模型的支持。

该存储库包含了在Amazon SageMaker中构建基于BERT的分类模型的Docker镜像和代码。 请参考我的博客使用FastBert和Amazon SageMaker训练和部署强大的基于BERT的NLP模型，该博客详细解释了如何将SageMaker与FastBert结合使用。

引用

如果您在已发表或开源项目中使用本工作，请提及此库和HuggingFace的pytorch-transformers库，并附上本仓库的链接。

同时还请引用我关于这个主题的博客文章：

• FastBert简介 — 一个简单的BERT模型深度学习库

• 使用BERT进行多标签文本分类 — 强大的Transformer

• 使用FastBert和Amazon SageMaker训练和部署强大的基于BERT的NLP模型