English

ZipMoE: Efficient On-Device MoE Serving via Lossless Compression and Cache-Affinity Scheduling

Distributed, Parallel, and Cluster Computing 2026-05-25 v2 Artificial Intelligence Machine Learning

Abstract

While Mixture-of-Experts (MoE) architectures substantially bolster the expressive power of large-language models, their prohibitive memory footprint severely impedes the practical deployment on resource-constrained edge devices, especially when model behavior must be preserved without relying on lossy quantization. In this paper, we present ZipMoE, an efficient and semantically lossless on-device MoE serving system. ZipMoE exploits the synergy between the hardware properties of edge devices and the statistical redundancy inherent to MoE parameters via a caching-scheduling co-design with provable performance guarantee. Fundamentally, our design shifts the paradigm of on-device MoE inference from an I/O-bound bottleneck to a compute-centric workflow that enables efficient parallelization. We implement a prototype of ZipMoE and conduct extensive experiments on representative edge computing platforms using popular open-source MoE models and real-world workloads. Our evaluation reveals that ZipMoE achieves up to 72.77%72.77\% inference latency reduction and up to 6.76×6.76\times higher throughput than the state-of-the-art systems.Our code is available at: https://github.com/npnothard/ZipMoE-ICML26.

Keywords

Cite

@article{arxiv.2601.21198,
  title  = {ZipMoE: Efficient On-Device MoE Serving via Lossless Compression and Cache-Affinity Scheduling},
  author = {Yuchen Yang and Yaru Zhao and Pu Yang and Shaowei Wang and Zhi-Hua Zhou},
  journal= {arXiv preprint arXiv:2601.21198},
  year   = {2026}
}

Comments

ICML 2026

R2 v1 2026-07-01T09:24:54.808Z