arXiv:2607.09686cs.LG2026-07

让本地MoE模型在内存受限设备上运行,通过按需加载专家实现高效推理。

MawForge: Memory-Bounded Expert Materialization for Local Mixture-of-Experts Inference

  • 将完整模型存于磁盘,仅常驻高频专家和关键数据,按需加载其余部分。
  • 在受限内存设备上实现稳定推理,性能受专家复用率与缓存大小共同影响。
  • 适合资源有限环境下的MoE模型部署,尤其对macOS用户有实用价值。

稀疏混合专家(MoE)语言模型将总参数量与每令牌活跃计算分离,但本地推理系统通常仍需将完整模型、键值缓存、运行时缓冲区及操作系统预留空间全部放入快速内存。MawForge提出一种新系统假设:通过将完整模型存储在磁盘上,仅保持常用张量驻留内存,并按需将路由的专家张量材料化至有界执行缓存,可在统一内存受限设备上实现可行的本地MoE服务。核心发现是,MawForge作为有界执行机制和性能测量基线有效,但不适合作为最大化缓存的策略。性能取决于专家复用率、驻留占用、KV缓存大小、量化程度、路由局部性以及macOS内存压力之间的权衡。

原文摘要 · Abstract (English)

Sparse Mixture-of-Experts (MoE) language models separate total parameter count from per-token active computation, but local inference systems often still require the full model, key-value cache, runtime buffers, and operatingsystem headroom to fit in fast memory. MawForge tests a different systems hypothesis: local MoE serving can be made practical on constrained unified-memory machines by storing the full model on disk, keeping common tensors resident, and materializing routed expert tensors into a bounded execution cache on demand. The central finding is that MawForge is effective as a bounded execution mechanism and measurement substrate for local MoE inference, but not as a cache-maximization policy. Performance depends on balancing expert reuse against resident footprint, KV-cache size, quantization, route locality, and macOS memory pressure.

MoE推理内存优化本地部署缓存机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。