English
Related papers

Related papers: HetuMoE: An Efficient Trillion-scale Mixture-of-Ex…

200 papers

Sparse mixture of expert architectures (MoEs) scale model capacity without significant increases in training or inference costs. Despite their success, MoEs suffer from a number of issues: training instability, token dropping, inability to…

Machine Learning · Computer Science 2024-05-28 Joan Puigcerver , Carlos Riquelme , Basil Mustafa , Neil Houlsby

With the growing prominence of the Mixture of Experts (MoE) architecture in developing large-scale foundation models, we investigate the Hierarchical Mixture of Experts (HMoE), a specialized variant of MoE that excels in handling complex…

Machine Learning · Statistics 2025-03-10 Huy Nguyen , Xing Han , Carl Harris , Suchi Saria , Nhat Ho

Mixture-of-experts (MoE) is gaining increasing attention due to its unique properties and remarkable performance, especially for language tasks. By sparsely activating a subset of parameters for each token, MoE architecture could increase…

Computation and Language · Computer Science 2025-06-24 Ka Man Lo , Zeyu Huang , Zihan Qiu , Zili Wang , Jie Fu

In today's landscape, Mixture of Experts (MoE) is a crucial architecture that has been used by many of the most advanced models. One of the major challenges of MoE models is that they usually require much more memory than their dense…

Machine Learning · Computer Science 2025-11-11 Shuning Lin , Yifan He , Yitong Chen

Mixture-of-Expert (MoE) models enable efficient inference by employing smaller experts and activating only a subset of them per token. MoE serving engines distribute experts across multiple GPUs and route tokens to appropriate GPUs at…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-20 Sourish Wawdhane , Avinash Kumar , Poulami Das

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

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 pervasive "memory wall" bottleneck is significantly amplified in modern large-scale Mixture-of-Experts (MoE) architectures. MoE's inherent architectural sparsity leads to sparse arithmetic compute and also introduces substantial…

Machine Learning · Computer Science 2026-01-12 Jiyuan Zhang , Yining Liu , Siqi Yan , Lisen Deng , Jennifer Cao , Shuqi Yang , Min Ni , Bi Xue , Shen Li

Sparse Mixture of Experts (MoE) models offer a scalable and efficient architecture for training large neural networks by activating only a subset of parameters ("experts") for each input. A learned router computes a distribution over these…

Machine Learning · Computer Science 2025-10-14 Nabil Omi , Siddhartha Sen , Ali Farhadi

The sparse Mixture-of-Experts (MoE) model is powerful for large-scale pre-training and has achieved promising results due to its model capacity. However, with trillions of parameters, MoE is hard to be deployed on cloud or mobile…

Machine Learning · Computer Science 2022-06-03 Tianyu Chen , Shaohan Huang , Yuan Xie , Binxing Jiao , Daxin Jiang , Haoyi Zhou , Jianxin Li , Furu Wei

Mixture-of-Experts (MoE) language models can reduce computational costs by 2-4$\times$ compared to dense models without sacrificing performance, making them more efficient in computation-bounded scenarios. However, MoE models generally…

Machine Learning · Computer Science 2024-04-09 Bowen Pan , Yikang Shen , Haokun Liu , Mayank Mishra , Gaoyuan Zhang , Aude Oliva , Colin Raffel , Rameswar Panda

The Mixture-of-Experts (MoE) paradigm has emerged as a promising solution to scale up model capacity while maintaining inference efficiency. However, deploying MoE models across heterogeneous end-cloud environments poses new challenges in…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-11 Zheming Yang , Yunqing Hu , Sheng Sun , Wen Ji

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 Mixture of Experts (MoE) architecture has emerged as a powerful paradigm for scaling large language models (LLMs) while maintaining inference efficiency. However, their enormous memory requirements make them prohibitively expensive to…

Machine Learning · Computer Science 2025-06-24 Zichong Li , Chen Liang , Zixuan Zhang , Ilgee Hong , Young Jin Kim , Weizhu Chen , Tuo Zhao

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…

The Mixture-of-Experts (MoE) structure scales the Transformer-based large language models (LLMs) and improves their performance with only the sub-linear increase in computation resources. Recently, a fine-grained DeepSeekMoE structure is…

Machine Learning · Computer Science 2025-03-10 Zewen Jin , Shengnan Wang , Jiaan Zhu , Hongrui Zhan , Youhui Bai , Lin Zhang , Zhenyu Ming , Cheng Li

Mixture-of-Experts (MoE) architectures offer the promise of larger model capacity without the prohibitive costs of fully dense designs. However, in real-world inference serving, load skew across experts often leads to suboptimal device…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-05-30 Shaoyu Wang , Guangrong He , Geon-Woo Kim , Yanqi Zhou , Seo Jin Park

Real-world model deployment across multiple domains requires multimodal models to operate under two complementary regimes: (1) multi-task pretraining, tasks are co-available at design time where related tasks could borrow representational…

Machine Learning · Computer Science 2026-05-12 Xing Han , Shravan Chaudhari , Tanvi Ranade , Rama Chellappa , Suchi Saria

Modern Mixture-of-Experts (MoE) models increasingly rely on large-scale AI accelerator clusters for efficient training. Ascend NPUs expose heterogeneous on-chip compute resources, including matrix-oriented AIC units and vector-oriented AIV…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-25 Zewen Jin , Congkun Ai , Guangpeng Zhang , Hanbo Zhang , Haoran Wang , Shihan Xiao , Da Lei , Xuefeng Jin , Teng Su , Cheng Li

Mixture-of-Experts (MoE) models have become a widely-adopted solution to continue scaling model sizes without a corresponding linear increase in compute. During MoE model training, each input token is dynamically routed to a subset of…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-20 Athinagoras Skiadopoulos , Mark Zhao , Swapnil Gandhi , Thomas Norrie , Shrijeet Mukherjee , Christos Kozyrakis