NLP新秀 : BERT的优雅解读

语言模型有什么作用？通过语言模型，可以量化地衡量一段文本存在的可能性。对于一段长度为n的文本，文本里每个单词都有上文预测该单词的过程，所有单词的概率乘积便可以用来评估文本。在实践中，如果文本很长，P(wi|context(wi))的估算会很困难，因此有了简化版：N元模型。在N元模型中，通过对当前词的前N个词进行计算来估算该词的条件概率。对于N元模型，常用的有unigram、bigram和trigram，N越大，越容易出现数据稀疏问题，估算结果越不准。此外，N元模型没法解决一词多义和一义多词问题。

为了解决N元模型估算概率时的数据稀疏问题，研究者提出了神经网络语言模型，代表作有2003年Bengio等提出了的NNLM，但效果并不吸引人，足足沉寂了十年。在另一计算机科学领域机器视觉，深度学习混得风生水起，特别值得一提的是预训练处理，典型代表：基于ImageNet预训练的Fine-Tuning模型。图像领域的预处理跟现在NLP领域的预训练处理思路相似，基于大规模图像训练数据集，利用神经网络预先训练，将训练好的网络参数保存。当有新的任务时，采用相同的网络结构，加载预训练的网络参数初始化，基于新任务的数据训练模型，Frozen或者Fine-Tuning。Frozen指底层加载的预训练网络参数在新任务训练过程中不变，Fine-Tuning指底层加载的预训练网络参数会随着新任务训练过程不断调整以适应当前任务。深度学习是适用于大规模数据，数据量少训练出来的神经网络模型效果并没有那么好。所以，预训练带来的好处非常明显，新任务即使训练数据集很小，基于预训练结果，也能训练出不错的效果。

深度学习预训练在图像领域的成果，吸引研究者探索预训练在NLP领域的应用，譬如Word Embedding。2013年开始大火的Word Embedding工具Word2Vec，Glove跟随其后。Word2Vec有2种训练方式：CBOW和 Skip-gram 。CBOW指抠掉一个词，通过上下文预测该词； Skip-gram 则与CBOW相反，通过一个词预测其上下文。不得不说，Word2Vec的CBOW训练方式，跟BERT“完形填空”的学习思路有异曲同工之妙。

一个单词通过Word Embedding表示，很容易找到语义相近的单词，但单一词向量表示，不可避免一词多义问题。于是有了基于上下文表示的 ELMo 和 OpenAI  GPT。

ELMo ，Embedding from Language Models，基于上下文对Word Embedding动态调整的双向神经网络语言模型。 ELMo 采用的是一种“Feature-based Approaches”的预训练模式，分两个阶段：第一阶段采用双层双向的LSTM模型进行预训练；第二阶段处理下游任务时，预训练网络中提取出的Word Embedding作为新特征添加到下游任务中，通过双层双向LSTM模型补充语法语义特征。相比Word2Vec， ELMo 很好地解决了一词多义问题，在6个NLP测试任务中取得SOTA。

Transformer：谷歌提出的新网络结构，这里指Encoder特征提取器。LSTM提取长距离特征有长度限制，而Transformer基于self-Attention机制，任意单元都会交互，没有长度限制问题，能够更好捕获长距离特征。

GPT，Generative Pre-Training，OpenAI提出的基于生成式预训练的单向神经网络语言模型。GPT采用的是一种“Fine-Tuning Approaches”的预训练模式，同样分两个阶段： 第一阶段采用Transformer模型通过上文预测的方式进行预训练；第二阶段采用Fine-Tuning的模式应用到下游任务。GPT的效果同样不错，在9个NLP测试任务中取得SOTA。不过，GPT这种单向训练模式，会丢失下文很多信息，在阅读理解这类任务场景就没有双向训练模式那么优秀。

BERT，Bidirectional Encoder Representations from Transformers，基于Transformer的双向语言模型。同样，BERT采用跟GPT一样的“Fine-Tuning Approaches”预训练模式，分两个阶段：第一阶段采用双层双向Transformer模型通过MLM和NSP两种策略进行预训练；第二阶段采用Fine-Tuning的模式应用到下游任务。有人戏称：Word2Vec + ELMo + GPT =BERT，不过也并无道理，BERT吸收了这些模型的优点：“完形填空”的学习模式迫使模型更多依赖上下文信息预测单词，赋予了模型一定的纠错能力；Transformer模型相比LSTM模型没有长度限制问题，具备更好的能力捕获上下文信息特征；相比单向训练模式，双向训练模型捕获上下文信息会更加全面；等等。当然，效果才是王道，集大成者BERT拿了11项SOTA。

论文解读：BERT原理

相关论文：

2017年，谷歌发表《Attention Is All You Need》，提出Transformer模型；

2018年，谷歌发表《BERT：Pre-training of Deep Bidirectional Transformers for Language Understanding》，提出基于Transformer的语言模型BERT。

在未来NLP领域的研究和应用，BERT有两点值得被借鉴：其一，基于Transformer编码器作特征提取，结合MLM&NSP策略预训练；其二，超 大数据 规模预训练Pre-Training+具体任务微调训练Fine-Tuning的两阶段模式。

1.特征提取器

