arXiv Machine Learning

DAOP: Data-Aware Offloading and Predictive Pre-Calculation for Efficient MoE Inference

arXiv Machine Learning
Oct 21

Specter: Fast MoE Inference via Quantization-Enabled Multi-Step Speculative Offloading

Specter is an inference system for mixture-of-experts (MoE) models that uses quantization to enable speculative decoding and expert offloading. By running a lightweight quantized draft model on the device, Specter predicts future expert usage and pre‑organizes expert movement, reducing PCIe I/O bottlenecks. The approach yields an average 1.50× speedup (up to 1.88×) in time per output token compared to state‑of‑the‑art systems, making single‑GPU MoE inference more practical on commodity hardware.

By Wenfeng Wang, Xiaofeng Hou, Xinfeng Xia, Xinkai WAng, Mingxuan Zhang, Peng Tang, Jiacheng Liu, Chao Li
arXiv AI
Oct 5

Hardware-Native Joint Sparse-Quantization for Trillion-Scale Mixture-of-Experts

The paper introduces a hardware-software co‑design framework that compresses Mixture‑of‑Experts (MoE) model weights into low‑precision, hardware‑native sparse representations, enabling efficient execution on Sparse Tensor Cores (SpTCs). By relaxing discrete support selection through continuous reparameterization, the method jointly optimizes quantized weights and supports a router‑weighted reconstruction objective, achieving up to 4.35 percentage‑point gains in joint sparse‑quantization accuracy while retaining 96.09% of the original model’s performance. A custom grouped sparse GEMM kernel further boosts inference speed, outperforming NVIDIA’s baseline by up to 1.65× and reducing latency by up to 4.03× on B200 GPUs.

By Kwanhee Lee, Namhoon Lee, Dan Alistarh