AI魔法学院客服
大模型再添重磅玩家:清华大学NLP实验室开源可商用大模型CPM-Bee
 2023-06-01
复制地址分享海报

5月27日,OpenBMB发布了一个最高有100亿参数规模的开源大语言模型CPM-BEE,OpenBMB是清华大学NLP实验室联合智源研究院成立的一个开源组织。该模型针对高质量中文数据集做了训练优化,支持中英文。根据官方的测试结果,其英文测试水平约等于LLaMA-13B,中文评测结果优秀。

本文原文:国产开源中文大语言模型再添重磅玩家:清华大学NLP实验室发布开源可商用大语言模型CPM-Bee | 数据学习者官方网站(Datalearner)



此前,国产开源大语言模型最强的应该是ChatGLM-6B(ChatGLM-6B(ChatGLM-6B)详细信息 | 名称、简介、使用方法,开源情况,商用授权信息 | 数据学习 (DataLearner) ),这也是清华大学研究成果,是清华大学KEG小组发布的,在国内外都有很好的反响,截止5月26日,ChatGLM-6B 全球下载达到200万。相比较ChatGLM-6B,CPM-Bee的主要特点如下:

  1. 开源协议更加友好,只要获取书面授权即可商用。代码、预训练结果均开源,训练过程也是全球直播。
  2. 从评测结果看,CPM-Bee模型效果也很优秀,中文ZeroCLUE评测中远超其它模型,英文多个任务评测结果也与LLaMA-13B差不多,部分任务上的表现超越GPT-3。
  3. 推理成本与ChatGLM-6B差不多,最低10亿参数规模的模型仅需要4.1GB显存可做推理,而100亿参数规模的推理仅需20GB显卡,单张消费级显卡即可。

总的来说,CPM-Bee是值得大家关注的模型,如果商用协议未来保持友好,且不断升级迭代,那也许会是一个可以和ChatGLM-6B模型竞争的一个国产替代好选择~

CPM-Bee简介

CPM全称Chinese Pretrained Model,Bee是该系列模型的第二个里程碑版本。

CPM-Bee模型是基于CPM-Ant模型继续训练得到。后者是2022年5月到9月训练的大语言模型。而CPM-Bee则是从2022年10月13日开启训练,相比之前,模型在很多任务上做了优化,包括文字填空、文本生成、问答等。

这是一个基于transformer架构的自回归模型,在高质量的中英文数据集上训练了半年左右的时间。包含4个系列的预训练结果,参数分别是10亿、20亿、50亿和100亿。CPM-Bee模型的相关信息如下:

CPM-Bee模型参数

CPM-Bee模型参数结果

模型名称

CPM-Bee

模型架构

transformer

基础模型

CPM-Ant

模型最大参数

100亿

训练时间

2022年10月13日-2023年3月29日

模型类型

基础语言模型

训练数据集

200GB高质量中文数据集+400GB多语言数据集

训练数据集tokens数

1万亿

根据官方的描述,CPM-Bee的中文数据集原始数据有1TB左右,做了预处理清洗之后有200GB的高质量中文数据集结果。并加入了400GB的多语言数据集,但没有透露数据集的具体信息。

CPM-Bee训练时间230天,成本85万左右。训练的tokens数量约6380亿左右(本部分数据与GitHub介绍有冲突)。

此外,CPM-Bee的一个特点是在推理上占用的资源很小:

CPM-Bee模型版本

推理的显存

推荐的硬件

CPM-Bee-10B

20GB

RTX3090(24GB)

CPM-Bee-5B

11 GB

RTX 3090(24 GB)

CPM-Bee-2B

6.7 GB

GTX 1080(8 GB)

CPM-Bee-1B

4.1 GB

GTX 1660(6 GB)

可以看到,最小的10亿参数版本的模型仅需要4.1GB显存即可运行!50亿参数规模版本需要11GB显存。另一个清华大学著名的开源模型ChatGLM-6B版本的推理需要13GB显存。

CPM-Bee的效果

为了测试CPM-Bee的训练效果,官方在中英文的数据集上做了多项测试。

首先是在中文的ZeroCLUE上的测试结果。这是一个专门针对中文大语言模型的评测排行。CPM-Bee的评测得分排行第二,仅次于人类。超过了百度、阿里云等相关模型。

在英文评测上,其测试结果也超过了LLaMA-7B,与LLaMA-13B的得分相当。甚至在BooQ、ARC-c和OBQA的任务上超过了GPT-3!可以说是十分不错。

模型

Average

BoolQ

PIQA

SIQA

HellaSwag

WinoGrande

ARC-e

ARC-c

OBQA

GPT-3


60.5

81

-

78.9

70.2

68.8

51.4

57.6

Gopher


79.3

81.8

50.6

79.2

70.1

-

-

-

Chinchilla


83.7

81.8

51.3

80.8

74.9

-

-

-

PaLM


84.8

80.5

-

79.7

77

75.2

52.5

50.4

LLaMA-7B

66.13

76.5

79.8

48.9

76.1

70.1

72.8

47.6

57.2

LLaMA-13B

68.08

78.1

80.1

50.4

79.2

73

74.8

52.7

56.4

CPM-Bee

67.80

78.69

77.58

61.11

78.89

61.88

66.88

54.18

63.20

CPM-Bee的开源协议

CPM-Bee作为一个完全开源的中文大语言模型,不仅测试结果很好,在开源方面也十分“大方”。

官方使用了自己的一个自定义开源协议,不过限制较为宽松。代码和模型预训练结果均开源,而模型的预训练结果的商用也只是需要获取官方书面授权即可(官方未说明是否收费)。

另一个比较有意思的是CPM-Bee模型的训练过程也是一个完全开放的过程,官方在社区上公布了完整的模型训练过程,目前最新的训练停留在2023年3月27日,loss结果0.691974,成本85万,时间230天。

除此之外,官方公布了训练过程中遭遇到的一系列问题,包括显存不足、磁盘空间不足等一系列问题,也是大家观察大模型训练最好的方法。

具体的模型信息和下载地址大家参考CPM-Bee模型的模型信息卡信息:CPM-Bee(Chinese Pretrained Model - Bee)详细信息 | 名称、简介、使用方法,开源情况,商用授权信息 | 数据学习 (DataLearner)