跳到正文
原文
OpenBMB· @OpenBMB · X·· 3 小时前AI 评分52

面壁智能 MiniCPM-V 4.7 用 LES 加速端侧 MoE 推理

AI 导读

面壁智能 OpenBMB 介绍 MiniCPM-V 4.7 采用的 Lookahead Expert Scheduling(LES),用于解决 35B MoE 模型在有限内存设备上按需从 SSD 加载专家权重导致生成卡顿的问题。

正文 · 原文

⚡ Lookahead Expert Scheduling: Making On-Device MoE Inference Faster

Running a 35B MoE model on a device with limited memory presents a key challenge: expert weights can be stored on SSD, but loading them on demand can stall generation.

MiniCPM-V 4.7 addresses this with LES, which coordinates expert loading and computation rather than waiting for every required expert to load before continuing.

🔹 Expert prefetch and compute scheduling: LES predicts which experts the next layer may need through cross-layer routing prediction. It prioritizes computation for experts already loaded and runs their computation while loading the remaining required experts in parallel. Predictions guide preloading and scheduling only; the original token-level router still determines the experts used for actual computation. No additional training is required.

🔹 Optimized weight access: Fine-grained, layer-wise mixed-precision quantization reduces weight storage and transfer overhead while preserving model capability. Reorganizing expert weights into contiguous layouts also reduces scattered reads and improves sequential flash-storage access.

Lookahead Expert Scheduling: github.com/tc-mb/llama.cpp-o…

来源:OpenBMB · x.com