NLP预训练模型

Models Corpus

---

RoBERTa: A Robustly Optimized BERT Pretraining Approach

与BERT主要区别在于：

• large mini-batches 保持总训练tokens数一致，使用更大的学习率、更大的batch size，adam ${\beta }_2=0.98$；
• dynamic masking 动态掩盖，同一份样本重复10次；
• FULL-SENTENCES without NSP 做了四种输入格式实验，验证了NSP任务的无效性，DOC-SENTENCES方式最优：
  • SEGMENT-PAIR+NSP: BERT输入类型，以一对文本段做了输入，文本段包含多句输入，总token数小于512；
  • SENTENCE-PAIR+NSP: 以句对作为输入，总长度可能远小于512，增加batch size使得单批次总tokens数接近其它方法；
  • FULL-SENTENCES:从单篇文章或多篇文章采样的连续句子，不同文章句采用特殊标记拼接，总长度最多512；
  • DOC-SENTENCES: 从单篇文章采样的连续句子，总长度可能不足512，增加batch_size保持单批次总tokens数接近其它方法；
• larger byte-level BPE 词表大小从30K提升至50K，无预处理步骤，无unknown token；

Byte-Pair Encoding (BPE)

A hybrid between character- and word-level representations that allows handling the large vocabularies common in natural language corpora.
Instead of full words, BPE relies on subwords units, which are extracted by performing statistical analysis of the training corpus.

Radford et al. (2019) introduce a clever implementation of BPE that uses bytes instead of unicode characters as the base subword units. Using bytes makes it possible to learn a subword vocabulary of a modest size (50K units) that can still encode any input text without introducing any “unknown” tokens.

---

ELECTRA: PRE-TRAINING TEXT ENCODERS AS DISCRIMINATORS RATHER THAN GENERATORS

给定一个mask的序列，对于各掩码位置，生成器预测：

判别器判别生成器预测序列中各token是否与原始token一致（Replaced Token Detection，RTD）：

与GAN的区别在于，生成器以极大似然估计方式训练，不需要欺骗判别器。判别器梯度不反向传播至生成器，下游任务仅使用判别器。

生成器生成的token与原始token的语义接近，使用生成器预测序列作为判别器输入（更难区分），比随机替换token的方式更有效。

---

DeBERTa: Decoding-enhanced BERT with Disentangled Attention

DISENTANGLED SELF-ATTENTION

改进自注意力机制，将不同位置的内容向量和相对位置向量的cross attention分数作为自注意力分数：

标准自注意力机制：

引入相对位置的分散自注意力机制：

上述矩阵各行对应各位置的向量表示，其中：

• $H,H_o\in {\mathbb{R}}^{N\times d}$，表示注意力层输入、输出隐状态；
• $Q_c,K_c,V_c$，表示经投影矩阵$W_{q,c},W_{k,c},W_{v,c}\in {\mathbb{R}}^{d\times d}$，投影后的内容向量；
• $P\in {\mathbb{R}}^{2k\times d}$，表示相对位置嵌入，所有层共享；
• $Q_r,K_r$，表示经投影矩阵$W_{q,r},W_{k,r}\in {\mathbb{R}}^{d\times d}$，投影后的相对位置向量；
• $\text{softmax}$，表示做行向量归一化，输出矩阵行向量为单位向量；

ENHANCED MASK DECODER ACCOUNTS FOR ABSOLUTE WORD POSITIONS

Given a sentence “a new store opened beside the new mall” with the words “store” and “mall” masked for prediction. Using only the local context (e.g., relative positions and surrounding words) is insufficient for the model to distinguish store and mall in this sentence, since both follow the word new with the same relative positions. For example, the subject of the sentence is “store” not “mall”. These syntactical nuances depend, to a large degree, upon the words’ absolute positions in the sentence.

语法上的细微差别，很大程度上取决于单词在句子中的绝对位置。

In DeBERTa, we incorporate them right after all the Transformer layers but before the softmax layer for masked token prediction, as shown in Figure 2. In this way, DeBERTa captures the relative positions in all the Transformer layers and only uses absolute positions as complementary information when decoding the masked words. Thus, we call DeBERTa’s decoding component an Enhanced Mask Decoder (EMD).

在所有transformers层之后、softmax之前，合并绝对位置信息，预测掩盖的token。

SCALE INVARIANT FINE-TUNING，SiFT

向标准化的word embeddings增加扰动，增强模型泛化性。

---

DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding Sharing

• 使用ELECTRA架构训练，以RTD代替MLM；
• 提出词向量梯度分散共享，优化生成器和判别器词向量共享，避免发生“tug-of-war”（激烈竞争）；

DeBERTa with RTD

Replacing the MLM objective used in DeBERTa with the RTD objective.

Token Embedding Sharing in ELECTRA

The tasks of MLM and RTD pull token embeddings into very different directions. MLM tries to map the tokens that are semantically similar to the embedding vectors that are close to each other. RTD, on the other hand, tries to discriminate semantically similar tokens, pulling their embeddings as far as possible to optimize the classification accuracy.

使用不同词向量共享和梯度传播方式，词向量平均余弦相似度比较：

Gradient-Disentangled Embedding Sharing

The training of GDES follows that of NES. $E_{\Delta }$ is initialized as a zero matrix. In each training pass:

• Run a forward pass with the generator to generate the inputs for the discriminator, and then run a backward pass with respect to the MLM loss to update $E_G$, which is shared by both the generator and the discriminator.
• Run a forward pass for the discriminator using the inputs produced by the generator, and run a backward pass with respect to the RTD loss to update ED by propagating gradients only through $E_{\Delta }$.
• After model training, $E_{\Delta }$ is added to $E_G$ and the sum is saved as $E_D$ in the discriminator.