English
Related papers

Related papers: Accelerating MoE with Dynamic In-Switch Computing …

200 papers

Tensor parallelism (TP) in large-scale LLM inference and training introduces frequent collective operations that dominate inter-GPU communication. While in-switch computing, exemplified by NVLink SHARP (NVLS), accelerates collective…

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

Mixture of Experts (MoE) has become a mainstream architecture for building Large Language Models (LLMs) by reducing per-token computation while enabling model scaling. It can be viewed as partitioning a large Feed-Forward Network (FFN) at…

Machine Learning · Computer Science 2025-08-27 Weilin Cai , Le Qin , Shwai He , Junwei Cui , Ang Li , Jiayi Huang

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 Expert (MoE) parallelism is a recent advancement that scales up the model size with constant computational cost. MoE selects different sets of parameters (i.e., experts) for each incoming token, resulting in a…

Machine Learning · Computer Science 2022-12-13 Chaoyang He , Shuai Zheng , Aston Zhang , George Karypis , Trishul Chilimbi , Mahdi Soltanolkotabi , Salman Avestimehr

Despite many recent works on Mixture of Experts (MoEs) for resource-efficient Transformer language models, existing methods mostly focus on MoEs for feedforward layers. Previous attempts at extending MoE to the self-attention layer fail to…

Machine Learning · Computer Science 2024-10-02 Róbert Csordás , Piotr Piękos , Kazuki Irie , Jürgen Schmidhuber

Mixture-of-Experts (MoE) models enable scalable performance but face severe memory constraints on edge devices. Existing offloading strategies struggle with I/O bottlenecks due to the dynamic, low-information nature of autoregressive expert…

Machine Learning · Computer Science 2026-03-12 Shuhuai Li , Jianghao Lin , Dongdong Ge , Yinyu Ye

In recent years, Mixture-of-Experts (MoE) has emerged as an effective approach for enhancing the capacity of deep neural network (DNN) with sub-linear computational costs. However, storing all experts on GPUs incurs significant memory…

Machine Learning · Computer Science 2025-03-11 Suraiya Tairin , Shohaib Mahmud , Haiying Shen , Anand Iyer

Mixture of experts (MoE) models achieve state-of-the-art results in language modeling but suffer from inefficient hardware utilization due to imbalanced token routing and communication overhead. While prior work has focused on optimizing…

Mixture-of-Experts (MoE) has successfully scaled up models while maintaining nearly constant computing costs. By employing a gating network to route input tokens, it selectively activates a subset of expert networks to process the…

Machine Learning · Computer Science 2025-04-22 Mohan Zhang , Pingzhi Li , Jie Peng , Mufan Qiu , Tianlong Chen

Mixture-of-Experts (MoE) has emerged as a promising approach to scale up deep learning models due to its significant reduction in computational resources. However, the dynamic nature of MoE leads to load imbalance among experts, severely…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-16 Chenqi Zhao , Wenfei Wu , Linhai Song , Yuchen Xu , Yitao Yuan

The Mixture-of-Expert (MoE) technique plays a crucial role in expanding the size of DNN model parameters. However, it faces the challenge of extended all-to-all communication latency during the training process. Existing methods attempt to…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-05-01 Chenyu Jiang , Ye Tian , Zhen Jia , Shuai Zheng , Chuan Wu , Yida Wang

The computational sparsity of Mixture-of-Experts (MoE) models enables sub-linear growth in compute cost as model size increases, thus offering a scalable path to training massive neural networks. However, existing implementations suffer…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-11 Osayamen Jonathan Aimuyo , Byungsoo Oh , Rachee Singh

The Mixture of Experts (MoE) architecture has become a fundamental building block in state-of-the-art large language models (LLMs), improving domain-specific expertise in LLMs and scaling model capacity without proportionally increasing…

Machine Learning · Computer Science 2026-05-13 Ankit Jyothish , Ali Jannesari , Aishwarya Sarkar , Joseph Zuber

As large language models (LLMs) continue to scale up, mixture-of-experts (MoE) has become a common technology in SOTA models. MoE models rely on expert parallelism (EP) to alleviate memory bottleneck, which introduces all-to-all…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-30 Xinru Tang , Jingxiang Hou , Dingcheng Jiang , Taiquan Wei , Jiaxin Liu , Jinyi Deng , Huizheng Wang , Qize Yang , Haoran Shang , Chao Li , Yang Hu , Shouyi Yin

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…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-16 Tian Wu , Liming Wang , Zijian Wen , Xiaoxi Zhang , Xu Chen , Jingpu Duan , Xianwei Zhang , Jinhang Zuo

Mixture-of-Experts (MoE) enjoys performance gain by increasing model capacity while keeping computation cost constant. When comparing MoE to dense models, prior work typically adopt the following setting: 1) use FLOPs or activated…

Machine Learning · Computer Science 2024-07-02 Xianzhi Du , Tom Gunter , Xiang Kong , Mark Lee , Zirui Wang , Aonan Zhang , Nan Du , Ruoming Pang

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

Mixture-of-Experts (MoE) layers activate a subset of model weights, dubbed experts, to improve model performance. MoE is particularly promising for deployment on process-in-memory (PIM) architectures, because PIM can naturally fit experts…

Hardware Architecture · Computer Science 2026-02-12 Hanyuan Gao , Xiaoxuan Yang

Mixture-of-Experts (MoE) has become a popular architecture for scaling large models. However, the rapidly growing scale outpaces model training on a single DC, driving a shift toward a more flexible, cross-DC training paradigm. Under this,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-23 Weihao Yang , Hao Huang , Donglei Wu , Ningke Li , Yanqi Pan , Qiyang Zheng , Wen Xia , Shiyi Li , Qiang Wang

The Mixture of Experts (MoE) is an advanced model architecture in the industry that combines multiple specialized expert models from various domains into a single supermodel. This approach enables the model to scale without significantly…

Machine Learning · Computer Science 2024-11-04 Jingming Guo , Yan Liu , Yu Meng , Zhiwei Tao , Banglan Liu , Gang Chen , Xiang Li