English
Related papers

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

200 papers

Mixture-of-Experts (MoE) has become a dominant architecture for scaling large language models (LLMs). However, the execution characteristics of MoE inference are changing rapidly and increasingly mismatch the assumptions underlying existing…

Hardware Architecture · Computer Science 2026-05-13 Jungwoo Kim , Rubens Lacouture , Genghan Zhang , Gina Sohn , Qizheng Zhang , Swapnil Gandhi , Christos Kozyrakis , Kunle Olukotun

The Mixture of Experts (MoE) is a widely known neural architecture where an ensemble of specialized sub-models optimizes overall performance with a constant computational cost. However, conventional MoEs pose challenges at scale due to the…

Computation and Language · Computer Science 2023-09-12 Ted Zadouri , Ahmet Üstün , Arash Ahmadian , Beyza Ermiş , Acyr Locatelli , Sara Hooker

Mixture-of-Experts (MoE) models improve the scalability of large language models (LLMs) by activating only a small subset of relevant experts per input. However, the sheer number of expert networks in an MoE model introduces a significant…

Machine Learning · Computer Science 2026-03-03 Qian Chen , Xianhao Chen , Kaibin Huang

Recent advancements in Mixture of Experts (MoE) models have significantly increased their parameter scale as well as model performance. Extensive offloading techniques have been proposed to address the GPU memory limitations of MoE…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-03 Zhibin Wang , Zhonghui Zhang , Yuhang Zhou , Zibo Wang , Mo Zhou , Peng Jiang , Weilin Cai , Chengying Huan , Rong Gu , Sheng Zhong , Chen Tian

Sparse Mixture of Experts (MoE) large language models (LLMs) are gradually becoming the mainstream approach for ultra-large-scale models. Existing optimization efforts for MoE models have focused primarily on coarse-grained MoE…

Computation and Language · Computer Science 2025-05-07 Haoqi Yang , Luohe Shi , Qiwei Li , Zuchao Li , Ping Wang , Bo Du , Mengjia Shen , Hai Zhao

Mixture-of-Experts (MoE) large language models (LLM) have memory requirements that often exceed the GPU memory capacity, requiring costly parameter movement from secondary memories to the GPU for expert computation. In this work, we present…

Machine Learning · Computer Science 2024-05-30 Taehyun Kim , Kwanseok Choi , Youngmock Cho , Jaehoon Cho , Hyuk-Jae Lee , Jaewoong Sim

Mixture-of-Experts (MoE) models scale large language models through conditional computation, but inference becomes memory-bound once expert weights exceed the capacity of GPU memory. In this case, weights must be offloaded to external…

Machine Learning · Computer Science 2025-12-05 Zehao Fan , Zhenyu Liu , Yunzhen Liu , Yayue Hou , Hadjer Benmeziane , Kaoutar El Maghraoui , Liu Liu

Sparsely-gated mixture-of-experts (MoE) has been widely adopted to scale deep learning models to trillion-plus parameters with fixed computational cost. The algorithmic performance of MoE relies on its token routing mechanism that forwards…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-06-06 Changho Hwang , Wei Cui , Yifan Xiong , Ziyue Yang , Ze Liu , Han Hu , Zilong Wang , Rafael Salas , Jithin Jose , Prabhat Ram , Joe Chau , Peng Cheng , Fan Yang , Mao Yang , Yongqiang Xiong

The Mixture of Experts (MoE) models are emerging as the latest paradigm for Large Language Models (LLMs). However, due to memory constraints, MoE models with billions or even trillions of parameters can only be deployed in multi-GPU or even…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-14 Bowen Zhou , Jinrui Jia , Wenhao He , Yong Zhang , Fang Dong

Mixture-of-Experts (MoE) has garnered significant attention for its ability to scale up neural networks while utilizing the same or even fewer active parameters. However, MoE does not alleviate the massive memory requirements of networks,…

Machine Learning · Computer Science 2026-04-21 Mingyu Cao , Gen Li , Jie Ji , Jiaqi Zhang , Ajay Jaiswal , Li Shen , Xiaolong Ma , Shiwei Liu , Lu Yin

Expert parallelism has emerged as a key strategy for distributing the computational workload of sparsely-gated mixture-of-experts (MoE) models across multiple devices, enabling the processing of increasingly large-scale models. However, the…

Machine Learning · Computer Science 2025-05-30 Weilin Cai , Juyong Jiang , Le Qin , Junwei Cui , Sunghun Kim , Jiayi Huang

Mixture-of-Experts (MoE) has become a practical architecture for scaling LLM capacity while keeping per-token compute modest, but deploying MoE models on a single, memory-limited GPU remains difficult because expert weights dominate the HBM…

Performance · Computer Science 2026-02-09 Kexin Chu , Dawei Xiang , Zixu Shen , Yiwei Yang , Zecheng Liu , Wei Zhang

Mixture of Experts (MoE) models enable parameter-efficient scaling through sparse expert activations, yet optimizing their inference and memory costs remains challenging due to limited understanding of their specialization behavior. We…

Machine Learning · Computer Science 2026-03-09 Marmik Chaudhari , Idhant Gulati , Nishkal Hundia , Pranav Karra , Shivam Raval

Mixture-of-Experts (MoE) models have recently demonstrated exceptional performance across a diverse range of applications. The principle of sparse activation in MoE models facilitates an offloading strategy, wherein active experts are…

Computation and Language · Computer Science 2025-10-15 Yushu Zhao , Yubin Qin , Yang Wang , Xiaolong Yang , Huiming Han , Shaojun Wei , Yang Hu , Shouyi Yin

The Mixture of Experts (MoE) architecture has demonstrated significant advantages as it enables to increase the model capacity without a proportional increase in computation. However, the large MoE model size still introduces substantial…

Machine Learning · Computer Science 2025-04-09 Shuzhang Zhong , Yanfan Sun , Ling Liang , Runsheng Wang , Ru Huang , Meng Li

Large Language Models (LLMs) with the Mixture-of-Experts (MoE) architectures have shown promising performance on various tasks. However, due to the huge model sizes, running them in resource-constrained environments where the GPU memory is…

Machine Learning · Computer Science 2025-05-02 Keisuke Kamahori , Tian Tang , Yile Gu , Kan Zhu , Baris Kasikci

The expansion of large language models is increasingly limited by the constrained memory capacity of modern GPUs. To mitigate this, Mixture-of-Experts (MoE) architectures activate only a small portion of parameters during inference,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-31 Zixu Shen , Kexin Chu , Yifan Zhang , Dawei Xiang , Runxin Wu , Wei Zhang

The application of mixture-of-experts (MoE) is gaining popularity due to its ability to improve model's performance. In an MoE structure, the gate layer plays a significant role in distinguishing and routing input features to different…

Machine Learning · Computer Science 2024-02-05 Zhitian Xie , Yinger Zhang , Chenyi Zhuang , Qitao Shi , Zhining Liu , Jinjie Gu , Guannan Zhang

The Mixture of Experts (MoE) model becomes an important choice of large language models nowadays because of its scalability with sublinear computational complexity for training and inference. However, existing MoE models suffer from two…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-04-25 Xin Chen , Hengheng Zhang , Xiaotao Gu , Kaifeng Bi , Lingxi Xie , Qi Tian

We present ScatterMoE, an implementation of Sparse Mixture-of-Experts (SMoE) on GPUs. ScatterMoE builds upon existing implementations, and overcoming some of the limitations to improve inference and training speed, and memory footprint.…

Machine Learning · Computer Science 2024-10-07 Shawn Tan , Yikang Shen , Rameswar Panda , Aaron Courville