MiLo: Efficient Quantized MoE Inference with Mixture of Low-Rank Compensators
On the map: Mixture-of-experts → Efficient MoE training and inference systems
Abstract (arXiv)
A critical approach for efficiently deploying Mixture-of-Experts (MoE) models with massive parameters is quantization. However, state-of-the-art MoE models suffer from non-negligible accuracy loss with extreme quantization, such as under 4 bits. To address this, we introduce MiLo, a novel method that augments highly quantized MoEs with a mixture of low-rank compensators. These compensators consume only a small amount of additional memory but significantly recover accuracy loss from extreme quantization. MiLo also identifies that MoEmodels exhibit distinctive characteristics across weights due to their hybrid dense-sparse architectures, and employs adaptive rank selection policies along with iterative optimizations to close the accuracy gap. MiLo does not rely on calibration data, allowing it to generalize to different MoE models and datasets without overfitting to a calibration set. To avoid the hardware inefficiencies of extreme quantization, such as 3-bit, MiLo develops Tensor Core-friendly 3-bit kernels, enabling measured latency speedups on 3-bit quantized MoE models. Our evaluation shows that MiLo outperforms existing methods on SoTA MoE models across various tasks.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Placed in Efficient MoE training and inference systems, one of 12 directions in Mixture-of-experts. Builds on 26 of its 43 references that are in the map. First posted on arXiv 3 Apr 2025.
Builds on
26 of 43 references.
Closest ancestors · 2
Same direction, newest first
- 2023Fast Inference of Mixture-of-Experts Language Models with Offloading28 Dec
- 2022DeepSpeed-MoE: Advancing Mixture-of-Experts Inference and Training to Power Next-Generation AI Scale14 Jan
Other references · 24
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2024Mixtral of Expertscited by 1,823
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,386
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
12 more
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 632
- 2022ZeroQuant: Efficient and Affordable Post-Training Quantization for Large-Scale Transformerscited by 526
- 2022The case for 4-bit precision: k-bit Inference Scaling Lawscited by 251
- 2021Efficient Large Scale Language Modeling with Mixtures of Expertscited by 175
- 2023ASVD: Activation-aware Singular Value Decomposition for Compressing Large Language Modelscited by 169
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policycited by 108
- 2022LUT-GEMM: Quantized Matrix Multiplication based on LUTs for Efficient Inference in Large-Scale Generative Language Modelscited by 105
- 2022Task-Specific Expert Pruning for Sparse Mixture-of-Expertscited by 72
- 2023Mixture of Quantized Experts (MoQE): Complementary Effect of Low-bit Quantization and Robustnesscited by 34
- 2022Who Says Elephants Can't Run: Bringing Large Scale MoE Models into Cloud Scale Productioncited by 27
Built on by
Citing papers per year: 2025: 2; 2026: 6.
8 papers.
- 2025PuzzleMoE: Efficient Compression of Large Mixture-of-Experts Models via Sparse Expert Merging and Bit-packed inferencecited by 9
- 2026PagedWeight: Efficient MoE LLM Serving with Dynamic Quality-Aware Weight Quantizationcited by 2
- 2025MoE-Compression: How the Compression Error of Experts Affects the Inference Accuracy of MoE Model?cited by 1
- 2026TileQ: Efficient Low-Rank Quantization of Mixture-of-Experts with 2D Tilingcited by 1
- 2026BitsMoE: Efficient Spectral Energy-Guided Bit Allocation for MoE LLM Quantizationcited by 0
- 2026Every Expert Counts: ExactMoE for Memory-Efficient W4A16 Inferencecited by 0
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Modelscited by 0
- 2026DuoRate: Two-Rate Per-Expert Precision Allocation for Mixture-of-Experts Servingcited by 0
- 2026DuoRate: Two-Rate Per-Expert Precision Allocation for Mixture-of-Experts Serving9 Oct 2026
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Models1 Oct 2026
- 2026Every Expert Counts: ExactMoE for Memory-Efficient W4A16 Inference15 Aug 2026
- 2026PagedWeight: Efficient MoE LLM Serving with Dynamic Quality-Aware Weight Quantization17 Jul 2026
- 2026BitsMoE: Efficient Spectral Energy-Guided Bit Allocation for MoE LLM Quantization22 May 2026
- 2026TileQ: Efficient Low-Rank Quantization of Mixture-of-Experts with 2D Tiling10 May 2026
- 2025PuzzleMoE: Efficient Compression of Large Mixture-of-Experts Models via Sparse Expert Merging and Bit-packed inference6 Nov 2025
- 2025MoE-Compression: How the Compression Error of Experts Affects the Inference Accuracy of MoE Model?9 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2026Hardware-Native Joint Sparse-Quantization for Trillion-Scale Mixture-of-Experts
- 2025MxMoE: Mixed-precision Quantization for MoE with Accuracy and Performance Co-Design
- 2026AlphaQ: Calibration-Free Bit Allocation for Mixture-of-Experts Quantization
- 2024QuantMoE-Bench: Examining Post-Training Quantization for Mixture-of-Experts
- 2026Efficient Quantization of Mixture-of-Experts with Theoretical Generalization Guarantees
- 2026MonoMoE: An Efficient Fused Mega-kernel for Quantized MoE Decoding
- 2025LExI: Layer-Adaptive Active Experts for Efficient MoE Model Inference
- 2023Pre-gated MoE: An Algorithm-System Co-Design for Fast and Scalable Mixture-of-Expert Inference