新知一下
海量新知
6 0 8 8 0 3 3

解密 BERT

Python和AI派 | Python,人工智能 2019/10/30 08:25

来源:AI开发者

BERT简介

毫不夸张的讲, 谷歌AI实验室的BERT深刻影响了NLP的格局

想象一下这样一个在大量未标注数据集中训练的模型,你仅仅只需要做一点的微调,就可以在11个不同的NLP任务上取得 SOTA结果。 没错,BERT就是这样,它彻底改变了我们设计NLP模型的方式。

BERT之后,许多NLP架构、训练方法与语言模型如雨后春笋般涌现,比如谷歌的TransformerXL、OpenAI’s GPT-2、 XLNet、ERNIE2.0、 RoBERTa等。

注: 在这篇文章中,我会提及许多Transformer的内容,如果你对Transformer不了解的话,可以先看看这篇文章—— How do Transformers Work in NLP? A Guide to the Latest State-of-the-Art Models .

什么是BERT?

你一定听说过BERT,也知道了它是多么不可思议以致改变了NLP的领域格局,但BERT究竟是什么?

以下是BERT团队对该框架的描述:

BERT全称Bidirectional Encoder Representations from Transformers(Transformers的双向编码表示),对未标注的文本,通过上下文约束预训练深层双向表示。 训练完成后,只需要对BERT预训练模型进行fine-tune,再加上针对特定任务的输出层就可以取得SOTA结果。

对新人来说这样的解释不够明白,但这确实很好的总结了BERT的机制。 接下来,我们一点点的进行剖析。

首先可以明确的是,BERT全称Bidirectional Encoder Representations from Transformers,名字中的每一个单词都有其意义,我们会在后面的段落一一介绍。 从BERT的名字中,我们能得到最重要信息就是: BERT是基于Transformer架构的。

其次,BERT是在大量的未标注文本上预训练得到,包括整个Wikipedia(有25亿单词)和图书语料库(8亿单词)。

预训练这一步对BERT来讲是至关重要的。 正是由于如此庞大的语料库的支撑,模型在训练过程中才能对语言的工作原理进行更深入、更准确的提取,通过此过程提取到的知识对所有NLP任务来说都是‘万滑油’。

然后,BERT是“深度双向”模型,双向就意味着BERT在训练过程中关注当前位置的上下文信息。

上下文信息对准确理解语义很重要的。 看下面这个例子,两句话中都包含了同一个单词“bank”:

新知达人, 解密 BERT

BETR捕获上下文信息

如果我们想仅依靠上文或者下文的信息去理解“bank”的含义,那么对这两句话中的“bank”,我们是无法区分它们的不同含义的。

解决方法就是在预测之前同时考虑上下文信息,BERT就是这样做的。

最后,BERT最吸引人的在于,我们仅仅通过在模型后根据自己的需求加上输出层部分就可以在各类NLP任务取得SOTA结果。

从Word2Vec到BERT: NLP对语言表示的探索

“自然语言处理领域最大的挑战之一就是训练数据的短缺。 NLP是一个多元领域,任务繁多,大多数特定领域的数据集仅仅包含几千或几十万人工标注的数据。 ”——谷歌AI

Word2Vec和GloVe

预训练模型从大量未标注文本数据中学习语言表示的思想来源于词嵌入,如 Word2Vec and GloVe

词嵌入改变了进行NLP任务的方式。 通过嵌入,我们能够捕获单词的上下文关系。

新知达人, 解密 BERT

图中所示的这些嵌入方法被广泛用于下游NLP任务的训练模型,以便获取较好的预测结果。

之前的嵌入方法的一大缺陷在于只使用了很浅的语言模型,那就意味着它们捕获到的信息是有限的。

另外一个缺陷就是这些嵌入模型没有考虑单词的上下文。 就像之前提到的“bank”例子,在不同的语境下同一个单词可能会有不同的含义。

然而,WordVec之类的模型将不同语境中的“bank”以同样的向量表示。

于是,一些重要的信息被遗漏了。

ELMo与ULMFiT

新知达人, 解密 BERT

ELMo 是对语言多义性问题提出的解决方案——针对那些在不同上下文中具有不同含义的单词。

从训练浅层前馈网络(Word2vec)开始,我们逐步过渡到使用复杂的双向LSTM结构来训练词嵌入。

