Accelerating Dense LLMs via L0-regularized Mixture-of-Experts
In the authors' words
Large language models (LLMs) achieve strong performance but suffer from slow and costly inference. Existing acceleration methods often lead to noticeable performance degradation, while Mixture-of-Experts (MoE) models require extensive computational resources. In this paper, we propose L0-MoE, a lightweight MoE approach using L0-regularization to accelerate dense LLMs nearly without performance loss. Our method introduces a cluster confusion matrix for domain-aware dataset curation and applies dynamic batching for efficient training. Experiments show that L0-MoE achieves up to 2.5x speedup over dense models while maintaining competitive performance, outperforming existing LLM acceleration baselines.
Appeared: Monday, September 21. arXiv. Preprint with a published version.
Published version: Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics, 2025