跳到正文
返回论文列表
cs.LG提交于 已译

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

中文摘要

混合专家(Mixture-of-Experts, MoE)模型具备可扩展性能,但在边缘设备上面临严重的内存约束。由于自回归专家激活具有动态性和低信息量,现有卸载策略难以突破 I/O 瓶颈。MoE-SpAc 将推测解码(Speculative Decoding, SD)重新用作内存管理的高信息量前瞻传感器,并通过理论与实证分析予以支持。该框架集成推测效用估计器以跟踪专家需求、异构工作负载均衡器以在线整数规划动态划分计算,以及异步执行引擎,在统一效用空间中整合预取与驱逐。七个 benchmark 上的实验表明,MoE-SpAc 的 TPS 较当前最优 SD 基线提升 42%,相较所有标准基线平均加速 4.04x。

关键要点

  1. 01现有 MoE 卸载受动态、低信息量专家激活导致的 I/O 瓶颈限制
  2. 02将推测解码用作内存管理的高信息量前瞻传感器
  3. 03通过效用估计、在线整数规划和异步执行协同管理边缘推理
  4. 04七个 benchmark 上 TPS 较最优 SD 基线提升 42%
  5. 05相较所有标准基线取得平均 4.04x 加速

解读

尚无解读。

原始英文摘要

arXiv:2603.09983v2 Announce Type: replace-cross Abstract: 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 .

同方向论文 · cs.LG

查看全部 →

反向发现 · 这篇论文相关的工具与提示词

去交叉发现 →

关联提示词 · 1