这意味着同一单词根据其所在的上下文可以具有多个ELMO嵌入。

从那时起,我们开始注意到预训练的优势将使其在NLP任务中发挥重要作用。

新知达人, 解密 BERT

ULMFiT 更进一步,在文档分类任务中,即使只有很少的数据(少于100),对该框架训练的语言模型进行微调就能够提供出色的结果。 这意味着ULMFiT解决了NLP任务中的 迁移学习 问题。

这是我们提出的NLP迁移学习黄金公式:

NLP迁移学习 = 预训练 + 微调

在ULMFIT之后,许多NLP任务根据上述公式进行训练,并获得了新的基准。

OpenAI的GPT

OpenAI’s GPT 进一步扩展了ULMFiT和ELMo中引入的pre-training和fine-tuning方法。

GPT的关键是用基于Transformer的结构取代了基于LSTM的语言建模结构。

不仅是文档分类任务,GPT模型还可以对其他NLP任务进行

fine-tuned,例如常识推理,语义相似性和阅读理解。

OpenAI的GPT在多项任务获得SOTA结果,验证了Transformer架构的鲁棒性和有效性。

就这样,BERT在Transformer的基础上横空出世,并给NLP领域带来巨大变革。

BERT出世

至此,解决NLP任务离不开这两步:

1. 在未标注的大型文本语料库上训练语言模型(无监督或半监督)

2. 针对特定的NLP任务对大型语言模型进行微调,以充分利用预训练模型的大量知识(监督)

接下来,我们将详细了解BERT如何训练模型,并在未来一段时间内成为NLP领域的行业标杆。

BERT是如何工作的? 干货讲解

深入BERT,理解为什么BERT建立的语言模型如此有效。

1. BERT的结构

BERT架构建立在Transformer之上。 我们目前有两个可用的变体:

  • BERT Base: 12层(transformer模块),12层注意力,1.1亿参数

  • BERT Large: 24层(transformer模块),16层注意力,3.4亿参数

新知达人, 解密 BERT

图源

与OpenAI的GPT模型相比,BERT Base模型大小与其相似,同时BERT Base的所有transformer层都仅包括编码部分。

如果你对transformer结构了解不是很清楚,建议你先读一下 这篇文章

现在我们已经了解了BERT的整体架构。 在正式构建模型之前,需要先进行一些文本处理工作。

新知达人, 解密 BERT

2. 文本预处理

BERT背后的开发人员添加了一组特定的规则来表示模型的输入文本。 其中许多都是创造性的设计选择可以让模型表现更好。

首先,每个输入嵌入都是三个嵌入的组合:

1.位置嵌入: BERT学习并使用位置嵌入来表达单词在句子中的位置。 添加该嵌入是为了克服Transformer的局限性,与RNN不同,Transformer无法捕获“序列”或“顺序”信息

2.段嵌入: BERT也可以将句子作为任务的输入(问题-解答)。 因此,它为第一句话和第二句话学习了独特的嵌入,以帮助模型区分它们。 在上面的示例中,所有为EA的标记都属于句子A(对于EB一样)

3.令牌嵌入: 这些是从WordPiece令牌词汇表中为特定令牌学习的嵌入

对于给定的令牌,其输入表示形式是通过将相应的令牌,段和位置嵌入相加而构造的。

这种全面的嵌入方案包含许多有用的模型信息。

这些预处理步骤的组合使BERT如此通用。

3. 预训练任务

BERT对两项NLP任务进行预训练:

  • 遮掩语言模型

  • 下句预测

让我们更详细地理解这两个任务!

a. 遮掩语言模型(双向)

BERT是深层的双向模型,该网络从第一层到最后一层始终关注当前单词的上下文进行信息捕获。

新知达人, 解密 BERT

单词序列预测

传统的语言模型要么是利用从右到左的文本信息进行训练预测下一个单词(例如GPT),要么是使用从左到右的文本信息进行训练,这使模型不可避免的丢失一些信息,从而导致错误。

ELMo试图通过训练两个LSTM语言模型(一个利用从左到右的文本信息,一个利用从右到左的文本信息),并将它们进行连接来解决这个问题。 这样虽然在一定程度上取得进步,但还远远不够。

新知达人, 解密 BERT

相对于GPT与ELMo,BERT在利用上下文信息这一方面取得重要突破,如上图所示。

