Pytorch实战笔记(3)——BERT实现情感分析

本文展示的是使用 Pytorch 构建一个 BERT 来实现情感分析。本文的架构是第一章详细介绍 BERT，其中包括 Self-attention，Transformer 的 Encoder，BERT 的输入与输出，以及 BERT 的预训练和微调方式；第二章是核心代码部分。

1 BERT

1.1 self-attention

Self-attention 接受一个序列输入，并输出等长的序列。其运行流程如下。

上图是 self-attention 的部分实例，因为仅展示了 $b_1$ 的计算过程。其计算过程如下所述（这里仅说明 $b_1$ 的计算过程，$b_2$ 到 $b_4$ 的计算方式与 $b_1$ 一样）：

1. 对于输入序列 $\{a_1,a_2,a_3,a_4\}$，当我们计算 $a_1$ 对该输入序列的注意力向量时，$a_1$ 会经过三次不同的线性变换，得到 $q_1$、$k_1$、$v_1$ 向量，公式如下。这里的 q (query)、k (key)、v (value) 可以用数据库来理解，q 对应的就是 SQL 语句，来查询某个键，最后返回这个键的值，就比如 q 是 ‘select age from girlfriend’，这里 query 就是这个 sql 语句，key 就是 age，value 就是 18。
   $$\begin{gathered} q_1=W^q{a}_1, \\ {k}_1=W^k{a}_1, \\ {v}_1=W^v{a}_1. \end{gathered}$$
2. 而对于 $\{a_2,a_3,a_4\}$ 而言，它们是被查询注意力的对象，所以只生成 k 和 v（这里需要注意的是，self-attention 是会计算自己对自己的注意力的，所以会有 k1 和 v1）。
3. 接着 $q_1$ 会与 $\{k_1,k_2,k_3,k_4\}$ 分别做一次点积操作，得到注意力权重 $\{{\alpha }_{1,1},{\alpha }_{1,2},{\alpha }_{1,3},{\alpha }_{1,4}\}$，公式如下。这里需要注意的是，由于 $k$ 会经过一次转置，所以注意力权重 $\alpha$ 是标量。同时，由于点积操作可以看做是一次相似度的计算（因为余弦相似度的计算公式是 $\cos \theta =\frac{a\cdot b}{|a||b|}$，即 $a\cdot b=|a||b|\cos \theta$，所以内积可以看做是计算两个向量的相似度），所以这里内积就可以理解为计算 $q_1$ 与$\{k_1,k_2,k_3,k_4\}$ 的一次相似度权重计算（因为 $\alpha$ 是标量，所以是相似度权重）。
   $$\{a_{1,1},{\alpha }_{1,2},{\alpha }_{1,3},{\alpha }_{1,4}\}=q_1\{k_1,k_2,k_3,k_4{\}}^{\mathrm{T}}.$$
4. 最后，相似度权重 $\{{\alpha }_{1,1},{\alpha }_{1,2},{\alpha }_{1,3},{\alpha }_{1,4}\}$ 与 $\{v_1,v_2,v_3,v_4\}$ 相乘，分别得到 $a_1$ 对 $a_1$ 的注意力向量、$a_1$ 对 $a_2$ 的注意力向量、$a_1$ 对 $a_3$ 的注意力向量、和$a_1$ 对 $a_4$ 的注意力向量。接着将这些向量拼起来，就得到了 $b_1$。$b_1$ 里面就包含了 $a_1$ 对整个输入序列的所有注意力向量。

1.2 multi-head self-attention

多头自注意力机制实际上就是计算多次 self-attention。如下图所示。

multi-head self-attention 就是输入的向量会经过 $h$ 个不同的线性变换，得到 $h$ 个 q、k、v。比如 $h=2$ 的时候，$a_1$ 会通过以下公式得到 $q_1^1$、$k_1^1$、$v_1^1$ 和 $q_1^2$、$k_1^2$、$v_1^2$：
$$\begin{gathered} q_1^1=W_1^q{a}_1, \\ {k}_1^1=W_1^k{a}_1, \\ {v}_1^1=W_1^v{a}_1, \\ {q}_1^2=W_2^q{a}_1, \\ {k}_1^2=W_2^k{a}_1, \\ {v}_1^2=W_2^v{a}_1. \end{gathered}$$
接着，每个 self-attention 后的输出，会拼在一起，再通过一个线性转换，得到 multi-head self-attention 的输出。设第一个头的输出为 $hea{d}_1$，第二个头的输出为 $hea{d}_2$，最后的输出为 $O$，则其计算公式为：
$$O=\mathrm{concat}(hea{d}_1,hea{d}_2)W^o.$$

