本論文では、自然言語理解タスクと生成タスクの両方で微調整が可能な新しい統合事前学習言語モデル(UniLM)を紹介する。このモデルは、一方向性、双方向性、シーケンス間予測の3種類の言語モデリングタスクを用いて事前に学習される。統一されたモデリングは、共有されたトランスフォーマーネットワークを採用し、予測がどのような文脈で行われるかを制御するために特定の自己注意マスクを利用することで達成される。UniLMは、GLUEベンチマーク、SQuAD 2.0やCoQAの質問応答タスクにおいて、BERTと比較して良好な結果を示した。さらにUniLMは、CNN/DailyMail抽象要約ROUGE-Lを40.51に(2.04絶対改善)、Gigaword抽象要約ROUGE-Lを35.75に(0.86絶対改善)、CoQA生成的質問応答F1スコアを82.5に(37.1絶対改善)、SQuAD質問生成BLEU-4を22.12に(3.75絶対改善)、DSTC7文書に基づく対話応答生成NIST-4を2.67に改善(人間の成績は2.65)するなど、5つの自然言語生成データセットにおいて新たな最先端の結果を達成している。コードと事前訓練モデルは、https://github.com/microsoft/unilm から入手可能。
This paper presents a new Unified pre-trained Language Model (UniLM) that can be fine-tuned for both natural language understanding and generation tasks. The model is pre-trained using three types of language modeling tasks: unidirectional, bidirectional, and sequence-to-sequence prediction. The unified modeling is achieved by employing a shared Transformer network and utilizing specific self-attention masks to control what context the prediction conditions on. UniLM compares favorably with BERT on the GLUE benchmark, and the SQuAD 2.0 and CoQA question answering tasks. Moreover, UniLM achieves new state-of-the-art results on five natural language generation datasets, including improving the CNN/DailyMail abstractive summarization ROUGE-L to 40.51 (2.04 absolute improvement), the Gigaword abstractive summarization ROUGE-L to 35.75 (0.86 absolute improvement), the CoQA generative question answering F1 score to 82.5 (37.1 absolute improvement), the SQuAD question generation BLEU-4 to 22.12 (3.75 absolute improvement), and the DSTC7 document-grounded dialog response generation NIST-4 to 2.67 (human performance is 2.65). The code and pre-trained models are available at https://github.com/microsoft/unilm.