HyperAIHyperAI

Command Palette

Search for a command to run...

超大规模神经网络:稀疏门控专家混合层

Noam Shazeer; Azalia Mirhoseini; Krzysztof Maziarz; Andy Davis; Quoc Le; Geoffrey Hinton; Jeff Dean

摘要

神经网络吸收信息的能力受其参数数量的限制。理论上,条件计算(即根据每个样本激活网络的部分组件)被提出作为一种在不大幅增加计算量的情况下显著提升模型容量的方法。然而,在实际应用中,该方法面临着重大的算法和性能挑战。在这项工作中,我们解决了这些挑战,最终实现了条件计算的潜力,在现代GPU集群上仅以轻微的计算效率损失就实现了超过1000倍的模型容量提升。我们引入了一种稀疏门控专家混合层(Sparsely-Gated Mixture-of-Experts, MoE),该层包含多达数千个前馈子网络。一个可训练的门控网络决定了每个样本应使用哪些专家的稀疏组合。我们将MoE应用于语言建模和机器翻译任务,这些任务中模型容量对于吸收训练语料库中的大量知识至关重要。我们展示了在堆叠LSTM层之间卷积应用具有多达1370亿参数的MoE的模型架构。在大规模语言建模和机器翻译基准测试中,这些模型在较低的计算成本下取得了显著优于现有最先进水平的结果。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
超大规模神经网络:稀疏门控专家混合层 | 论文 | HyperAI超神经