English
Related papers

Related papers: MegaBlocks: Efficient Sparse Training with Mixture…

200 papers

Mixture-of-Experts (MoE) models face deployment challenges due to their large parameter counts and computational demands. We explore quantization for MoE models and highlight two key insights: 1) linear blocks exhibit varying quantization…

Machine Learning · Computer Science 2025-05-12 Haojie Duanmu , Xiuhong Li , Zhihang Yuan , Size Zheng , Jiangfei Duan , Xingcheng Zhang , Dahua Lin

The rapid advancement of large language models (LLMs) has led to architectures with billions to trillions of parameters, posing significant deployment challenges due to their substantial demands on memory, processing power, and energy…

Machine Learning · Computer Science 2024-07-02 Enshu Liu , Junyi Zhu , Zinan Lin , Xuefei Ning , Matthew B. Blaschko , Shengen Yan , Guohao Dai , Huazhong Yang , Yu Wang

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

Large Language Models (LLMs) have achieved remarkable success across many applications, with Mixture of Experts (MoE) models demonstrating great potential. Compared to traditional dense models, MoEs achieve better performance with less…

Machine Learning · Computer Science 2026-02-17 Zongle Huang , Lei Zhu , Zongyuan Zhan , Ting Hu , Weikai Mao , Xianzhi Yu , Yongpan Liu , Tianyu Zhang

Mixture of Experts (MoEs) have become a central component of many state-of-the-art open-source and proprietary large language models. Despite their widespread adoption, it remains unclear how close existing MoE architectures are to optimal…

While Mixture-of-Experts (MoE) scales model capacity without proportionally increasing computation, its massive total parameter footprint creates significant storage and memory-access bottlenecks, which hinder efficient end-side deployment…

Machine Learning · Computer Science 2026-05-21 Chenyang Song , Weilin Zhao , Xu Han , Chaojun Xiao , Yingfa Chen , Zhiyuan Liu

The Mixture of Experts (MoE) architecture reduces the training and inference cost significantly compared to a dense model of equivalent capacity. Upcycling is an approach that initializes and trains an MoE model using a pre-trained dense…

Computation and Language · Computer Science 2025-03-18 Taishi Nakamura , Takuya Akiba , Kazuki Fujii , Yusuke Oda , Rio Yokota , Jun Suzuki

Mixture of Experts (MoE) offers remarkable performance and computational efficiency by selectively activating subsets of model parameters. Traditionally, MoE models use homogeneous experts, each with identical capacity. However, varying…

Computation and Language · Computer Science 2024-08-21 An Wang , Xingwu Sun , Ruobing Xie , Shuaipeng Li , Jiaqi Zhu , Zhen Yang , Pinxue Zhao , J. N. Han , Zhanhui Kang , Di Wang , Naoaki Okazaki , Cheng-zhong Xu

Sparse Mixture-of-Experts (MoE) allows scaling of language and vision models efficiently by activating only a small subset of experts per input. While this reduces computation, the large number of parameters still incurs substantial memory…

Mixtures of Experts (MoEs) have gained prominence in (self-)supervised learning due to their enhanced inference efficiency, adaptability to distributed training, and modularity. Previous research has illustrated that MoEs can significantly…

Machine Learning · Computer Science 2024-06-27 Timon Willi , Johan Obando-Ceron , Jakob Foerster , Karolina Dziugaite , Pablo Samuel Castro

Scaling the size of language models usually leads to remarkable advancements in NLP tasks. But it often comes with a price of growing computational cost. Although a sparse Mixture of Experts (MoE) can reduce the cost by activating a small…

Computation and Language · Computer Science 2023-11-23 Shwai He , Run-Ze Fan , Liang Ding , Li Shen , Tianyi Zhou , Dacheng Tao

As an enabling architecture of Large Models (LMs), Mixture of Experts (MoE) has become prevalent thanks to its sparsely-gated mechanism, which lowers computational overhead while maintaining learning performance comparable to dense LMs. The…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-17 Weihao Zhu , Long Shi , Kang Wei , Zhen Mei , Zhe Wang , Jiaheng Wang , Jun Li

Mixture-of-Experts (MoE) represents an ensemble methodology that amalgamates predictions from several specialized sub-models (referred to as experts). This fusion is accomplished through a router mechanism, dynamically assigning weights to…

Machine Learning · Computer Science 2024-03-27 Jinze Zhao , Peihao Wang , Zhangyang Wang

Traditional multi-task learning (MTL) methods use dense networks that use the same set of shared weights across several different tasks. This often creates interference where two or more tasks compete to pull model parameters in different…

The Mixture of Experts architecture allows for outrageously large neural networks by scaling model parameter size independently from computational demand (FLOPs). However, current DNN frameworks cannot effectively support the dynamic data…

Machine Learning · Computer Science 2022-08-03 Ferdinand Kossmann , Zhihao Jia , Alex Aiken

We propose Tensor-Trained Low-Rank Adaptation Mixture of Experts (TT-LoRA MoE), a novel computational framework integrating Parameter-Efficient Fine-Tuning (PEFT) with sparse MoE routing to address scalability challenges in large model…

Machine Learning · Computer Science 2026-01-27 Pradip Kunwar , Minh N. Vu , Maanak Gupta , Mahmoud Abdelsalam , Manish Bhattarai

Fine-grained, per-micro-batch load balancing is essential for efficient Mixture-of-Experts (MoE) training, yet every prior dynamic scheduling scheme pays for it with extra communication that is hard to hide. Especially on modern…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-22 Shuyao Qi , Haoyuan Liu , Shizhen Zhao

Mixture of Experts (MoE) is able to scale up vision transformers effectively. However, it requires prohibiting computation resources to train a large MoE transformer. In this paper, we propose Residual Mixture of Experts (RMoE), an…

Computer Vision and Pattern Recognition · Computer Science 2022-10-05 Lemeng Wu , Mengchen Liu , Yinpeng Chen , Dongdong Chen , Xiyang Dai , Lu Yuan

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

Efficient deployment of large language models, particularly Mixture of Experts (MoE), on resource-constrained platforms presents significant challenges, especially in terms of computational efficiency and memory utilization. The MoE…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-11-20 Shiyi Cao , Shu Liu , Tyler Griggs , Peter Schafhalter , Xiaoxuan Liu , Ying Sheng , Joseph E. Gonzalez , Matei Zaharia , Ion Stoica