中国电信把新一代星辰大模型开源了,29B总参只激活4B

星辰新一代开源了,29B总参数,实际激活4B,官方明确说能本地部署。

29B激活4B这个配比看着就是MoE的路子,稀疏度七倍出头。思路不难理解:知识尽量往总参里塞,算力省在激活上,单次前向只走其中一部分专家,吞吐才能上去。

但“能本地部署”这句得拆开看。等权重到手,我准备按这几步验:

  1. 先看权重按什么精度存。 卡个人机器的从来是显存放不放得下这份权重,不是算得快不快。激活只有4B,省下来的是每一步的计算量,加载的时候该占的位置一点不少。这一关过不去,后面都免谈。
  2. 再看放出来的量化版本做到哪一档。 量化到什么位宽、哪些层保持原精度、有没有给显存不够的机器留分层加载的配置,这些直接决定单卡塞不塞得进去、塞进去之后掉多少质量。
  3. 然后跑一遍激活分布,看路由策略调得怎么样。 MoE最容易翻车的地方就是专家利用率不均:中文长文本和代码这两类输入,分布差得很远,很容易出现一部分专家被反复命中、另一部分几乎全程闲着。闲着的专家照样占显存,账面上那个稀疏度也就跟着打折。

配置文件里的路由超参,和实际跑出来的激活直方图能对上,这版才算调得可以。

顶

本地部署这四个字水很深,24G显存跑不跑得动才是关键。

我拿上一代在两张4090上试过,纯推理还行,一上长上下文KV cache就炸。MoE省的是算力不是显存,这点很多人搞混。真要塞单卡还得等社区出量化。

开源协议是哪个?能不能商用比参数重要多了

1 个赞

四楼问协议的,这个确实得先看清楚,前几年在这上面吃过亏。

激活4B的话端侧其实有戏,就是不知道还有没有更小的尺寸 :thinking: