Accelerating Edge Inference for Distributed MoE Models with Latency-Optimized Expert Placement

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wu, Tian, Wang, Liming, Wen, Zijian, Zhang, Xiaoxi, Chen, Xu, Duan, Jingpu, Zhang, Xianwei, Zuo, Jinhang
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915938306097152
author Wu, Tian
Wang, Liming
Wen, Zijian
Zhang, Xiaoxi
Chen, Xu
Duan, Jingpu
Zhang, Xianwei
Zuo, Jinhang
author_facet Wu, Tian
Wang, Liming
Wen, Zijian
Zhang, Xiaoxi
Chen, Xu
Duan, Jingpu
Zhang, Xianwei
Zuo, Jinhang
contents The emergence of Mixture-of-Experts (MoE) has transformed the scaling of large language models by enabling vast model capacity through sparse activation. Yet, converting these performance gains into practical edge deployment remains difficult, as the massive memory footprint and communication demands often overwhelm resource-limited environments. While centralized cloud-based solutions are available, they are frequently plagued by prohibitive infrastructure costs, latency issues, and privacy concerns. Moreover, existing edge-oriented optimizations largely overlook the complexities of heterogeneous hardware, focusing instead on isolated or uniform device setups. In response, this paper proposes Prism, an inference framework engineered for collaborative MoE serving across diverse GPU-equipped edge servers. By leveraging the intrinsic sparsity and input locality of MoE workloads, Prism minimizes inter-server communication and optimizes expert placement within diverse resource constraints. The framework integrates an activation-aware placement strategy that balances local request coverage with memory utilization, supplemented by a runtime migration mechanism to adapt expert distribution to dynamic workload changes. Experiments on contemporary MoE models and datasets demonstrate that Prism reduces inference latency by up to 30.6% and significantly lowers communication costs compared to state-of-the-art baselines, confirming the effectiveness of cooperative edge-based MoE serving.
format Preprint
id arxiv_https___arxiv_org_abs_2508_12851
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Accelerating Edge Inference for Distributed MoE Models with Latency-Optimized Expert Placement
Wu, Tian
Wang, Liming
Wen, Zijian
Zhang, Xiaoxi
Chen, Xu
Duan, Jingpu
Zhang, Xianwei
Zuo, Jinhang
Distributed, Parallel, and Cluster Computing
The emergence of Mixture-of-Experts (MoE) has transformed the scaling of large language models by enabling vast model capacity through sparse activation. Yet, converting these performance gains into practical edge deployment remains difficult, as the massive memory footprint and communication demands often overwhelm resource-limited environments. While centralized cloud-based solutions are available, they are frequently plagued by prohibitive infrastructure costs, latency issues, and privacy concerns. Moreover, existing edge-oriented optimizations largely overlook the complexities of heterogeneous hardware, focusing instead on isolated or uniform device setups. In response, this paper proposes Prism, an inference framework engineered for collaborative MoE serving across diverse GPU-equipped edge servers. By leveraging the intrinsic sparsity and input locality of MoE workloads, Prism minimizes inter-server communication and optimizes expert placement within diverse resource constraints. The framework integrates an activation-aware placement strategy that balances local request coverage with memory utilization, supplemented by a runtime migration mechanism to adapt expert distribution to dynamic workload changes. Experiments on contemporary MoE models and datasets demonstrate that Prism reduces inference latency by up to 30.6% and significantly lowers communication costs compared to state-of-the-art baselines, confirming the effectiveness of cooperative edge-based MoE serving.
title Accelerating Edge Inference for Distributed MoE Models with Latency-Optimized Expert Placement
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2508.12851