1.3 Encoder

这里的 Encoder 特指的是 Transformer[1] 中的 Encoder（左边是 Transformer 的 Encoder，右边是 Decoder），其模型结构如下：

Encoder 中一共有以下几个部分：

• Multi-head self-attention：在前面已介绍过了。
• 残差连接 (Residual connection)[2]：对应的是图中的 Add。残差连接如下图所示。简单来说，残差连接就是将一个模块的输入与其输出相加，通常使用在层次较深的结构当中。那么为什么残差连接在深层次模型中有效？具体而言，如果不采用残差连接，那么前向传播为 $F(x)$，反向传播的时候，求梯度就为 $\frac{\partial (F(x))}{\partial x}$，当梯度消失的时候，$\frac{\partial (F(x))}{\partial x}$ 就为0，就无法回传梯度。而当采用了残差连接后，前向传播变为 $F(x)+x$。从直觉上来讲，这样能够让模型更关注于经过了这个模块后变化的部分；而从数学上来将，在反向传播的时候会变成 $\frac{\partial (F(x)+x)}{\partial x}=\frac{\partial (F(x))}{\partial x}+1$。当梯度消失后，那么 $\frac{\partial (F(x))}{\partial x}$ 趋近于0，所以 $\frac{\partial (F(x)+x)}{\partial x}$ 趋近于1，使得梯度无法消失，始终能够回传。
  
• 层归一化 (layer norm)[3]：对应的是图中的 Norm。层归一化的公式如下所示。其中，$m$ 是向量 $x_i$ 的均值，$\sigma$ 是向量 $x_i$ 的标准差。层归一化的示例图如下所示。具体而言，如果数据不做归一化，有可能在某些方向上梯度下降很快（从左下到右上），这样会导致越过最优点；有的方向上下降很慢（从右下到左上），这样会导致半天收敛不到最优点。而通过层归一化后，能够使得数据在各个方向上都能够下降的一样快，使得能够更快收敛。
  $$x_i^{\prime }=\frac{x_i-m}{\sigma }$$
  
• 位置嵌入 (Positional Encoding)：对应的是图中最下面的 Positional Encoding。为什么要位置嵌入？由于 self-attention 可以看做是下图这样，两个位置之间间隔为1。如果不能理解为什么是 1，可以再回过头看看上面那个 gif。那么这样会导致一个问题，对于自然语言处理的任务而言，词语的先后顺序肯定是很重要的，就比如我现在这里写到了 positional encoding，那么和第一小节写的 self-attention 关联就很弱了，所以需要通过位置嵌入来控制词语与词语之间的位置。
  
• 全连接层：对应图中的 Feed forward，没什么好说的，唯一要注意的是，这里是两层全连接层，公式如下：
  $$FFN(x)=W_2(\mathrm{ReLU}(W_{1}x+b_1))+b_2$$

1.4 BERT 的输入与输出

1.4.1 BERT 的输入

BERT 的输入与传统的语言模型输入不同，传统的语言模型的输入就只是整个句子，而 BERT 在输入中还加入了几个特殊的字符。其中包括：

• [CLS]：[CLS] 一定出现在句首，这个特殊字符通过 BERT 后得到的隐藏状态代表了该句子的句向量。 [CLS] 是一定会有的。
• [SEP]：[SEP] 一定出现在句子的结尾。由于 BERT 支持单句和两句话输入，所以用 [SEP] 来区分哪句话是哪句话。[SEP] 是一定会有的。
• [MASK]：[MASK] 会出现在 [CLS] 与 [SEP] 中的任意位置，该特殊字符是让 BERT 去预测这个位置是什么词语。[MASK] 不一定会有。

以以下两句话为例 练习时长两年半 和 唱跳 rap 打篮球，那么输入进 BERT 后会变成以下这样：[CLS] 练习时长两年半 [SEP] 唱跳 rap 打篮球 [SEP]；如果只有前一句话输入，并且掩盖掉 半 的话，那么是如下这样：[CLS] 练习时长两年[MASK] [SEP]

1.4.2 BERT 的输出

与传统的序列模型一样，BERT 的输出有两部分：

• 句向量：通过模型后，[CLS] 的隐藏状态即句向量。如果是一句话输入，那么就是这句话的句向量；如果是两句话输入，那么就是这两句话的句向量。
• 每个词语的隐藏状态：和 LSTM 一样，BERT 也会输出每个词语的隐藏状态。这里特别需要注意的是，所谓的 BERT 的词嵌入，实际上指的就是这个通过 BERT 后的隐藏状态，而非 BERT 的嵌入层。 这是因为 BERT 是基于上下文的词嵌入（contextualized word embedding），你得有上下文信息，才能叫词嵌入。

