ERNIE 2.0

ERNIE 2.0 预训练模型详情

模型全称

Enhanced Representation through kNowledge IntEgration 2.0

发布组织

Baidu

模型大小

类型

自然语言处理

发布论文

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

模型简介

最近,预训练模型在各种语言理解任务中取得了最先进的成果,这表明在大规模语料库上的预训练可能在自然语言处理中发挥了关键作用。目前的预训练程序通常侧重于用几个简单的任务来训练模型,以掌握单词或句子的共现性。然而,除了共现之外,训练语料中还存在其他有价值的词汇、句法和语义信息,如命名实体、语义接近度和话语关系。为了最大限度地提取训练语料中的词汇、句法和语义信息,我们提出了一个名为ERNIE 2.0的持续预训练框架,该框架通过不断的多任务学习,逐步建立和学习预训练任务。实验结果表明,ERNIE 2.0在16个任务上的表现优于BERT和XLNet,包括GLUE基准上的英语任务和中文中的几个常见任务。