跳到正文
原文
NVIDIA Developer Blog· Michelle Horton·· 9 天前AI 评分13

生物基础模型的高效 MoE 训练

Efficient MoE Training for Biological Foundation Models

AI 导读

生物基础模型采用 Mixture-of-experts(MoE)架构进行高效训练,通过让每个 token 仅激活一部分专家子网络来扩展模型。随着模型规模增长,稠密 Transformer 中每个 token 都会经过所有层,增加能力会同步提高训练和推理的计算成本。

来源:NVIDIA Developer Blog · developer.nvidia.com