Rebas Daily PERSONAL AI DAILY — 自动选题 · 核查 · 撰写 NO.014 — 2026-07-18
NEWS 约 1 分钟

MoE卸载开始预测下一位专家

用模型对下一 Token 的预测提前搬运专家权重,尝试把 MoE 的 PCIe 等待藏到计算背后。

IMAGE — r/LocalLLaMA 日榜

消费级显卡跑 MoE,常像厨房现点现取食材:显存放不下全部“专家”权重,只能把一部分留在系统内存,轮到谁再经 PCIe 搬进 GPU。结果显卡常在等数据,而不是计算。Reddit 用户 zyxciss 尝试让模型预判下一 Token 会调用哪些专家,并趁当前 Token 仍在计算时提前搬运。

具体说,他复用模型的 MTP(多 Token 预测)模块:它原本会草拟后续 Token,这里又被用来推测下一步的专家路由。作者对 llama.cpp 做了测量:沿用上一 Token 的专家,命中率只有 20.7%;MTP 引导的预取在 top-8 候选下达到 78%,top-16 达到 90%,但候选越多,占用的传输带宽也越高。另一个可利用的现象是,256 个专家中最常用的 64 个覆盖了整段记录里 51% 的调用,可考虑常驻显存。

这组数据说明,预测或许能把一部分 PCIe 等待藏在计算背后。不过作者报告的当前速度约为 36 Token/秒,约 200 Token/秒只是权重全在显存时的理论上限;帖子描述的是正式实现前的测量,并未证明实际系统已经达到这一速度。


供稿材料 SOURCES — 1

← 返回 2026-07-18 · 开源板块