返回关卡地图
闯关题:0 / 3 XP 0 / 300

Level 47 | 零基础导学关卡

MoE 专家并行

MoE Expert Parallel

MoE 专家像不同专科窗口:Router 决定 token 去哪里,专家并行再把专家放到不同设备。窗口数量多并不能保证任务分得均匀。

本课官方 Notebook ↗
47_MoE_Expert_Parallel.ipynb 并行通信MoEExpert ParallelAll-to-All
Mission 1 01 / 建立直觉

四个窗口都开着,为什么还是有人排长队?

图解原理

MoE 专家像不同专科窗口:Router 决定 token 去哪里,专家并行再把专家放到不同设备。窗口数量多并不能保证任务分得均匀。

先认识这三个词

expert
处理部分 token 的子网络。
dispatch
把 token 按路由结果发往对应专家。
all-to-all
多个设备彼此交换各自目的地的数据。
试着说给朋友听

先不看公式:用上面的生活场景,说一说这节课想减少哪种浪费、需要付出什么代价。

闯关题

本课中的「expert」指什么?

学完这一段,试着做

用一个小动作确认自己理解了;最后再进入官方题目。

Mission 2 02 / 操作与推演

把一个小例子算到最后

图解原理

先用默认数值手算,再只改一个参数。让结果来检验你的猜想。

  1. 先数路由

    8 个 token 分给两个专家,负载是 6 和 2。

  2. 找关键路径

    假设两个专家同速,完成时间由较忙的 6 决定;总工作量/设备数=4 是理想均分值。

  3. 看通信

    如果每个 token 都需要跨设备发送,还会增加传输和等待。必须一起记录负载和通信量。

动手实验室 / 只在浏览器中演示

把八份工作分到两个窗口

改变任务分配,观察最忙窗口怎样决定完成时间。

教学简化模型:所有数值来自上方规则,不是 GPU 性能实测。播放可以暂停,键盘方向键可调整滑块。

闯关题

两个同速专家处理 6 与 2 份工作,理想并行完成时间由谁决定?

学完这一段,试着做

用一个小动作确认自己理解了;最后再进入官方题目。

Mission 3 03 / 纠错与迁移

从会看,走到会写与会判断

图解原理

下面是一段独立的小练习。它把计算关系写清楚,帮助你进入官方题目;不是整份作业的答案。

读懂这段最小 Python

loads = [6, 2]
ideal = sum(loads) / len(loads)
efficiency = sum(loads) / (max(loads) * len(loads))
print(ideal, round(efficiency,2))  # 4, 0.67

先找输入变量,再找中间量,最后核对注释里的输出。改一个输入,手算后再运行。

最容易踩的坑

把 expert 数当作有效并行度,或只看平均负载而忽略最忙设备。

去官方题目做什么

  1. 到官方 Notebook 阅读题目函数与测试;先写出输入、输出和一个最小例子。
  2. 把本课手算过程转换成实现,先跑最小测试,再检查空输入、边界值或未达门槛的情况。
  3. 记录一个与预期不同的结果,并用本课术语说明原因。

闯关题

同学提出下面的做法,哪一项会导致本课讨论的误判?

学完这一段,试着做

用一个小动作确认自己理解了;最后再进入官方题目。

把理解变成自己的代码

准备好,去官方题目试一试

在本页用小例子建立直觉,再去官方 Notebook 完成实现。先运行你自己的测试,遇到困难时再查看官方提示与参考答案。

前往本课官方 Notebook ↗

这节课会遇到的代码对象

MoEParallelSummary · summarize_expert_parallel · recommend_moe_parallel_plan · compare_dense_vs_moe_cost

先找题目里的输入、输出与 TODO,再把本课的手算过程对应进去;以官方题目中的函数说明和测试为准。

练习来源:Datawhale 官方仓库 · 518cc45。这里的入口直接打开官方版本,不读取或分享你的本地 Notebook。

完成 3 道闯关题后,本关即算完成;作业 checklist 用来辅助你回 notebook 练习。
进度只保存在当前浏览器 localStorage,分享 HTML 不会带走你的记录。
上一关:L46 下一关:L48