Transformer Encoder，特征提取器，由Nx个完全一样的layer组成，每个layer有2个sub-layer，分别是：Multi-Head Self-Attention机制、Position-Wise全连接前向神经网络。对于每个sub-layer，都添加了2个操作：残差连接Residual Connection和归一化Normalization，用公式来表示sub-layer的输出结果就是LayerNorm(x+Sublayer(x))。

Attention Mechanism。为什么要有 注意力机制 ？换句话说， 注意力机制 有什么好处？类比人类世界，当我们看到一个人走过来，为了识别这个人的身份，眼睛注意力会关注在脸上，除了脸之后的其他区域信息会被暂时无视或不怎么重视。对于语言模型，为了模型能够更加准确地判断，需要对输入的文本提取出关键且重要的信息。怎么做？对输入文本的每个单词赋予不同的 权重 ，携带关键重要信息的单词偏向性地赋予更高的 权重 。抽象来说，即是：对于输入Input，有相应的向量query和key-value对，通过计算query和key关系的function，赋予每个value不同的 权重 ，最终得到一个正确的向量输出Output。在Transformer编码器里，应用了两个Attention单元：Scaled Dot-Product Attention和Multi-Head Attention。

Scaled Dot-Product Attention。Self-Attention机制是在该单元实现的。对于输入Input，通过线性变换得到Q、K、V，然后将Q和K通过Dot-Product相乘计算，得到输入Input中词与词之间的依赖关系，再通过尺度变换Scale、掩码Mask和Softmax操作，得到Self-Attention矩阵，最后跟V进行Dot-Product相乘计算。

Multi-Head Attention。通过h个不同线性变换，将d_model维的Q、K、V分别 映射 成d_k、d_k、d_v维，并行应用Self-Attention机制，得到h个d_v维的输出，进行拼接计算Concat、线性变换Linear操作。

2.输入特征处理

BERT的输入是一个线性序列，支持单句文本和句对文本，句首用符号[CLS]表示，句尾用符号[SEP]表示，如果是句对，句子之间添加符号[SEP]。输入特征，由Token向量、Segment向量和Position向量三个共同组成，分别代表单词信息、句子信息、位置信息。

3.预训练

BERT采用了MLM和NSP两种策略用于模型预训练。为了证明这两种策略的效果，谷歌额外增加了两组对照实验。对照组一：No NSP，保留MLM，但没有NSP；对照组二：LTR & No NSP，没有MLM和NSP，替换成一个Left-to-Right（LTR）模型，甚至为了增强可信性，在对照组二的基础上增加一个随机初始化的BiLSTM。实验数据表明，BERT采用MLM&NSP策略完胜其他。

MLM，Masked LM。对输入的单词序列，随机地掩盖15%的单词，然后对掩盖的单词做预测任务。相比传统标准条件语言模型只能left-to-right或right-to-left单向预测目标函数，MLM可以从任意方向预测被掩盖的单词。不过这种做法会带来两个缺点：1.预训练阶段随机用符号[MASK]替换掩盖的单词，而下游任务微调阶段并没有Mask操作，会造成预训练跟微调阶段的不匹配；2.预训练阶段只对15%被掩盖的单词进行预测，而不是整个句子，模型 收敛 需要花更多时间。对于第二点，作者们觉得效果提升明显还是值得；而对于第一点，为了缓和，15%随机掩盖的单词并不是都用符号[MASK]替换，掩盖单词操作进行了以下改进，同时举例“my dog is hairy”挑中单词“hairy”。

80%用符号[MASK]替换：my dog is hairy -> my dog is [MASK]

10%用其他单词替换：my dog is hairy -> my dog is apple

10%不做替换操作：my dog is hairy -> my dog is hairy

NSP，Next Sentence Prediction。许多重要的下游任务譬如QA、NLI需要语言模型理解两个句子之间的关系，而传统的语言模型在训练的过程没有考虑句对关系的学习。NSP，预测下一句模型，增加对句子A和B关系的预测任务，50%的时间里B是A的下一句，分类标签为IsNext，另外50%的时间里B是随机挑选的句子，并不是A的下一句，分类标签为NotNext。

Input = [CLS] the man went to [MASK] store [SEP]

he brought a gallon [MASK] milk [SEP]

Label = IsNext

Input = [CLS] the man went to [MASK] store [SEP]

penguin [MASK] are flight ##less birds [SEP]

Label = NotNext

4.任务微调

BERT提供了4种不同下游任务的微调方案：

（a）句对关系判断，第一个起始符号[CLS]经过Transformer编码器后，增加简单的Softmax层，即可用于分类；

（b）单句分类任务，具体实现同（a）一样；

（c）问答类任务，譬如SQuAD v1.1， 问答系统 输入文本序列的question和包含answer的段落，并在序列中标记answer，让BERT模型学习标记answer开始和结束的向量来训练模型；

（d）序列标准任务，譬如命名实体标注NER，识别系统输入标记好实体类别（人、组织、位置、其他无名实体）的文本序列进行微调训练，识别实体类别时，将序列的每个Token向量送到预测NER标签的分类层进行识别。

源码分析：BERT实现

PyTorch：Torch的 python 包，而Torch是一个用于机器学习和科学计算的模块化开源库。Anaconda Python针对Interl架构调整了MKL，使得PyTorch在Interl CPU上的性能达到最佳；此外，PyTorch支持NVIDIA GPU，能够利用GPU加速训练。