Google发布24个小型BERT模型,直接通过MLM损失进行预训练

新闻 人工智能
Google最新推出24个精简的BERT模型供下载,赶在白色情人节之前为广大NLP研究者送上一份厚礼。

 [[318598]]

Google最新推出24个精简的BERT模型供下载,赶在白色情人节之前为广大NLP研究者送上一份厚礼。

 

BERT胜在模型大,也败在模型大!
BERT一出世就带着「狂破11项记录」的光环,用压倒性的参数量暴力突围。可以说,BERT的成功,成就成在模型大。但大,一方面成就了BERT,另一方面也成了它无法被更广泛应用的障碍,这体现在三个方面。

障碍一:占资源

巨大的体积使得它必须占用巨大的存储资源,而维护大量存储资源的同时也在不断消耗大量的能源。

障碍二:太耗时

BERT作者Jacob曾提到「BERT-Large模型有24层、2014个隐藏单元,在33亿词量的数据集上需要训练40个Epoch,在8块P100上可能需要1年」。

障碍三:费用高

有人曾经计算过,三大主流模型的训练成本大概是:

  • BERT:1.2 万美元
  • GPT-2:4.3 万美元
  • XLNet:6.1 万美元

为了解决上述问题,人们不断推出精简版BERT

Size matters。其实上述3个问题,源头就在于Size,因此大家开始不断的研究精简版BERT。在资源有限的情况下,小体积的模型有着不言而喻的优势。

DistillBERT:Hugging Face团队在BERT的基础上用知识蒸馏技术训练出来的小型化BERT。模型大小减小了40%(66M),推断速度提升了60%,但性能只降低了约3%。

ALBERT:通过改变模型架构,极大的降低了体积。最小的ALBERT只有12M,最大ALBERT-XXLarge有233M,而BERT-Base就有110M。不过虽然体积下去了,推理速度没有太明显的提升。

TinyBERT:用知识蒸馏的方法来压缩模型。这个模型由华中科大和华为共同出品。

Google亲自推出更小型BERT模型

就在昨天,Google更新了BERT的GitHub仓库,发布了24个较小的BERT模型,仅限英语词汇且不分大小写。

该24个模型使用WordPiece masking进行训练,直接通过MLM损失进行预训练,可以通过标准训练或最终任务提炼针对下游任务进行微调,在MLM蒸馏的更精细的预训练策略方面极具优势。

理论指导来自论文《Well-Read Students Learn Better: On the Importance of Pre-training Compact Models》。论文地址:https://arxiv.org/abs/1908.08962

小的BERT模型只需要按照原有BERT微调方式进行微调即可,只不过在由更大、更精准的teacher制作的知识蒸馏情况下最有效。

需要注意的是,本发行版中的BERT-Base模型仅是出于完整性考虑,在和原始模型相同的条件下进行了重新训练。

GLUE分数:


24个小型BERT下载地址:https://storage.googleapis.com/bert_models/2020_02_20/all_bert_models.zip

 

 

责任编辑:张燕妮 来源: 新智元
相关推荐

2020-03-17 09:42:00

谷歌开源机器学习

2023-02-28 13:09:53

训练模型

2024-01-29 00:24:07

图像模型预训练

2017-08-09 10:02:12

NMT神经网络自然语言处理

2019-07-21 19:45:23

GitHub代码开发者

2021-11-18 11:18:51

人工智能机器学习技术

2019-03-10 22:22:06

PyTorchKeras深度学习

2021-08-30 09:25:25

Bert模型PyTorch语言

2022-08-01 10:07:34

训练模型图片

2021-12-06 09:53:09

自然语言神经网络人工智能

2017-12-26 13:53:31

深度学习迁移学习

2015-10-14 11:42:44

安全漏洞Chrome 46浏览器安全

2022-12-31 13:55:28

模型训练

2023-09-04 19:19:36

语言模型LLM

2022-09-30 15:28:05

BERT语言模型自然语言

2023-02-27 12:57:32

BERTChatGPT机构

2018-12-24 09:42:53

人工智能机器学习技术

2020-10-26 15:48:32

NLP自然语言处理数据

2023-06-12 07:50:45

2020-07-29 12:16:12

预训练自训练神经网络
点赞
收藏

51CTO技术栈公众号