全球首创能用4090推理 昆仑万维开源2千亿稀疏大模型天工MoE (全球首创用英语怎么说)

昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

2024年6月3日,昆仑万维宣布开源 2 千亿稀疏大模型 Skywork-MoE , 性能强劲, 同时推理成本更低。Skywork-MoE 基于之前昆仑万维开源的 Skywork-13B 模型中间 checkpoint 扩展而来,是 首个完整将 MoE Upcycling 技术应用并落地的开源千亿 MoE大模型,也是首个支持用单台 4090 服务器推理的开源千亿 MoE大模型。

开源地址:

Skywork-MoE 的模型权重、技术报告完全开源,免费商用,无需申请:

•模型权重下载:

•模型开源仓库:

•模型技术报告:

•模型推理代码:(支持 8x4090 服务器上 8 bit 量化加载推理)

模型架构:

本次开源的 Skywork-MoE 模型隶属于天工 3.0 的研发模型系列,是其中的中档大小模型(Skywork-MoE-Medium),模型的总参数量为 146B,激活参数量 22B,共有 16 个 Expert,每个 Expert 大小为 13B,每次激活其中的 2 个 Expert。

天工 3.0 还训练了 75B (Skywork-MoE-Small) 和 400B (Skywork-MoE-Large)两档 MoE 模型,并不在此次开源之列。

模型能力:

我们基于目前各大主流模型评测榜单评测了 Skywork-MoE,在相同的激活参数量 20B(推理计算量)下,Skywork-MoE 能力在行业前列,接近 70B 的 Dense 模型。使得模型的推理成本有近 3 倍的下降。同时 Skywork-MoE 的总参数大小比 DeepSeekV2 的总参数大小要小 1/3,用更小的参数规模做到了相近的能力。

昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

技术创新:

为了解决 MoE 模型训练困难,泛化性能差的问题,相较于 Mixtral-MoE, Skywork-MoE 设计了两种训练优化算法

1.Gating Logits 归一化操作

我们在 Gating Layer 的 token 分发逻辑处新增了一个 normalization 操作,使得 Gating Layer 的参数学习更加趋向于被选中的 TOP-2 experts,增加 MoE 模型对于 top-2 的置信度:

昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

2.自适应的 Aux Loss

有别于传统的固定系数(固定超参)的 aux loss, 我们在 MoE 训练的不同阶段让模型自适应的选择合适的 aux loss 超参系数,从而让 Drop Token Rate 保持在合适的区间内,既能做到 expert 分发的平衡,又能让 expert 学习具备差异化,从而提升模型整体的性能和泛化水平。在 MoE 训练的前期,由于参数学习不到位,导致 Drop Token Rate 太高(token 分布差异太大),此时需要较大的 aux loss 帮助 token load balance;在 MoE 训练的后期,我们希望 Expert 之间仍保证一定的区分度,避免 Gating 倾向为随机分发 Token,因此需要较低的 aux loss 降低纠偏。

昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

训练 Infra

如何对 MoE 模型高效的进行大规模分布式训练是一个有难度的挑战,目前社区还没有一个最佳实践。Skywork-MoE 提出了两个重要的并行优化设计,从而在千卡集群上实现了 MFU 38% 的训练吞吐,其中 MFU 以 22B 的激活参数计算理论计算量。

1. Expert> 昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

一个最简单的 EDP 的例子,两卡情况下 TP = 2, EP = 2, 其中 Attention 部分采用 Tensor Parallel , Expert 部分采用 Expert Parallel

2.非均匀切分流水并行

由于 first stage 的 Embedding 计算和 last stage 的 Loss 计算,以及 Pipeline Buffer 的存在, 流水并行下均匀切分 Layer 时的各 stage 计算负载和显存负载均有较明显的不均衡情况。我们提出了非均匀的流水并行切分和重计算 Layer 分配方式,使得总体的计算/显存负载更均衡,约有 10% 左右的端到端训练吞吐提升。

昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

比较均匀切分和非均匀切分下的流水并行气泡:对于一个 24 层 Layer 的 LLM, (a) 是均匀切分成 4 个 stage,每个 stage 的 layer 数量是:[6, 6, 6, 6].(b) 是经过优化后的非均匀切分方式,切成 5 个 stage, 每个 stage 的 layer 数量是:[5, 5, 5, 5, 4] , 在中间流水打满的阶段,非均匀切分的气泡更低。

MoE Know-how