图中的箭头表示一层到下一层的信息流,顶部的绿色框表示每个输入单词的最终表示。

从以上图片可以明显看到: BERT是双向的,GPT是单向的(从左到右的信息流),ELMo是浅层双向的。

关于遮掩语言模型——这是BERT双向编码的奥秘。

对这样一个句子——“I love to read data science blogs on Analytics Vidhya”,我们怎样用它训练双向语言模型呢。

我们首先将“Analytics”替换为“[MASK]”, “[MASK]”  表示将该位置的单词掩盖。

然后我们需要训练一个模型,使其能够预测mask掉的单词: “I love to read data science blogs on [MASK] Vidhya.”

这就是遮掩语言模型的关键。 BERT的作者还介绍了一些遮掩语言模型的注意事项:

  • 为了防止模型过于关注特定位置或被遮掩的标记,研究人员随机遮掩15%的单词

  • 被遮掩的单词并不总是[MASK]取代,在针对特定任务的微调阶段是不需要[MASK]标记的

  • 为此,研究人员的一般做法是: (对 15%需要[MASK] 单词 )

  • (15%的)80%的单词被[MASK]遮掩

  • 其余10%的单词被其他随机单词取代

  • 其余10%的单词保持不变

在我之前的一篇文章中,我详细介绍了如何在Python中实现遮掩语言模型: Introduction to PyTorch-Transformers: An Incredible Library for State-of-the-Art NLP (with Python code)

b. 下句预测

遮掩语言模型(MLMs)学习单词之间的关系。

此外,BERT还对下句预测任务进行训练以学习句子之间的关系。

这类任务的典型例子就是 问答系统

任务很简单,给A和B两个句子,判断B是A之后的下一句,或只是一个随机句子?

由于这是一个二分类问题,将语料库中的句子分解为句子对就可以得到大量训练数据。 与MLMs类似,作者也给出在进行下句预测任务时的注意事项。 具体通过这个例子进行说明:

对于一个包含10万句子的数据集,我们可以得到5万句子对作训练数据。

  • 训练数据中的50%,第二句是真实的下句

  • 另外的50%,第二句是语料库中的随机句子

  • 前50%的标签是‘IsNext’,后50%的标签是‘NotNext’

在建模过程中结合遮掩语言模型(MLMs)和下句预测(NSP)两个预训练任务,这就使得BERT成为一个与任务无关的模型,经过简单fine-tuning即可适用到其他下游任务。

在python中使用BERT进行文本分类

你对BERT的可能性一定有各种期待。 确实如此,我们在具体的NLP应用中可以通过各种方式利用BERT预训练模型的优势。

最有效的方法之一就是根据自己的任务和特定数据进行微调, 然后,我们可以将BERT中的嵌入用作文本文档的嵌入。

接下来,我们将学习如何将BERT的嵌入应用到自己的任务上。 至于如何对整个BERT模型进行微调,我会在另一篇文章中进行介绍。

为了提取BERT的嵌入,我们将使用一个非常实用的开源项目 Bert-as-Service

新知达人, 解密 BERT

BERT-As-Service

由于BERT需要大量代码以及许多安装包的支持,对普通用户而言直接运行BERT是不现实的,为此开源项目BERT-As-Service来帮助我们便捷的使用BERT。 通过该项目,我们仅仅通过两行代码就可以调用BRRT对句子进行编码。

安装BERT-As-Service

BERT-As-Service运行方式十分简单。 它创建了一个BERT服务器,我们可以在notebook中编写ython代码进行访问。 通过该方式,我们只需将句子以列表形式发送,服务器就会返回这些句子的BERT嵌入。

我们可以通过pip安装服务器和客户端。 它们可以单独安装在本地计算机,也可以安装到不同的计算机上:

$ pip install bert-serving-server # server

$ pip install bert-serving-client # client

另外,由于运行BERT对GPU要求较高,我建议你在云GPU平台或是其他具有高计算能力的计算机上安装bert-serving-server。

同时,bert-serving-server对python和Tensorflow的版本要求为: Python >= 3.5 ; TensorFlow >= 1.10。

然后,在终端下载图示的预训练模型(选择你需要的即可),并对下载的zip文件进行解压。

下图是已发布的BERT预训练模型:

新知达人, 解密 BERT

BERT模型

我这里选择BERT Uncased下载并解压:

