English
Related papers

Related papers: Efficient MoE Inference with Fine-Grained Scheduli…

200 papers

To deploy large Mixture-of-Experts (MoE) models cost-effectively, offloading-based single-GPU heterogeneous inference is crucial. While GPU-CPU architectures that offload cold experts are constrained by host memory bandwidth, emerging…

Hardware Architecture · Computer Science 2026-03-03 Yudong Pan , Yintao He , Tianhua Han , Lian Liu , Shixin Zhao , Zhirong Chen , Mengdi Wang , Cangyuan Li , Yinhe Han , Ying Wang

Mixture-of-Experts (MoE) models have become a dominant paradigm for scaling large language models, but their rapidly growing parameter sizes introduce a fundamental inefficiency during inference: most expert weights remain idle in GPU…

Machine Learning · Computer Science 2026-05-01 Qingxiu Liu , Cyril Y. He , Hanser Jiang , Zion Wang , Alan Zhao , Patrick P. C. Lee

The immense memory requirements of state-of-the-art Mixture-of-Experts (MoE) models present a significant challenge for inference, often exceeding the capacity of a single accelerator. While offloading experts to host memory is a common…

Machine Learning · Computer Science 2025-11-19 Wenfeng Wang , Jiacheng Liu , Xiaofeng Hou , Xinfeng Xia , Peng Tang , Mingxuan Zhang , Chao Li , Minyi Guo

Despite the significant breakthrough of Mixture-of-Experts (MoE), the increasing scale of these MoE models presents huge memory and storage challenges. Existing MoE pruning methods, which involve reducing parameter size with a uniform…

Computation and Language · Computer Science 2025-09-22 Sikai Bai , Haoxi Li , Jie Zhang , Zicong Hong , Song Guo

Mixture-of-Experts (MoE) has been adopted by many leading large models to reduce computational requirements. However, frequent inter-GPU communication in MoE expert parallelism (EP) becomes a performance challenge. We observe substantial…

Hardware Architecture · Computer Science 2026-05-08 Qijun Zhang , Chen Zhang , Zhuoshan Zhou , Haibo Wang , Zhe Zhou , Zhipeng Tu , Guangyu Sun , Zhiyao Xie , Yijia Diao , Zhigang Ji , Jingwen Leng , Guanghui He , Minyi Guo

Expert Parallelism (EP) permits Mixture of Experts (MoE) models to scale beyond a single GPU. To address load imbalance across GPUs in EP, existing approaches aim to balance the number of tokens each GPU processes. Surprisingly, we find…

Sparse Mixture-of-Experts (MoE) models can outperform dense large language models at similar computation by activating only a small set of experts per token. However, stacking many expert modules introduces substantial parameter memory,…

Artificial Intelligence · Computer Science 2026-04-03 Xin He , Shunkang Zhang , Kaijie Tang , Shaohuai Shi , Yuxin Wang , Zihao Zeng , Zhenheng Tang , Xiaowen Chu , Haiyan Yin , Ivor W. Tsang , Yew Soon Ong

Mixture-of-Experts (MoE) model architecture has emerged as a promising solution for scaling transformer models efficiently, offering sparse activation that reduces computational costs while increasing model capacity. However, as MoE models…

Machine Learning · Computer Science 2025-02-11 Seokjin Go , Divya Mahajan

Mixture-of-Experts (MoE) models offer computational efficiency during inference by activating only a subset of specialized experts for a given input. This enables efficient model scaling on multi-GPU systems that use expert parallelism…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-06-18 Zachary Doucet , Rishi Sharma , Martijn de Vos , Rafael Pires , Anne-Marie Kermarrec , Oana Balmau

Mixture-of-Experts (MoE) models have gained popularity as a means of scaling the capacity of large language models (LLMs) while maintaining sparse activations and reduced per-token compute. However, in memory-constrained inference settings,…

Machine Learning · Computer Science 2026-03-23 Vivan Madan , Prajwal Singhania , Abhinav Bhatele , Tom Goldstein , Ashwinee Panda

Mixture of Experts (MoE) models have emerged as the de facto architecture for scaling up language models without significantly increasing the computational cost. Recent MoE models demonstrate a clear trend towards high expert granularity…

Machine Learning · Computer Science 2026-03-30 Wentao Guo , Mayank Mishra , Xinle Cheng , Ion Stoica , Tri Dao

Mixture of Experts (MoE) models have enabled the scaling of Large Language Models (LLMs) and Vision Language Models (VLMs) by achieving massive parameter counts while maintaining computational efficiency. However, MoEs introduce several…

Prevailing LLM serving engines employ expert parallelism (EP) to implement multi-device inference of massive MoE models. However, the efficiency of expert parallel inference is largely bounded by inter-device communication, as EP embraces…

Machine Learning · Computer Science 2026-03-02 Yan Li , Zhenyu Zhang , Zhengang Wang , Pengfei Chen , Pengfei Zheng

Mixture-of-Experts (MoE) has recently emerged as the mainstream architecture for efficiently scaling large language models while maintaining near-constant computational cost. Expert parallelism distributes parameters by partitioning experts…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-01 Adrian Zhao , Zhenkun Cai , Zhenyu Song , Lingfan Yu , Haozheng Fan , Jun Wu , Yida Wang , Nandita Vijaykumar

Mixture-of-Experts (MoE) models are typically pre-trained with explicit load-balancing constraints to ensure statistically balanced expert routing. Despite this, we observe that even well-trained MoE models exhibit significantly imbalanced…

Machine Learning · Computer Science 2026-01-27 Xuan-Phi Nguyen , Shrey Pandit , Austin Xu , Caiming Xiong , Shafiq Joty

Large Language Models (LLMs) with Mixture-of-Expert (MoE) architectures achieve superior model performance with reduced computation costs, but at the cost of high memory capacity and bandwidth requirements. Near-Memory Processing (NMP)…

Performance · Computer Science 2025-09-12 Haochen Huang , Shuzhang Zhong , Zhe Zhang , Shuangchen Li , Dimin Niu , Hongzhong Zheng , Runsheng Wang , Meng Li

The Mixture-of-Experts (MoE) approach has demonstrated outstanding scalability in multi-task learning including low-level upstream tasks such as concurrent removal of multiple adverse weather effects. However, the conventional MoE…

Computer Vision and Pattern Recognition · Computer Science 2023-12-29 Rongyu Zhang , Yulin Luo , Jiaming Liu , Huanrui Yang , Zhen Dong , Denis Gudovskiy , Tomoyuki Okuno , Yohei Nakata , Kurt Keutzer , Yuan Du , Shanghang Zhang

Mixture-of-Experts (MoE) is an emerging technique for scaling large models with sparse activation. MoE models are typically trained in a distributed manner with an expert parallelism scheme, where experts in each MoE layer are distributed…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-11-26 Fahao Chen , Peng Li , Zicong Hong , Zhou Su , Song Guo

The training of large-scale Mixture of Experts (MoE) models faces a critical memory bottleneck due to severe load imbalance caused by dynamic token routing. This imbalance leads to memory overflow on GPUs with limited capacity, constraining…

With the widespread adoption of Mixture-of-Experts (MoE) models, there is a growing demand for efficient inference on memory-constrained devices. While offloading expert parameters to CPU memory and loading activated experts on demand has…

Machine Learning · Computer Science 2025-05-13 Yuxin Zhou , Zheng Li , Jun Zhang , Jue Wang , Yiping Wang , Zhongle Xie , Ke Chen , Lidan Shou