FluxMoE: Decoupling Expert Residency for High-Performance MoE Serving
针对 MoE 模型推理中大量 expert 权重闲置占用 GPU 显存、与 KV Cache 竞争内存的问题,提出 FluxMoE 系统。核心思想是将 expert 权重从 GPU 常驻内存中解耦,引入 expert paging 抽象,按需加载 expert 并在使用后立即驱逐,将 GPU 内存优先分配给 KV Cache 等性能关键状态。系统基于 vLLM 实现,在严重内存受限场景下显著提升 MoE 模型的服务吞吐。
INSIGHTFluxMoE 提出的 expert paging 抽象将 MoE 推理的内存管理从静态驻留转向动态流式,类似操作系统的虚拟内存管理思想。这一设计洞察对搜索推荐系统部署 MoE 排序模型具有直接参考价值:在 GPU 显存受限的线上服务环境中,通过按需加载 expert 权重可以腾出更多显存给 KV Cache,从而提升并发请求处理能力和整体服务吞吐。