预训练语言表达模型之BERT

2019-05-30 本文已影响0人萧风萧雨

BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

介绍

语言模型的预训练能够提高语言任务的性能（就像cv中的imagenet预训练），将预训练的语言表达应用到下游的任务(down-stream tasks)有两种策略：
（1）Feature-based：将预训练得到的特征作为下游任务的额外特征，下游任务的模型结构是根据具体任务设计的（task-specific）；
（2）Fine-tuning：下游任务使用预训练语言模型的结构，通过增加特定任务的层和finetune的方式来训练模型（结构重用）；
（PS：预训练的方法可以是无监督的，也可以是有监督的(机器翻译任务的迁移)。）
论文提出了基于fine-tuning的语言表达模型BERT(Bidirectional Encoder Representation from Transformers)，基于双向的上下文信息的transformers编码器，使用MLM(Masked Language Model)和NSP(Next Sentence Prediction)任务来预训练得到语言的隐含表达(token embedding)。同时基于finetune的模型，可以减少针对具体任务的模型设计。BERT在11个语言任务上取得了非常好的效果。

BERT

下图为BERT的预训练和fine-tuning，可以看出一个预训练的BERT可以重用于不同语言任务上，然后finetune。

模型结构：

多层双向Transformer，原论文：Attention is all you need
$\rm BERT_{BASE}$ : 层数L=12，隐含层大小H=768，head数量A=12；（110M，与Open AI GPT同样大小）
$\rm BERT_{LARGE}$ : L=24, H=1024, A=16

输入输出

输入为token，输出为token的隐含向量T
[CLS]是分类token，隐含状态为C，用于分类任务。
区分sentence：用[SEP] token；用segment embedding表示sentence A or B。
下图为Token的representation，由token，segment，position的embedding相加得到

预训练

不同于传统的left-to-right or right-to-left 的语言模型，BERT训练是通过MLM和NSP任务。

Masked Language Model

随机掩盖部分输入tokens，然后对应地输出被掩盖的tokens。论文使用15%地masks。
论文用token [MASK] 表示mask，但是在语言任务中一般是不会有这种token的，所以，论文使用如下策略解决：80%用[MASK]，10%用随机token代替，10%用原本的token。这个比例的实验如下表所示，可以看出还是有点影响的。

Next Sentence Prediction

有一些任务如QA，自然语言推断NLI需要理解句子间的关系，于是论文通过预测句子是不是下一句的关系来预训练。具体通过C来做分类。
附录中可知，MLM和NSP是结合到一起做预训练的。

Fine-tuning

Input：sentence A， B对应于sentence pairs，hypothesis-premise，question-passage，等
Output：输出T feed into 输出层做token level或者输出C用于分类。

与OPT，ELMo对比

每一层都是双向

实验

GLUE benchmark结果

论文还给出在QA数据集SQuAD 1.1， 2.0的结果

Ablation 研究，可看出NSP和MLM起了很大的作用。

总结

得益于双向transformer的结构，MLM和NSP的预训练方法，以及fine-tuning的多任务共用，使得BERT取得了非常不错的效果。另外，最近有一篇将BERT结合知识图谱的论文，也是值得一看的：ERNIE: Enhanced Language Representation with Informative Entities，ACL2019。