LightMoE: Task-Aware Expert Availability Management for Memory-Efficient MoE-LLM Inference
Puhan Luo, Yunhao Yao, Junyang Wang, Junyang Zhang, Xiangyang Li
Abstract
Mixture-of-Experts (MoE) models offer a promising path for scaling model capacity, yet their massive memory footprint poses significant challenges for deployment on resource-constrained edge devices. Existing solutions, such as static pruning or dynamic offloading, often struggle to balance model accuracy with inference latency due to irreversible information loss or prohibitive I/O overhead. In this paper, we propose LightMoE, a novel framework for memory-efficient MoE inference that exploits the inherent functional redundancy and temporal locality of expert activation. LightMoE employs a frequency-aware expert initialization strategy to retain a compact core of resident experts and introduces a similarity-based redirection mechanism to compensate for missing experts without incurring I/O costs. Furthermore, it incorporates a lightweight runtime manager that performs coarse-grained, task-level expert replacement to adapt to shifting data distributions. Empirical evaluations on representative edge platforms demonstrate that LightMoE achieves a superior accuracy-efficiency trade-off, improving average accuracy by 4.3% over static pruning and 2.4% over dynamic swapping methods, while maintaining inference latency comparable to strictly pruned models.- Anthology ID:
- 2026.findings-acl.1062
- Volume:
- Findings of the Association for Computational Linguistics: ACL 2026
- Month:
- July
- Year:
- 2026
- Address:
- San Diego, California, United States
- Editors:
- Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
- Venue:
- Findings
- SIG:
- Publisher:
- Association for Computational Linguistics
- Note:
- Pages:
- 21124–21134
- Language:
- URL:
- https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1062/
- DOI:
- Cite (ACL):
- Puhan Luo, Yunhao Yao, Junyang Wang, Junyang Zhang, and Xiangyang Li. 2026. LightMoE: Task-Aware Expert Availability Management for Memory-Efficient MoE-LLM Inference. In Findings of the Association for Computational Linguistics: ACL 2026, pages 21124–21134, San Diego, California, United States. Association for Computational Linguistics.
- Cite (Informal):
- LightMoE: Task-Aware Expert Availability Management for Memory-Efficient MoE-LLM Inference (Luo et al., Findings 2026)
- PDF:
- https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1062.pdf