AI知识库

53AI知识库

学习大模型的前沿技术与行业应用场景


OLMoE,首个“完全开源”MOE模型,干货满满
发布日期:2024-09-12 17:29:13 浏览次数: 1561



正常的MOE结构,用到了qk norm

全文61页,细节满满,感兴趣的同学可以自己看。这里简单提一些有意思的点~

1B-7BMOE vs 1B / 7B dense起点更高,终点接近或超过7B全量全参dense的效果

更细粒度的专家组合可以得到更好的训练损失,但是收益递减。(这里提到了很多相关的研究,去做预测最优组合)

共享专家会消除模型的灵活性,让性能更差,与一些历史研究不符

确定哪些专家处理每个输入token ,有2种类型

  • EC,每个专家从输入序列中选择固定数量的token。(确保完美的负载平衡,但是不利于自回归的生成模式,也可能导致token丢失)
  • TC,每个token会选择固定数量的专家。)这可能会导致许多token选择同一位专家,从而损害训练效率)

相同的预算下,TC稳定优于EC

专家是从头初始化,还是从MLP复制得到。实验发现只需要几百B的token,从头开始的模型就赶上了复制的结果,从600B token之后优于复制的结果。

使用负载平衡损失也能带来更好的性能

路由z-loss可以提高MoE模型的稳定性和质量。这个辅助损失会惩罚进入门控网络的大logits

预训练的早期阶段路由就达到了饱和状态

训练结束后,通一层的专家之间不存在强协同激活,也就是说不同专家之间几乎没有冗余

专家对于特定领域和词汇的专业化程度区分度还是比较高的。某些专家可能专门处理一些科学相关的,(如arXiv数据集),而其他专家可能更擅长处理编程语言或一般文本。



53AI,企业落地应用大模型首选服务商

产品:大模型应用平台+智能体定制开发+落地咨询服务

承诺:先做场景POC验证,看到效果再签署服务协议。零风险落地应用大模型,已交付160+中大型企业

联系我们

售前咨询
186 6662 7370
预约演示
185 8882 0121

微信扫码

与创始人交个朋友

回到顶部

 
扫码咨询