Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

ERNIE 2.0: A Continual Pre-training Framework for Language Understanding

About

Recently, pre-trained models have achieved state-of-the-art results in various language understanding tasks, which indicates that pre-training on large-scale corpora may play a crucial role in natural language processing. Current pre-training procedures usually focus on training the model with several simple tasks to grasp the co-occurrence of words or sentences. However, besides co-occurring, there exists other valuable lexical, syntactic and semantic information in training corpora, such as named entity, semantic closeness and discourse relations. In order to extract to the fullest extent, the lexical, syntactic and semantic information from training corpora, we propose a continual pre-training framework named ERNIE 2.0 which builds and learns incrementally pre-training tasks through constant multi-task learning. Experimental results demonstrate that ERNIE 2.0 outperforms BERT and XLNet on 16 tasks including English tasks on GLUE benchmarks and several common tasks in Chinese. The source codes and pre-trained models have been released at https://github.com/PaddlePaddle/ERNIE.

Yu Sun, Shuohuan Wang, Yukun Li, Shikun Feng, Hao Tian, Hua Wu, Haifeng Wang• 2019

Related benchmarks

TaskDatasetResultRank
Natural Language UnderstandingGLUE (dev)
SST-2 (Acc)96
504
Natural Language UnderstandingGLUE (test)
SST-2 Accuracy95.6
416
Natural Language InferenceXNLI (test)
Average Accuracy81
167
Natural Language InferenceSNLI (dev)
Accuracy82.6
71
Machine Reading ComprehensionDRCD (dev)
EM89.7
45
Machine Reading ComprehensionDRCD (test)
EM89
45
Machine Reading ComprehensionCMRC 2018 (dev)
EM71.5
34
Sentiment AnalysisChnSentiCorp (dev)
Accuracy96.1
33
Sentiment AnalysisChnSentiCorp (test)
Accuracy95.8
33
Question MatchingLCQMC (dev)
Accuracy90.9
24
Showing 10 of 36 rows

Other info

Code

Follow for update