1.5 BERT 预训练

BERT 预训练有两个部分，第一个部分是 masked language model (MLM)，第二部分是 next sentence prediction (NSP)。

• MLM 简单来说就是随机将输入文本中 15% 的词语给提取出来，然后进行以下处理：1. 80% 的可能，将词语替换为 [MASK]，这是让模型通过上下文来预测这 [MASK] 是什么词语；10% 的可能，将词语随机替换为另外一个词语；10% 的可能，保持词语不变。MLM 如下图所示。MLM 是个 $V$ 分类任务，其中 $V$ 是词表大小。
  
• NSP 简单来说就是输入两句话到模型中，让模型判断后一句话是否与前一句话有关联。NSP 如下图所示。NSP 是个二分类任务，其中 1 代表上下两句话有关联，0 代表没有关联。
  

1.6 BERT 微调

微调阶段，首先 BERT 会先加载预训练好的参数，并额外添加上部分随机初始化的参数。如下图所示。图中，用橙色标出来的全连接层，就是在微调阶段随机初始化的参数。所以微调阶段，训练的为两部分内容：

• 模型本身：这部分是可以不参与训练的，因为模型已经在预训练阶段训练好了，不是必须训练的。
• 随机初始化的参数：这部分是必须训练的参数。
  

2 BERT 实现情感分析

• 全部代码在 github 上，网址为：https://github.com/Balding-Lee/Pytorch4NLP
• 我采用的是 IMDb 数据集，由于数据集没有验证集，而且读取起来很麻烦，所以我将数据给读取出来，放到了一个文件中，并且将训练集中的10%划分为了验证集，数据集链接如下： https://pan.baidu.com/s/128EYenTiEirEn0StR9slqw，提取码：xtu3 。
• 采用的词嵌入是谷歌的词嵌入，词嵌入的链接如下：链接：https://pan.baidu.com/s/1SPf8hmJCHF-kdV6vWLEbrQ，提取码：r5vx
  在本博客中仅介绍模型部分，详细代码见 github。

具体的模型代码如下：

import torch
import torch.nn as nn
from transformers import BertTokenizer, BertConfig, BertForSequenceClassification


class Config:
    def __init__(self):
        # 训练配置
        self.seed = 22
        self.batch_size = 64
        self.lr = 1e-5
        self.weight_decay = 1e-4
        self.num_epochs = 100
        self.early_stop = 512
        self.max_seq_length = 128
        self.save_path = '../model_parameters/BERT_SA.bin'

        # 模型配置
        self.bert_hidden_size = 768
        self.model_path = 'bert-base-uncased'
        self.num_outputs = 2


class Model(nn.Module):
    def __init__(self, config, device):
        super().__init__()
        self.config = config
        self.device = device
        tokenizer_class, bert_class, model_path = BertTokenizer, BertForSequenceClassification, config.model_path
        bert_config = BertConfig.from_pretrained(model_path, num_labels=config.num_outputs)
        self.tokenizer = tokenizer_class.from_pretrained(model_path)
        self.bert = bert_class.from_pretrained(model_path, config=bert_config).to(device)

    def forward(self, inputs):
        tokens = self.tokenizer.batch_encode_plus(inputs,
                                                  add_special_tokens=True,
                                                  max_length=self.config.max_seq_length,
                                                  padding='max_length',
                                                  truncation='longest_first')

        input_ids = torch.tensor(tokens['input_ids']).to(self.device)
        att_mask = torch.tensor(tokens['attention_mask']).to(self.device)

        logits = self.bert(input_ids, attention_mask=att_mask).logits

        return logits

实验结果如下：

test loss 0.281900 | test accuracy 0.878846 | test precision 0.853424 | test recall 0.915280 | test F1 0.883270

参考

[1] Ashish Vaswani, Noam Shazeer, Niki Parmar, et al. Attention is all you need [EB/OL]. https://arxiv.org/abs/1706.03762, 2017.
[2] Kaiming He, Xiangyu Zhang, Shaoqing Ren, et al. Deep Residual Learning for Image Recognition [EB/OL]. https://arxiv.org/abs/1512.03385, 2015.
[3] Jimmy Lei Ba, Jamie Ryan Kiros, Geoffrey E. Hinton. Layer Normalization [EB/OL]. https://arxiv.org/abs/1607.06450, 2016.