Edge0:35B 的 MoE 塞进 3 GiB 内存,在 Mac mini 上跑到 15 tok/s
4-bit 权重留在 SSD 上按需读,一个提前一步预测路由的小头把读盘藏进前向,常驻内存压到 2.9 GiB。
展开
MoE 推理在消费级硬件上卡住的从来不是算力而是字节数。稀疏激活省的是每 token 的计算量,要搬进内存的权重一点没少,35B 模型哪怕 4-bit 也有 19.5 GB。直接把专家放到 SSD 又救不了场,因为第 N+1 层要读哪些专家,得等第 N 层算完才知道,读盘根本来不及提前发起。
Edge0 的破法是训一个 prerouter,提前一个 token 预测下一层的路由,而且预测结果直接当成路由用,所以调进来的专家集合就等于真正被路由到的集合,不存在猜错要回退、要丢弃。int4 掉的精度用一个不合并的 recover-LoRA 蒸馏补回来,五个公开基准平均只比 fp16 base 低 3.9 分(AIME 2026 86.6 对 92.7,HumanEval 90.9 对 95.1)。
实测机器是 24 GB 的 Mac mini M4 Pro,解码 14.9 到 17.7 tok/s,预填充 140 tok/s,峰值活跃内存 2.9 GiB。后端目前只有 MLX,也就是只支持 Apple Silicon;作者自己标明这是 preview,agent 能力(工具调用、多步规划)还很弱。方法那一篇在 arXiv:2609.18063。