arXiv:2601.17063cs.LGcs.AI2026-01被引 1

用机器学习优化缓存,让边缘设备跑超大稀疏模型更快更省I/O。

FlashMoE: Reducing SSD I/O Bottlenecks via ML-Based Cache Replacement for Mixture-of-Experts Inference on Edge Devices

  • 用机器学习融合访问频次和时间,动态优化专家模型缓存策略。
  • 实测缓存命中率提升51%,推理速度比现有系统快2.6倍。
  • 适合资源受限的边缘设备部署超大模型,实测验证可行性。

近年来,混合专家(Mixture-of-Experts, MoE)模型因其高效扩展大语言模型的能力而受到关注。尽管模型极其庞大,但其稀疏激活特性使得推理时仅需访问部分模型,从而为边缘设备上的本地推理提供了可能。然而,此前的MoE推理系统如Fiddler或DAOP依赖于基于DRAM的卸载机制,在内存受限的设备上不适用。随着最新MoE模型规模突破数百吉字节,基于内存的卸载方案已不可行。为此,本文提出FlashMoE,将非活跃专家卸载至SSD,并引入轻量级机器学习驱动的缓存替换策略,通过自适应结合访问频率与时间信号以最大化专家复用,显著降低存储I/O开销。此外,我们构建了面向用户的桌面平台以验证其实用性。在真实硬件环境下,FlashMoE相比经典的LRU、LFU等策略,缓存命中率最高提升51%,推理速度相较现有系统最快提升2.6倍。

原文摘要 · Abstract (English)

Recently, Mixture-of-Experts (MoE) models have gained attention for efficiently scaling large language models. Although these models are extremely large, their sparse activation enables inference to be performed by accessing only a fraction of the model at a time. This property opens the possibility of on-device inference of MoE, which was previously considered infeasible for such large models. Consequently, various systems have been proposed to leverage this sparsity and enable efficient MoE inference for edge devices. However, previous MoE inference systems like Fiddler[8] or DAOP[13] rely on DRAM-based offloading and are not suitable for memory constrained on-device environments. As recent MoE models grow to hundreds of gigabytes, RAM-offloading solutions become impractical. To address this, we propose FlashMoE, a system that offloads inactive experts to SSD, enabling efficient MoE inference under limited RAM. FlashMoE incorporates a lightweight ML-based caching strategy that adaptively combines recency and frequency signals to maximize expert reuse, significantly reducing storage I/O. In addition, we built a user-grade desktop platform to demonstrate the practicality of FlashMoE. On this real hardware setup, FlashMoE improves cache hit rate by up to 51% over well-known offloading policies such as LRU and LFU, and achieves up to 2.6x speedup compared to existing MoE inference systems.

MoE边缘计算缓存优化SSD加速

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