SMoE: An Algorithm-System Co-Design for Pushing MoE to the Edge via Expert Substitution
The Mixture of Experts (MoE) architecture has emerged as a key technique for scaling Large Language Models by activating only a subset of experts per query. Deploying MoE on consumer-grade edge hardware, however, is constrained by limited device memory, making dynamic expert offloading essential. Unlike prior work that treats offloading purely as a scheduling problem, we leverage expert importance to guide decisions, substituting low-importance activated experts with functionally similar ones already cached in GPU memory, thereby preserving accuracy. As a result, this design reduces memory usage and data transfer, while largely eliminating PCIe overhead. In addition, we introduce a scheduling policy that maximizes the reuse ratio of GPU-cached experts, further boosting efficiency. Extensive evaluations show that our approach delivers 48% lower decoding latency with over 60% expert cache hit rate, while maintaining nearly lossless accuracy.
Code (0)
등록된 구현이 없습니다.
Similar Papers 제목 키워드 기반
Edge-Aware Autoencoder Design for Real-Time Mixture-of-Experts Image Compression
Steered-Mixtures-of-Experts (SMoE) models provide sparse, edge-aware representations, applicable to many use-cases in image processing. This includes denoising, super-resolution and compression of 2D- and higher dimensio…
DenoisingImage CompressionImage ReconstructionMixture-of-Experts+1Beyond AI as Assistants: Toward Autonomous Discovery in Cosmology
Recent advances in artificial intelligence (AI) agents are pushing AI beyond tools toward autonomous scientific discovery. We discuss two complementary agentic systems for cosmology: \texttt{CMBEvolve}, which targets tas…
Out-of-Distribution DetectionModeling Expert Interactions in Sparse Mixture of Experts via Graph Structures
Sparse Mixture of Experts (SMoE) has emerged as a promising solution to achieving unparalleled scalability in deep learning by decoupling model parameter count from computational cost. By activating only a small subset o…
Steered Mixture-of-Experts Autoencoder Design for Real-Time Image Modelling and Denoising
Research in the past years introduced Steered Mixture-of-Experts (SMoE) as a framework to form sparse, edge-aware models for 2D- and higher dimensional pixel data, applicable to compression, denoising, and beyond, and ca…
DecoderDenoisingMixture-of-ExpertsSuper-ResolutionSteered Mixture of Experts Regression for Image Denoising with Multi-Model-Inference
In this paper we introduce a novel block-based regression strategy for image denoising based on edge-aware Steered-Mixture-of-Experts (SMoE) models. SMoEs provide very sparse image representations, able to model sharp ed…
DenoisingImage DenoisingMixture-of-Expertsregression