MegaScale:超过 10,000 GPU 的生产级 LLM 训练系统

本页为第 13 篇(训练、并行与 GPU Kernel 类),全部定量内容取自本地论文 PDF 13_MegaScale_2402.15627.pdf(arXiv:2402.15627v1,cs.LG,水印 2024-02-23),并逐条标注 PDF 页码与章节;PDF 共 16 页,全部完成版式文本核对,其中第 2、4、5、6、8–12 页另经原页渲染逐项目检(第 13–16 页为参考文献)。

快速标签与阅读说明

最后核验日期:(PDF 全部 16 页文本核对 + 9 页原页渲染目检;论文链接的仓库已在线核验)。

证据完整度:关键定量结论均带 PDF 页码/表号与完整测试条件,未获取的字段一律写明「论文未明确披露」;论文脚注指向的组件仓库 veScale 与基线仓库 Megatron-LM 经在线核验(R);云上产品映射为示例(E,未逐一核验规格与价格);成本公式、放置与运维建议为参数化推断(I)。本页不设任何评分。

开源边界:MegaScale 生产系统本体未随论文开源;论文脚注 3 以「组件开源进行中」指向 github.com/volcengine/veScale P(第 2 页)+ R(2026-09-15 核验)。veScale 不应被当作 MegaScale 的官方完整实现引用。

证据标签图例: P=论文 R=核验仓库 E=外部资料 I=编辑推断

1. 摘要与一句话判断

