English
Related papers

Related papers: SwitchHead: Accelerating Transformers with Mixture…

200 papers

The Mixture-of-Experts (MoE) architecture has become a predominant paradigm for scaling large language models (LLMs). Despite offering strong performance and computational efficiency, large MoE-based LLMs like DeepSeek-V3-0324 and…

Machine Learning · Computer Science 2025-08-08 Xiaodong Chen , Mingming Ha , Zhenzhong Lan , Jing Zhang , Jianguo Li

Recent large language models (LLMs) have tended to leverage sparsity to reduce computations, employing the sparsely activated mixture-of-experts (MoE) technique. MoE introduces four modules, including token routing, token communication,…

Machine Learning · Computer Science 2025-01-22 Xinglin Pan , Wenxiang Lin , Lin Zhang , Shaohuai Shi , Zhenheng Tang , Rui Wang , Bo Li , Xiaowen Chu

In the era of large language models, Mixture-of-Experts (MoE) is a promising architecture for managing computational costs when scaling up model parameters. However, conventional MoE architectures like GShard, which activate the top-$K$ out…

Production LLM workloads increasingly serve discriminative tasks, such as classification, recommendation, and verification, whose answers are read from the logits of a single prefill pass with no autoregressive decoding. Serving these…

Machine Learning · Computer Science 2026-05-18 Zhaoyuan Su , Olatunji Ruwase , Karthik Ganesan , Aurick Qiao , Samyam Rajbhandari , Juncheng Yang , Yue Cheng , Yuxiong He

Mixture-of-Experts (MoE) models are widely used to scale language models, yet their expert routing behavior and adaptation in a multilingual setting remain underexplored. In this work, we study multilingual routing dynamics during continual…

Computation and Language · Computer Science 2026-05-29 Aditi Khandelwal , Marius Mosbach , Verna Dankers , Siva Reddy , Golnoosh Farnadi

The mixture of experts (MoE) model is a sparse variant of large language models (LLMs), designed to hold a better balance between intelligent capability and computational overhead. Despite its benefits, MoE is still too expensive to deploy…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-23 Haodong Wang , Qihua Zhou , Zicong Hong , Song Guo

Recently, Mixture-of-Experts (MoE) has become one of the most popular techniques to scale pre-trained models to extraordinarily large sizes. Dynamic activation of experts allows for conditional computation, increasing the number of…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-06-30 Zheng Zhang , Donglin Yang , Yaqi Xia , Liang Ding , Dacheng Tao , Xiaobo Zhou , Dazhao Cheng

MoE-PEFT methods combine Mixture of Experts with parameter-efficient fine-tuning for multi-task adaptation, but require separate adapters per expert causing trainable parameters to scale linearly with expert count and limiting applicability…

Machine Learning · Computer Science 2026-04-06 Md Kowsher , Haris Mansoor , Nusrat Jahan Prottasha , Ozlem Garibay , Victor Zhu , Zhengping Ji , Chen Chen

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

Scaling large language models has driven remarkable advancements across various domains, yet the continual increase in model size presents significant challenges for real-world deployment. The Mixture of Experts (MoE) architecture offers a…

Machine Learning · Computer Science 2025-03-18 Shwai He , Daize Dong , Liang Ding , Ang Li

The parameter size of modern large language models (LLMs) can be scaled up via the sparsely-activated Mixture-of-Experts (MoE) technique to avoid excessive increase of the computational costs. To further improve training efficiency,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-08 Yunqi Gao , Bing Hu , Mahdi Boloursaz Mashhadi , A-Long Jin , Yanfeng Zhang , Pei Xiao , Rahim Tafazolli , Merouane Debbah

The sparsely activated mixture-of-experts (MoE) transformer has become a common architecture for large language models (LLMs) due to its sparsity, which requires fewer computational demands while easily scaling the model size. In MoE…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-14 Wenxiang Lin , Xinglin Pan , Lin Zhang , Shaohuai Shi , Xuan Wang , Xiaowen Chu

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

Mixture-of-Experts (MoE) has emerged as a promising architecture for modern large language models (LLMs). However, massive parameters impose heavy GPU memory (i.e., VRAM) demands, hindering the widespread adoption of MoE LLMs. Offloading…

Machine Learning · Computer Science 2025-09-11 Jiaming Yan , Jianchun Liu , Hongli Xu , Liusheng Huang

Mixture-of-Expert (MoE) presents a strong potential in enlarging the size of language model to trillions of parameters. However, training trillion-scale MoE requires algorithm and system co-design for a well-tuned high performance…

Machine Learning · Computer Science 2021-03-25 Jiaao He , Jiezhong Qiu , Aohan Zeng , Zhilin Yang , Jidong Zhai , Jie Tang

Multi-head attention is a driving force behind state-of-the-art transformers, which achieve remarkable performance across a variety of natural language processing (NLP) and computer vision tasks. It has been observed that for many…

Machine Learning · Computer Science 2022-06-14 Tam Nguyen , Tan M. Nguyen , Dung D. Le , Duy Khuong Nguyen , Viet-Anh Tran , Richard G. Baraniuk , Nhat Ho , Stanley J. Osher

Mixture-of-Experts (MoE) models substantially improve performance by increasing the capacity of dense architectures. However, directly training MoE models requires considerable computational resources and introduces extra overhead in…

Computer Vision and Pattern Recognition · Computer Science 2026-03-31 Jiacheng Ruan , Daize Dong , Xiaoye Qu , Tong Zhu , Ting Liu , Yuzhuo Fu , Yu Cheng , Suncheng Xiang

Mixture-of-Experts (MoE) architectures are widely used in modern large language models and multimodal models. However, inference efficiency is often limited by highly dynamic and skewed expert workloads across different modalities. During…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-12 Yingping Wang , Yi Wu , Xiangyu Wu , Junwei Cui , Weilin Cai , Zhijiang Guo , Jiayi Huang

Mixture-of-Experts is a promising approach for edge AI with low-batch inference. Yet, on-device deployments often face limited on-chip memory and severe workload imbalance; the prevalent use of offloading further incurs off-chip memory…

Hardware Architecture · Computer Science 2026-03-31 Songchen Ma , Hongyi Li , Weihao Zhang , Yonghao Tan , Pingcheng Dong , Yu Liu , Lan Liu , Yuzhong Jiao , Xuejiao Liu , Luhong Liang , Kwang-Ting Cheng

Mixture-of-Experts (MoE) models have gained popularity in achieving state-of-the-art performance in a wide range of tasks in computer vision and natural language processing. They effectively expand the model capacity while incurring a…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-06-21 Haiyang Huang , Newsha Ardalani , Anna Sun , Liu Ke , Hsien-Hsin S. Lee , Anjali Sridhar , Shruti Bhosale , Carole-Jean Wu , Benjamin Lee