此外,Skywork-MoE 还通过一系列基于 Scaling Laws 的实验,探究哪些约束会影响 Upcycling 和 From Scratch 训练 MoE 模型的好坏。

昆仑万维开源2千亿稀疏大模型天工MoE,全球首创能用4090推理

一个可以遵循的经验规则是:如果训练 MoE 模型的 FLOPs 是训练 Dense 模型的 2 倍以上,那么选择 from Scratch 训练 MoE 会更好,否则的话,选择 Upcycling 训练 MoE 可以明显减少训练成本。

4090 推理

Skywork-MoE 是目前能在 8x4090 服务器上推理的最大的开源 MoE 模型。8x4090 服务器一共有 192GB 的 GPU 显存,在 FP8 量化下(weight 占用 146GB),使用我们首创的非均匀 Tensor Parallel 并行推理方式,Skywork-MoE 可以在合适的 batch size 内达到 2200 tokens/s 的吞吐。天工团队完整开源了相关的推理框架代码和安装环境,详情参见:

结语

我们希望本次开源的 Skywork-MoE 模型、技术报告和相关的实验结果可以给开源社区贡献更多的 MoE 训练经验和 Know-how,包括模型结构、超参选择、训练技巧、训练推理加速等各方面, 探索用更低的训练推理成本训更大更强的模型,在通往 AGI 的道路上贡献一点力量。

版权文章,未经授权禁止转载。详情见 转载须知 。

全局中部横幅
360图片

更精准的搜索,更流畅的观赏;360图片收录几十亿高清美图,为用户提供壁纸、素材、头像、写真、摄影、风景等最新、最全的高质量图片搜索服务!

手机主页

中关村在线手机频道为用户提供最新、最深度、最全面的手机相关资讯。能够让用户在最快的时间内了解到如何选手机、买手机、用手机。

有问题找答案来小问小答

小问小答是全球新型的中文问答互动平台,每天为网民答疑解惑。小问小答通过技术实现检索和推荐,让您的每个疑问能够快速获得有效解答。如果您有问题就来小问小答找答案吧。

青岛航顺设备安装有限公司

青岛航顺设备安装有限公司在机械行业服务多年。企业前身为青岛顺兴机械化装卸队,始建于1986年,以大型设备起重、装卸、搬运安装为主,2002改制为公司(股份)注册资本400万元,现有固定资产1800万元。刘经理:13705324183

垃圾分选设备

郑州市城盈机械设备有限公司是一家新型高新科技技术型企业,经营项目涵盖垃圾碳化机和垃圾分选机以及全系列垃圾处理设备,合作唯一原则,为客户提供可靠的产品,真实的技术指导,完善的技术服务,广交天下朋友,共创我国民族工业的辉煌!

招投标资质认证

君嘉资质,专注于企业资质认证服务咨询辅导,提供售后服务体系认证.业务范围涵盖:招投标资质认证,投标加分证书、ISO9001质量管理体系认证、招投标资质、AAA信用等级认证,信用评级、控标证书等

太阳能路灯厂家

江苏顺泰交通是专业生产中杆灯高杆灯,市政路灯太阳能路灯,庭院灯景观灯,玉兰灯智慧路灯,风光互补路灯厂家,免费咨询15052509897微信同号,品质好价格优,欢迎前来选购!

瑞美锦鑫

瑞美锦鑫以“中小学生身心健康发展”为重点,建立学生信息化电子档案系统,建立完善健康监测体系,通过人工智能与大数据分析,实现健康教育数据的常态化、信息化、标准化评价体系,助力中国学生健康教育。

LED显示屏租赁

上海锐屏LED显示屏租赁提供LED大屏幕租赁、显示屏租赁、LED大屏幕出租等,曾为上海各类演出、会展等提供LED大屏幕租赁、大屏幕出租、LED显示屏租赁,并取得了令人满意的成绩。

成都曜奇科技有限公司

成都曜奇科技有限公司是一家专注于网络与信息安全的高科技企业,业务涵盖软件开发、智能机器人的研发、新兴能源技术及物联网技术等多个领域。我们致力于为客户提供全面的技术服务,包括技术咨询、技术推广及市场营销策划等。公司还积极拓展互联网数据与安全服务,确保客户的数据安全与隐私保护。此外,我们提供专业的广告设计、会议展览服务以及文艺创作等多样化服务,以满足不同客户的需求。

全局底部横幅