arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.09983cs.LGcs.AIcs.CL
收藏

MoE-SpAc:基于推测激活效用的高效MoE推理:在异构边缘场景中

MoE-SpAc: Efficient MoE Inference Based on Speculative Activation Utility in Heterogeneous Edge Scenarios

Shuhuai Li, Jianghao Lin, Dongdong Ge, Yinyu Ye

AI总结:

MoE-SpAc通过整合推测效用估计器、异构工作负载平衡器和异步执行引擎,在异构边缘场景中实现高效MoE推理,TPS提升42%,速度提升4.04倍。

摘要:

混合专家(MoE)模型能够实现可扩展的性能,但在边缘设备上面临严峻的内存约束。现有的卸载策略由于自回归专家激活的动态性和低信息性,难以应对I/O瓶颈。在本文中,我们提出将推测解码(SD)不仅仅作为一种计算加速器,而是作为一种信息前瞻传感器用于内存管理,这得到了我们的理论和实证分析的支持。因此,我们引入了MoE-SpAc,一种MoE推理框架,它集成了一个推测效用估计器以跟踪专家需求,一个异构工作负载平衡器通过在线整数优化动态划分计算,以及一个异步执行引擎以在同一个效用空间中统一预取和驱逐。在七个基准测试中的广泛实验表明,MoE-SpAc在SOTA SD基线上的TPS提高了42%,并在所有标准基线上的平均速度提升了4.04倍。代码可在https://github.com/lshAlgorithm/MoE-SpAc上获得。

英文摘要:

Mixture-of-Experts (MoE) models enable scalable performance but face severe memory constraints on edge devices. Existing offloading strategies struggle with I/O bottlenecks due to the dynamic, low-information nature of autoregressive expert activation. In this paper, we propose to repurpose Speculative Decoding (SD) not merely as a compute accelerator, but as an informative lookahead sensor for memory management, supported by our theoretical and empirical analyses. Hence, we introduce MoE-SpAc, an MoE inference framework that integrates a Speculative Utility Estimator to track expert demand, a Heterogeneous Workload Balancer to dynamically partition computation via online integer optimization, and an Asynchronous Execution Engine to unify the prefetching and eviction in the same utility space. Extensive experiments on seven benchmarks demonstrate that MoE-SpAc achieves a 42% improvement in TPS over the SOTA SD-based baseline, and an average 4.04x speedup over all standard baselines. Code is available at https://github.com/lshAlgorithm/MoE-SpAc .

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