论文元数据 P(第 1-2 页)
标题MegaScale: Scaling Large Language Model Training to More Than 10,000 GPUs(编辑译名:把大语言模型训练扩展到超过 10,000 GPU)
作者与机构Ziheng Jiang、Haibin Lin、Yinmin Zhong(三人同等贡献)、Qi Huang、Yangrui Chen、Zhi Zhang、Yanghua Peng、Xiang Li、Cong Xie、Shibiao Nong、Yulu Jia、Sun He、Hongmin Chen、Zhihao Bai、Qi Hou、Shipeng Yan、Ding Zhou、Yiyao Sheng、Zhuo Jiang、Haohan Xu、Haoran Wei、Zhang Zhang、Pengfei Nie、Leqi Zou、Sida Zhao、Liang Xiang、Zherui Liu、Zhe Li、Xiaoying Jia、Jianxi Ye、Xin Jin(通讯)、Xin Liu(通讯),共 32 人;ByteDance、Peking University P(第 1 页与脚注)
arXiv / 版本arXiv:2402.15627v1(cs.LG),水印 2024-02-23;正文未标注会议录用信息,第 13 页致谢提及 NSDI 匿名审稿人与 shepherd(指向 NSDI 评审流程;正式发表信息属站外事实,本页不据以声称) P(第 1、13 页)
代码MegaScale 系统本体未随论文开源。论文脚注 3 以「working on open-sourcing components」链接 github.com/volcengine/veScale(组件仓库)P(第 2 页)+ R(2026-09-15 核验,见 #links);基线为 Megatron-LM 仓库 commit 285068c8 P(第 9 页 §6.1)
评估范围175B / 530B Transformer 模型;256–12,288 GPU(175B)与 2,240–11,200 GPU(530B);生产运行:专有模型、数万亿(multi-trillion)tokens、数周 P(第 9-11 页 §6)
一句话判断(编辑推断):I MegaScale 的价值不在单项技术,而在「把算法、通信、算子、数据管道、网络、容错拧成一个在万卡上仍保持 55.2% MFU 与 >90% 有效训练时间率的生产闭环」。对云平台团队,它是「万卡预训练平台该长什么样」的参考架构:算法层技巧(PTB/SWA/LAMB)可移植,而通信重叠、网络定制与稳健训练框架只有在自控集群时才能复刻;其不可复刻性(专有集群、专有模型、系统本体未开源)决定了它更像蓝图而非可引入的组件。

3 分钟速读

  1. 问题:万卡 LLM 训练有两大挑战——效率(模型切分后 GPU 间重度通信,MFU 直接决定训练速度)与稳定性(作业长达数周,失败与掉队是常态而非例外,且很多稳定性问题只在大规模出现)P(第 1 页 §1)。
  2. 方案:全栈协同设计:算法层(并行 transformer block、滑窗注意力、LAMB 批量 4×)+ 3D 并行通信-计算重叠(DP/PP/TP-SP)+ 高效算子(FlashAttention-2、融合 LayerNorm/GeLU)+ 数据管道(共享内存树状加载)+ 通信组初始化加速 + 定制网络(Tomahawk 4、multi-rail、Swift+DCQCN)+ 稳健训练框架与深度可观测 P(第 3-9 页 §3-§5)。
  3. 论文声称的主要结果:175B 模型、12,288 GPU:55.2% MFU、1.34× 于 Megatron-LM(迭代 6.34s、1,984k tokens/s、300B tokens 训练时间 1.75 天);消融显示各项叠加共 +17.6% MFU(256 GPU 口径);生产运行数周重启逾 100 次、>90% 故障自动修复、有效训练时间率 >90% P(第 1、10-12 页;条件见 #experiments)。
  4. 云上含义:I 吞吐来自「通信躲进计算」(TP/SP 融合、PP 解耦收发、DP 预取+优先级),可用性来自「心跳 → 诊断 → 驱逐补位 → 断点恢复」的自动化闭环;前者决定 GPU 租用成本,后者决定万卡集群的有效训练时间占比。
  5. 主要限制:系统本体与数据集未开源、专有模型细节未披露;GPU 仅称「NVIDIA Ampere」未给型号,部分结论(网络、容错)依赖自控集群前提;MFU 依赖峰值 FLOPs 假设口径;仓库侧仅论文脚注指向的 veScale 组件库 P(第 2、9 页)+ R(2026-09-15 核验)+ I

2. 问题背景

效率挑战:LLM 训练不是易并行的——模型切分到多卡后 GPU 间重度通信,通信之外的算子优化、数据预处理与显存占用同样显著影响 MFU(观测吞吐 ÷ 100% 峰值 FLOPs 假设下的理论吞吐)P(第 1 页 §1)。I 对架构师而言,MFU 是万卡集群「每元算力产出」的第一决定项,1 个百分点的 MFU 差异在数周作业里就是数万美元级。

稳定性挑战:LLM 训练周期以周计,失败与掉队是常态;失败恢复代价高、掉队拖慢整个作业;许多硬稳定性问题只在大规模出现,必须靠「深度可观测性」而非人工逐例排查 P(第 1-2 页 §1)。

并行策略底座(论文 §2 的背景知识):P(第 2-3 页 §2)

  1. 数据并行 + ZeRO:复制模型、均分数据;ZeRO 把优化器状态/梯度分片(第 2 阶段常用:分片两者且无额外通信开销),把 all-reduce 分解为 reduce-scatter + all-gather(图 1)。
  2. 流水线并行:按层切 stage、批切微批流水执行;Megatron-LM 的交错 1F1B 把每个 stage 细分为多个虚拟 stage(model chunk),warm-up → steady(1F1B)→ cool-down(图 2)。
  3. 张量/序列并行:TP 拆分计算密集算子(GEMM 需通信),LayerNorm/Dropout 等沿序列维 SP 分片省激活内存。
  4. 3D 组合规则:TP 通信最好限制在单节点内;DP 与 PP 更适合跨节点;优先把 DP 组建在 PP 之上,缓解跨 minipod 的 DP 通信。

相关工作站位:既有 LLM 技术报告(GPT-3/GPT-4/GShard/PaLM 与 OPT/BLOOM/Llama/Llama-2 等)多聚焦模型性能对比、略去系统基础设施细节,本文从系统视角补足超过 10,000 GPU 端到端预训练经验 P(第 12 页 §7)。

3. 核心机制

3.1 算法层优化(不损收敛前提下换效率)

3.2 3D 并行的通信-计算重叠

三个并行维度的通信与重叠手法(机制均为论文事实 P;「运维要点」列为编辑整理 I
维度关键通信重叠手法页码运维要点(I)
DP(ZeRO-2) forward 前参数 all-gather;backward 后梯度 reduce-scatter(图 1) 首个 all-gather 无法隐藏 → 受 PyTorch FSDP 启发在迭代开始预取、与数据加载重叠,通信时间按 1/(2·vpp_size) 因子缩短;高优先级通信先行(优先级=依赖该通信结果的计算算子顺序) P 第 4 页 §3.2 首个 AG/末个 RS 是结构性暴露段;vpp_size(虚拟 stage 数)越大暴露越小
PP(交错 1F1B) 阶段间点对点 send/receive warm-up 前向只依赖前一 receive → 解耦 send/receive(解除「被较慢一方阻塞」),send 与计算重叠;cool-down 反向应用同法;steady 阶段收发全部异步(图 4) P 第 4-5 页 §3.2 与图 4 send/recv 解耦是调度器级改动,复现时注意与框架自带实现的差异
TP/SP 输入 all-gather / 输出 reduce-scatter(在关键路径上) 与 FFN 路径并行 Linear 融合(FFN GEMM 大、隐藏更好);再把 GEMM kernel 切小块与通信流水执行;backward 同法;3D 下按 model chunk 粒度实施(图 3) P 第 4-5 页 §3.2 与图 3 通信融合进 kernel 是「需要改算子实现」的深水区,借库不如自研或跟上游

3.3 高效算子、数据管道与初始化

3.4 网络性能调优

控制面 / 数据面职责划分(编辑推断):I 论文未使用「控制面/数据面」术语。本页划分:控制面=稳健训练框架(Driver + 定制 Kubernetes + Executor 心跳 + 诊断/驱逐/补位 + 检查点恢复触发,§4);数据面=数据管道、微批计算与 3D 通信、优化器更新、检查点读写(§3)。该划分仅影响叙述,不改变论文事实。

4. GPU/系统数据路径

MegaScale 数据路径示意图:控制面(虚线)为稳健训练框架——Driver 与定制 Kubernetes 按节点分配 Executor(Pod),训练守护进程周期发送心跳,异常或心跳超时即暂停全集群、运行轻量自检诊断(主机内 RNIC 测试与 NCCL 测试)、屏蔽故障 IP 并等量补位,再从最新检查点恢复;数据面(实线)为一次训练迭代的关键路径:数据管道以每机单一 dataloader 写共享内存、预处理异步执行;启动期把 TCPStore 换为 Redis 并重排通信组初始化顺序,使 NCCL 初始化从 1047 秒降至 30 秒内;微批前向反向计算采用并行 transformer block(y = x + MLP(LN(x)) + Attention(LN(x)))、滑窗注意力、FlashAttention-2 与融合 LayerNorm/GeLU;TP/SP 把 all-gather 与 reduce-scatter 融合进 FFN 并行 Linear 并切块 GEMM 流水重叠,首个 all-gather 预取使通信时间按 1/(2·vpp_size) 缩短;PP 交错 1F1B 解耦 send/recv 异步重叠,LAMB 四倍批量把流水线气泡削减 87.5%;DP 梯度 reduce-scatter 与参数 all-gather 重叠且高优先级通信先行;LAMB 优化器更新后进入两阶段快速检查点——GPU 状态先写主机内存(数秒、训练不停),后台异步写 HDFS,恢复时组内单 worker 读取共享分区后广播;底部两条支撑带为定制数据中心网络(Tomahawk 4、三层 CLOS、1:1 无收敛、multi-rail、ECMP 抑制、Swift+DCQCN、adap_retrans)与可观测旁路(CUDA event 热力图约 0.5% 慢机、分布式 trace、3D 并行可视化、Kafka 实时入库)。
图 1:MegaScale 训练数据路径与关键路径(⑨控制面 + ①–⑧数据面步骤 + 两条支撑带)。实线=数据面关键路径,虚线=控制面与可观测旁路;编号对应下方文字序列。依据论文第 2-9 页图 1-图 8 与 §3-§5 绘制 P;平面划分、配色与编号为编辑标注(论文本身无此图)I

端到端文字序列(①–⑨;与图中编号一致)

  1. ① 数据管道(卸出关键路径):预处理异步执行、与步末梯度同步重叠;每机单一专用 dataloader 读入共享内存,各 GPU worker 拷贝所需数据到显存(两层树状、消除冗余磁盘读)P(第 5 页 §3.4)。
  2. ② 启动期(一次性):建立 NCCL 通信组——TCPStore 换 Redis(2,048 GPU 初始化 1047s→361s),再以组初始化顺序设计消除全局 barrier(O(n²)→O(n)),2,048 GPU <5s、>10,000 GPU <30s,为例行测试与快恢复扫清障碍 P(第 5-6 页 §3.5)。
  3. ③ 微批前向/反向计算(关键路径主干):PTB 并行块(y = x + MLP(LN(x)) + Attention(LN(x)))+ SWA 滑窗注意力 + FlashAttention-2 + 融合 LayerNorm/GeLU;微批按交错 1F1B 流经各 PP stage;激活驻留 HBM,通信操作交由 ④⑤⑥ 与计算重叠 P(第 3-5 页 §3.1/§3.3、第 3 页图 2)。
  4. ④ TP/SP 重叠:all-gather/reduce-scatter 与 FFN 并行 Linear 融合,GEMM 切小块与通信流水执行(backward 同法);按 model chunk 粒度;首个 all-gather 预取与数据加载重叠,通信时间按 1/(2·vpp_size) 因子缩短 P(第 4-5 页 §3.2 与图 3)。
  5. ⑤ PP 重叠 + 气泡压缩:交错 1F1B 的 send/receive 解耦异步发起,warm-up 只依赖前一 receive、steady 收发与计算全重叠、cool-down 反向应用同法;LAMB 把批量扩至 4×,流水线气泡减少 87.5% P(第 4 页 §3.1/§3.2)。
  6. ⑥ DP 梯度同步重叠:ZeRO-2 语义下梯度 reduce-scatter 聚合、参数 all-gather 取回;高优先级通信先行(优先级=依赖该通信结果的计算顺序);这些集合通信由底部「通信基座」承载(Tomahawk 4 三层 CLOS、1:1 无收敛、multi-rail、Swift+DCQCN)P(第 4 页 §3.2、第 6 页 §3.6、第 2-3 页图 1)。
  7. ⑦ 优化器更新:LAMB 以 4× 批量更新(微基准:13B 模型、约 250B tokens 后与 ADAM 1× 批量损失一致,随后在生产开启)P(第 4 页 §3.1、第 11 页图 10b)。
  8. ⑧ 两阶段快速检查点:阶段一 GPU 片上状态写主机内存(优化序列化 + pinned memory,数秒、训练不停);阶段二后台进程异步转存 HDFS;恢复时利用 DP 组共享分区:组内单 worker 读取后广播,线性缓解 HDFS 带宽瓶颈 P(第 7-8 页 §4.4)。
  9. ⑨ 控制面(旁路常驻):Driver + 定制 Kubernetes 按节点起 Executor(Pod);训练守护进程周期发心跳(携带状态、日志、RDMA 指标);异常/心跳超时 → 暂停全集群 → 轻量自检诊断(主机内 RNIC 回环/全互联、节点内 all-to-all、同 ToR 跨机 all-reduce)→ 屏蔽 IP、驱逐并等量补位 → 自最新检查点恢复;可观测旁路提供 CUDA event 热力图(约 0.5% 慢机)、分布式 trace 与 3D 并行可视化(超时定位故障节点)P(第 6-9 页 §4-§5 与图 5-图 8)。
路径上的关键数字与容量事实(口径非性能结论;性能结论见 #experiments
事实/数值对象与条件论文定位
LAMB 气泡:(4/v)·(p−1)/m → (1/v)·(p−1)/(4m),即减少 87.5%(批量 1×、4 步 → 4×、1 步)交错流水线,p=阶段数、v=虚拟 stage 数、m=微批数P 第 4 页 §3.1
首个 all-gather 预取:通信时间按 1/(2·vpp_size) 因子缩短;高优先级通信先行DP all-gather/reduce-scatter 重叠P 第 4 页 §3.2
NCCL 通信组初始化:1047s(Megatron-LM/TCPStore)→ 361s(Redis)→ <5s(2,048 GPU)/ <30s(>10,000 GPU);barrier O(n²)→O(n)2,048 NVIDIA Ampere GPU 起测(与 §6 同集群类型)P 第 5-6 页 §3.5
网络:Tomahawk 4 单芯片 25.6 Tbps、64×400G;三层 CLOS >10,000 GPU;下行:上行 1:1(32+32);400G→2×200G(AOC);8×200G NIC multi-rail;同组 ToR ≤64 台服务器自建数据中心网络P 第 6 页 §3.6
检查点:阶段一写主机内存降至数秒(pinned + 优化序列化,高 PCIe 带宽);阶段二后台异步写 HDFS;恢复按 DP 组单读多播自建部署(分布式文件系统为 HDFS)P 第 7-8 页 §4.4
诊断套件:主机内 RNIC 回环 + full-mesh;同主机 RNIC 互测;节点内 NCCL all-to-all;同 ToR 跨机 all-reduce轻量自检(时间与准确率折中)P 第 7 页 §4.3
监控:心跳含 IP/Pod/硬件/进程状态/stdout/stderr/RDMA 指标;毫秒级精度;流量完全停止自动触发恢复生产常开、开销可忽略P 第 7 页 §4.2、第 8 页 §5.1
P 口径提醒:论文网络与容错机制描述均以「自建数据中心 + 自研 Driver/Kubernetes + HDFS」为前提(第 6-8 页)。把这些数字(如初始化 <30s、诊断 <10 分钟)外推到共享集群或托管服务时,前提不再成立——引用时必须连带前提条件。

5. 架构权衡

6. 云上部署映射

以下映射为厂商中立示例;具体产品命名/规格仅作说明并标 E,以厂商当时目录为准,未逐一在线核验。论文事实单独标注 P

MegaScale 需求/机制 → 云上资源映射
论文需求/机制云上映射(示例)证据与说明
计算:大规模同代 GPU(论文口径 NVIDIA Ampere,>10,000 卡;每服务器 8×200G NIC) E 8×GPU 高带宽整机实例(各云 H100/A100 档 GPU 主机),同代同型整池;节点内 NVSwitch、节点间高带宽 RDMA GPU 代际与 NIC 配置为论文口径 P(第 6、9 页;具体 Ampere 型号论文未明确披露);实例命名为云目录示例 E
网络:三层 CLOS、1:1 无收敛、multi-rail(8×200G/服务器)、定制拥塞控制(Swift+DCQCN 融合) E RoCEv2/InfiniBand 后端网络、非收敛比(1:1)组网档、拓扑感知放置(TP 同机、DP/PP 跨机);拥塞控制用云厂商 ECN/PFC 配置近似 组网原则为论文事实 P(第 6 页 §3.6);自研拥塞控制在共享云环境通常不可复刻,以厂商配置近似 E+I
存储:检查点两阶段写(主机内存数秒 → 异步分布式文件系统 HDFS);恢复组内单读多播 E 高吞吐对象存储/并行文件系统(Lustre/GPFS 类)承载分片检查点;节点本地内存作一阶缓冲 两阶段机制与「单 worker 读 + 广播」为论文事实 P(第 7-8 页 §4.4);HDFS→云存储的吞吐换算需按目录压测 E/I
编排:定制 Kubernetes + Driver + 每节点 Executor(Pod)+ 心跳;故障节点屏蔽 IP、驱逐、等量补位 E Kubernetes + 自定义控制器/训练作业编排、节点污点与驱逐、自动换补;或托管批调度服务 控制面机制为论文事实 P(第 6 页 §4.1、图 5);云上控制器实现为工程示例 E
可观测:心跳聚合日志 + RDMA 指标 + 毫秒级监控;CUDA event 热力图/trace/3D 可视化;Kafka→分析库 E 日志/指标接入云监控(Prometheus/Grafana、托管 Kafka);GPU 事件计时自研采集 + 仪表盘 监控形态为论文事实 P(第 7-9 页 §4.2/§5);云产品接入为实践示例 E
弹性:从检查点跨节点拓扑恢复(驱逐/补位后继续训练);训练中动态换入健康节点 E 节点池维护 + 检查点续训(set 自动换卡重调度);把「换入节点需过诊断」固化为准入钩子 恢复语义为论文事实 P(第 6-8 页 §4);云上准入钩子为工程建议 I
放置要点(编辑推断):I 依论文组网:TP 组必须同机(节点内高速互联),DP 组优先于 PP 组跨 minipod 铺开(第 3 页 §2);数据密集节点尽量同置一个 ToR 域以减少 ECMP 冲突(第 6 页)。若云上只能拿到收敛比组网,优先保 TP 同机与 PP 邻机架,DP 慢一点影响最小(其通信已大幅重叠)。

7. 成本 / 性能 / SLO

7.1 训练场景的「SLO」口径

预训练没有在线时延 SLO,平台 SLA 通常转化为:I吞吐目标(tokens/s 或 MFU);②有效训练时间占比;③故障 MTTR;④收敛里程碑。论文对四者均给出参考:P 有效训练时间率 >90%(迭代数×迭代训练时长 ÷ 总训练时长,第 12 页 §6.3);故障检测+诊断平均 <10 分钟、崩溃后 15 分钟内自最新检查点追平(第 12 页);吞吐/MFU 见 #experiments;收敛里程碑对应生产运行损失持续收敛(第 11 页图 11)。

7.2 成本公式(参数化,编辑推断)

预计 GPU 时(小时)≈ 总 token 数 ÷ (单作业吞吐 tokens/s × 3600)(吞吐带完整条件,见表 2 口径)
算力成本 ≈ GPU 时 × 当时单价(记录查询日期)
MFU 杠杆:吞吐 ×(MFU_new/MFU_old) → 同规模作业 GPU 时等比例缩短
有效成本/token ≈ 算力成本 ÷ 总 token 数 ÷ 有效训练时间占比

I 公式中由论文支撑的变量是吞吐与 MFU(带完整条件,见 #experiments)与有效训练时间率(>90%,第 12 页);单价、能耗与存储价论文未披露,本页不给伪精确数字。

7.3 敏感项排序(论文口径,条件互不可比)

P 256 GPU 消融口径下,各项叠加对 MFU 的贡献:通信重叠(TP/PP/DP)+6.2%(合并披露)> 算法(PTB+SWA)+5.6% > LAMB 批量 256→768 +3.0% > 高效算子 +1.7% > 数据管道与问题代码消除 +1.1%;合计 +17.6%(第 10 页 §6.1、第 11 页表 3)。I 成本治理含义:先拿「不改算子实现」的算法与 LAMB 项,通信重叠与网络定制收益更大但工程更深;万卡规模上稳定性项(>90% 自动修复、15 分钟追平)对有效产出的影响可能大于最后一个 MFU 点。

7.4 数据缺口

8. 安全与可运维性

8.1 安全

8.2 可运维性

9. 适用 / 不适用场景

适用(触发条件 + 理由)

  1. 数千至数万卡自建/专有集群上的 LLM 预训练。触发:作业以周计、MFU 与稳定性同时是主要矛盾。理由:论文全部机制与数字均在该场景验证(175B/12,288 GPU 55.2% MFU;生产运行 >90% 有效训练时间率)P(第 1、10-12 页)。
  2. 可自控网络与编排的环境(自建数据中心或专属托管)。触发:能定制 CLOS 组网、拥塞控制与 Kubernetes 控制器。理由:通信重叠收益与容错闭环都建立在自控前提上 P(第 6 页 §3.6、第 6-8 页 §4)。
  3. 以 3D 并行为底座、基于 Megatron-LM 生态演进的平台。触发:已在用 Megatron-LM(论文用其 commit 285068c8 做基线)。理由:MegaScale 构建于 Megatron-LM 之上,各优化项可直接对照消融表排优先级 P(第 9-11 页)。
  4. 万卡作业的故障/掉队治理体系建设。触发:失败与掉队频发、人工排障不可扩展。理由:心跳→诊断→驱逐→恢复闭环 + 热力图/trace/3D 可视化是成套方法论 P(第 6-9 页 §4-§5)。
  5. 研究「生产级 LLM 训练系统该怎么设计」的团队。触发:需要系统视角的参考架构与踩坑清单。理由:论文价值正在于公开了常规技术报告省略的系统细节 P(第 12 页 §7)+ I

不适用(触发条件 + 理由)

  1. 期望「引入即用」的开源训练框架。触发:想 pip install 一个 MegaScale。理由:系统本体未开源,论文脚注指向的 veScale 是组件库(Apache-2.0,进行中),不是 MegaScale 官方完整实现 P(第 2 页脚注 3)+ R(2026-09-15 核验)。
  2. 数百卡以下的小规模训练。触发:单机或数机作业。理由:万卡特化机制(通信组初始化、multi-rail、稳健训练框架)的收益在小规模不成比例;论文最小实验口径也是 256 GPU P(第 10 页)+ I
  3. 共享/托管环境想完整复刻网络前提。触发:无法控制收敛比、拥塞控制与机架放置。理由:1:1 无收敛 CLOS、multi-rail、自研拥塞控制是吞吐与稳定性的前提,租用环境不可复制 P(第 6 页 §3.6)+ I
  4. 推理服务或微调为主的工作负载。触发:拿本文数字预估 serving/微调收益。理由:论文范畴是预训练系统,不含服务化组件与微调实验 P(第 1 页 §1)+ I
  5. 要求全部组件可复现审计的合规环境。触发:需要逐组件源码审计。理由:专有模型、数据集与部分自研组件(Driver、诊断、拥塞控制算法)细节未公开 P(第 6-11 页)+ I

10. 实验与指标

10.1 实验设置与口径 P(第 9 页 §6.1)

设置与指标口径(论文事实)
基线与框架MegaScale 构建于 Megatron-LM 之上;对比基线为 Megatron-LM 仓库 commit 285068c8(实验开始数月前选定);MegaScale 与 Megatron-LM 使用相同批量(公平对比)
模型175B:hidden 12,288 / heads 128 / layers 96 / TP 8 / PP 8;530B:hidden 20,480 / heads 160 / layers 105 / TP 8 / PP 35(表 1);交错流水线交错数:175B 为 6、530B 为 3;序列长 2,048、词表 64,000
硬件NVIDIA Ampere GPU(生产集群口径,截至 2023 年 9 月最大集群 >10,000 卡);具体型号/显存/峰值 FLOPs:论文未明确披露
口径MFU = 观测吞吐 ÷ 100% 峰值 FLOPs 假设下的理论吞吐(第 1 页引 [7]);表 2 训练时间按 300B tokens 计;消融在 256 GPU/批量 256 进行且网络优化双方均开启(第 10 页)
收敛验证微基准受资源所限用 13B 模型:PTB+SWA 在 >100B tokens 后损失与基线相当;LAMB 4× 批量在约 250B tokens 后与 ADAM 1× 损失一致(第 10-11 页图 10);生产运行为专有模型、数万亿(multi-trillion)tokens(第 11 页图 11)

10.2 强扩展主结果(论文表 2 全量转录)

175B 强扩展训练性能(P,第 10 页表 2;批量 6144 用于 3072–12288 GPU、768 用于 256–1024 GPU(GPU 内存所限);训练时间按 300B tokens;MFU 列括号内为 MegaScale 对 Megatron-LM 加速比)
批量方法GPUs迭代时间 (s)吞吐 (tokens/s)训练时间 (天)MFU聚合 PFlops/s
768Megatron-LM25640.039.3k88.3553.0%43.3
51221.274.1k46.8649.9%77.6
76815.2103.8k33.4546.7%111.9
102411.9132.7k26.1744.7%131.9
MegaScale25632.049.0k70.8665.3% (1.23×)52.2
51216.595.1k36.5163.5% (1.27×)101.4
76811.5136.7k25.4061.3% (1.31×)146.9
10248.9176.9k19.6259.0% (1.32×)188.5
6144Megatron-LM307229.02433.6k8.0148.7%466.8
614414.78851.6k4.0847.8%916.3
819212.241027.9k3.3843.3%1106.7
122888.571466.8k2.3741.2%1579.5
MegaScale307223.66531.9k6.5359.1% (1.21×)566.5
614412.211030.9k3.3757.3% (1.19×)1098.4
81929.561315.6k2.6454.9% (1.26×)1400.6
122886.341984.0k1.7555.2% (1.34×)2166.3

P 正文口径(第 10 页 §6.1):全部设置加速 1.19×–1.34×;GPU 增多时 MegaScale MFU 从 59.1% 降至 55.2%(批量固定、计算通信比下降);12,288 GPU 处原文表述为领先 Megatron-LM「by 14% MFU」。I 编辑注:55.2% − 41.2% = 14.0 个百分点,与 1.34× 一致;「14%」应按百分点差理解,不应再乘算。

10.3 弱扩展与消融

弱扩展(图 9,530B,批量随 GPU 等比放大;MFU 读图)与消融(表 3,175B、256 GPU、批量 256;P,第 10-11 页)
实验2240 GPU / 基线4480 GPU / +PTB11200 GPU / +SWA说明
图 9:Megatron-LM MFU49.20%48.80%48.20%正文称随掉队与通信下降 1.6%(第 10 页)I(柱标 49.20→48.20 点差 1.0,换算口径论文未说明,照录不调和)
图 9:MegaScale MFU54.30%54.10%54.30%领先至多 6.1%;3D 通信重叠带来近线性扩展(第 9-10 页)
消融:MFU 分解(P,第 11 页表 3;175B、256 GPU、批量 256;基线 Megatron-LM 47.7%,网络优化双方均开启;Δ 为相对基线累计)
Idx叠加项MFUΔMFU备注(编辑整理 I
1baseline47.7%原版 Megatron-LM
2(1) with PTB52.3%+4.6%算法:并行 transformer block
3(2) with SWA53.3%+5.6%算法累计口径(正文「两项算法 +5.6%」)
4(3) with TP overlap55.5%+7.8%通信重叠合计 +6.2%(正文口径,第 10 页)
5(4) with PP overlap58.0%+10.3%
6(5) with DP overlap59.5%+11.8%
7(6) with efficient operators61.2%+13.5%FlashAttention-2 + 融合 LayerNorm/GeLU
8(7) with misc optimizations62.3%+14.6%数据管道优化 + §6.3 问题代码消除
9(8) with LAMB (BS×3)65.3%+17.6%批量 256→768(LAMB 扩批量)

10.4 稳定性与生产运行

稳定性数字与生产运行(P,第 8、11-12 页)
指标/事件论文数字与口径定位
掉队机器占比约 0.5%热力图识别显著慢机;剔除后各次运行峰值 MFU 一致(第 8 页图 6/图 7)
慢主机影响前向约 +10% 耗时 → 移除后 MFU +0.7%计算掉队案例(第 12 页 §6.3)
生产运行>10,000 GPU、数周、重启 >100 次专有模型、数千亿级参数(hundreds of billions)、数万亿(multi-trillion)tokens;损失持续收敛(第 11 页图 11)
故障自动修复率>90%其余经排障工具处理;异常多为 CUDA error、段错误(第 11-12 页 §6.3)
MTTR 分解检测+诊断 <10 分钟;崩溃后 <15 分钟追平自最新检查点续训(第 12 页 §6.3)
有效训练时间率>90%=迭代数×迭代训练时长 ÷ 总训练时长(第 12 页 §6.3)
公平性限制(引用前必读):P ①MFU 依赖「100% 峰值 FLOPs」假设,而论文未披露 GPU 具体型号与峰值算力(第 1、9 页),MFU 不可跨集群复算;②表 2 各行批量不同(768/6144),且 GPU 数与 3D 并行配置在 175B 与 530B 实验间不同(12,288 vs 11,200,第 10 页);③消融仅在 256 GPU/批量 256 进行,网络优化双方均开启,各项增益不可外推到万卡(第 10 页);④收敛微基准用 13B 模型(资源所限),非 175B/530B 本体(第 11 页);⑤生产运行的模型/数据为专有、未披露细节(第 11 页)。I 各项百分比分属不同规模/基线,禁止跨行相加或与站内其他论文数字直接比较。

10.5 建议复现/验证步骤(编辑推断)

  1. I 以 Megatron-LM 为底座复现基线:pin 论文同源 commit 谱系 + 同代 GPU,先在 256 GPU 验证 47.7% MFU 量级(表 3 基线口径,含网络优化)。
  2. I 按消融表顺序逐项叠加(PTB → SWA → TP/PP/DP 重叠 → 算子 → 数据管道 → LAMB),每项跑 13B 级收敛对照后再上大规模(对齐论文微基准口径,第 10-11 页)。
  3. I 专项验证通信组初始化:>2,000 卡集群测启动时间,确认 Redis 替换与初始化顺序收益(目标 5 秒/30 秒量级,第 5-6 页)。
  4. I 演练容错闭环:注入进程级故障与慢机,验证心跳检出、诊断定位、驱逐补位与 15 分钟内追平(对齐第 12 页口径,验收阈值自定)。
  5. I 固化有效训练时间率与 MTTR 到平台报表;把热力图/trace 工具常开并验证开销可忽略(对齐第 8-9 页口径)。

12. 给架构师的决策清单

I 以下为落地前的勾选项;标注(P)的条目对应论文证据,(R)对应仓库核验,其余为工程判断。

13. 证据台账

下表为核心结论的证据映射;逐条引文与在线核验记录见构建文件 sources/megascale.evidence.json。核验日期均为

关键结论 → 证据级别 → 定位 → 核验状态
关键结论级别定位核验状态
题名/32 位作者/机构(ByteDance、Peking University)/arXiv v1(cs.LG,2024-02-23 水印);正文未标注录用会议(致谢提及 NSDI 审稿人)P第 1、13 页已核验(PDF 原页)
摘要级结论:>10,000 GPU 生产系统;175B/12,288 GPU 55.2% MFU、1.34× 于 Megatron-LM;全栈协同设计 + 容错与可观测P第 1 页摘要(另见第 10、13 页)已核验(PDF 原页;条件见 EV-21/22/32)
两大挑战(效率/稳定性)与两条原则(算法-系统协同设计、深度可观测);系统本体未开源、脚注指向 veScaleP第 1-2 页 §1 与脚注 3已核验(PDF 原页)
并行底座:ZeRO-2、交错 1F1B、TP/SP、3D 组合规则(TP 限节点内、DP 组优先于 PP)P第 2-3 页 §2 与图 1/图 2已核验(PDF 原页)
算法层:PTB 公式改写、SWA O(s×w)、LAMB 批量 4× 与气泡 (4/v)(p−1)/m → (1/v)(p−1)/(4m)、−87.5%P第 3-4 页 §3.1已核验(PDF 原页 + 300dpi 放大核对分数式)
通信重叠:DP 预取(1/(2·vpp_size))与优先级、PP 解耦 send/receive、TP/SP 融合 Linear + GEMM 切块(图 3/图 4)P第 4-5 页 §3.2已核验(PDF 原页)
算子(FlashAttention-2、融合 LayerNorm/GeLU)与数据管道(异步预处理、共享内存树状加载)P第 5 页 §3.3/§3.4已核验(PDF 原页)
初始化加速:1047s → 361s(Redis)→ <5s/<30s;barrier O(n²)→O(n)P第 5-6 页 §3.5已核验(PDF 原页)
网络:Tomahawk 4(25.6 Tbps、64×400G)、三层 CLOS 1:1、400G→2×200G、8×200G multi-rail、ECMP/Swift+DCQCN/adap_retransP第 6 页 §3.6已核验(PDF 原页)
容错:Driver+定制 K8s+Executor 心跳(图 5)、诊断套件(RNIC/NCCL)、两阶段检查点(数秒写主机内存 + 异步 HDFS、单读多播恢复)P第 6-8 页 §4 与图 5已核验(PDF 原页)
排障:CUDA event 热力图(约 0.5% 慢机)、trace 时间线、3D 可视化超时定位;Kafka 实时入库(图 6-图 8)P第 8-9 页 §5已核验(PDF 原页)
部署规模:截至 2023-09 生产最大集群 >10,000 NVIDIA Ampere GPU;Hopper 集群在建P第 2、9 页已核验(PDF 原页;具体型号未披露)
实验设置:Megatron-LM commit 285068c8、175B/530B 配置(表 1)、seq 2048/vocab 64000、交错数 6/3P第 9 页 §6.1 与表 1已核验(PDF 原页)
表 2 全量 16 行(强扩展 175B:迭代时间/吞吐/训练时间/MFU/聚合 PFlops/s);图 9 弱扩展读数;正文 59.1%→55.2% 与「by 14% MFU」口径P第 9-10 页 §6.1、图 9、表 2已核验(PDF 原页逐格转录;14% 按百分点理解,编辑注 I)
表 3 消融 9 行(47.7% → 65.3%,合计 +17.6%)与正文分解(算法 5.6%/通信 6.2%/算子 1.7%/其他 1.1%/LAMB 3.0%)P第 10 页 §6.1、第 11 页表 3已核验(PDF 原页逐格转录)
收敛与生产运行:13B 微基准(>100B/约 250B tokens 口径)、生产运行 >100 次重启、>90% 自动修复、<10 分钟/<15 分钟、有效训练时间率 >90%(图 10/图 11/图 12)P第 10-12 页 §6.2/§6.3已核验(PDF 原页)
veScale 可达性/归属(volcengine)/Apache-2.0/自述/最近提交(2026-03-03);Megatron-LM 与 arXiv 页可达R论文第 2 页脚注 3、第 9 页 §6.1 给出的 URL已核验(2026-09-15 在线核验;veScale 非 MegaScale 本体,见 EV-33)
云实例/网络/存储/编排/监控映射示例(高带宽 GPU 主机、RoCE/IB 非收敛组网、并行文件/对象存储、K8s 控制器、托管 Kafka/监控)E各云厂商公开目录未逐一在线核验;使用前按当时目录复核
成本公式、SLO 框架、放置与弹性建议、安全与回滚建议、复现步骤、控制面/数据面划分、图文口径编辑注(图 9 的 1.6%、表 2 的 14%)I本报告 §4-§10、§12;EV-20/21/36编辑标注完成;不含伪精确数字