$ wget https://storage.googleapis.com/bert_models/2018_10_18/uncased_L-12_H-768_A-12.zip && unzip uncased_L-12_H-768_A-12.zip

将所有文件解压之后,就可以启动BERT服务了:

$ bert-serving-start -model_dir uncased_L-12_H-768_A-12/ -num_worker=2 -max_seq_len 50

现在,你可以在Python中通过bert-serving-client调用BERT-As-Service。 看代码吧!

打开一个新的Jupyter Notebook,我们想要获取“I love data science and analytics vidhya”的BERT嵌入。

from bert_serving.client import BertClient

# 使用IP地址连接BERT服务器; 如果是本机服务器的话不需要IP

bc = BertClient(ip="服务器的IP地址")

# 获取嵌入

embedding = bc.encode(["I love data science and analytics vidhya."])

# 对返回的嵌入形状进行确认,应该是1x768

print(embedding.shape)

IP地址是BERT服务器或云平台的IP; 如果是本机服务器的话不需要填写IP

由于该句被BERT架构中的768个隐藏单元表示,最终返回的嵌入形状是(1,768)。

问题描述: 对Twitter上仇恨言论进行分类

接下来使用真实数据集测试BERT的效果。 我们将使用Twitter的“仇恨言论”分类数据集,该数据集中的推文被标注为是或者否。

你可以从此链接 problem statement on the DataHack platform 了解或下载该数据集。

为简单起见,如果一条推文带有种族主义或性别歧视情绪,我们就认为该推文包含仇恨言论。 于是,本次任务就是将种族主义或性别歧视推文与其他推文进行分类。

我们将使用BERT对数据集中的每条推文进行嵌入,然后使用这些嵌入训练文本分类模型。

新知达人, 解密 BERT

任务流程

接下来看代码部分:

import pandas as pd

import numpy as np

# 导入训练数据

train = pd.read_csv('BERT_proj/train_E6oV3lV.csv', encoding='iso-8859-1')

train.shape

相信你对推特并不陌生,在很多推文中总有一些随机符号和数字(又名聊天语言! )。 我们的数据集也是这样,为此,需要对数据集进行预处理,然后再传入BERT:

新知达人, 解密 BERT

现在,我们需要将清理后的数据集划分为训练集与验证集:

from sklearn.model_selection import train_test_split

# 划分训练集与验证集

X_tr, X_val, y_tr, y_val = train_test_split(train.clean_text, train.label, test_size=0.25, random_state=42)

print('X_tr shape:',X_tr.shape)

接下来,对测试集与验证集的所有推文进行BERT嵌入:

from bert_serving.client import BertClient

# 使用IP连接BERT服务器

bc = BertClient(ip="YOUR_SERVER_IP")

# 获得训练集与测试集的嵌入

X_tr_bert = bc.encode(X_tr.tolist())

X_val_bert = bc.encode(X_val.tolist())

该构建模型了!我们先来训练分类模型:

from sklearn.linear_model import LogisticRegression

# LR模型

model_bert = LogisticRegression()

# 训练

model_bert = model_bert.fit(X_tr_bert, y_tr)

# 预测

pred_bert = model_bert.predict(X_val_bert)

查看分类准确率:

from sklearn.metrics import accuracy_score

print(accuracy_score(y_val, pred_bert))

可以看到,即使只有很小的数据集,我们也很容易达到95%左右的准确率。实在是不可思议!

你最好在其他任务上亲自实践一下BERT嵌入,并将你的结果分享到下面的评论区。

下一篇文章,我会在另外一个数据集上使用Fine-tune的BERT模型,并比较其性能。

超越BERT: NLP的最新技术

BERT激起了人们对NLP领域的极大兴趣,尤其是Transformer的广泛应用。 这也导致越来越多的实验室和组织开始研究pre-training, transformers 和 fine-tuning等任务。

BERT之后,一些新的项目在NLP各项任务中取得了更好的结果。 比如RoBERTa,这是Facebook AI对BERT和DistilBERT的改进,而后者其实就是BERT的更轻巧,便捷的版本。

你可以在 regarding State-of-the-Art NLP in this article 了解更多BERT之后的改进模型。

via: https://www.analyticsvidhya.com ,2019/9/25.

更多“BERT”相关内容

更多“BERT”相关内容

新知精选

更多新知精选