ZipMoE: Efficient On-Device MoE Serving via Lossless Compression and Cache-Affinity Scheduling
Abstract
Lay Summary
Mixture-of-Experts (MoE) is a class of incredibly powerful LLMs. However, running them on everyday devices like phones or laptops has been a long-standing headache, usually requiring developers to downgrade the model and sacrifice its accuracy. We asked a simple question: Can we bring fast MoE inference directly to your mobile devices without compromising performance at all? Our paper gives an affirmative answer to this question by introducing ZipMoE, a new solution to losslessly accelerate MoE inference on edge platforms. Exploiting the observation that LLM parameters contain significant redundancy from an information-theoretic view, we use losslessly compressed parameters during inference. This transforms expert access from loading data across congested PCIe bandwidth into an expert reconstruction process that can be computed in parallel using multiple CPU cores. ZipMoE is the very first MoE inference engine for edge platforms utilizing lossless compression, demonstrating order-of-magnitude improvements in latency and throughput. Our solution opens up the possibility of a broad range of private, responsive, and always-available on-device LLM applications.