English
Related papers

Related papers: Accelerating Mixture-of-Expert Inference with Adap…

200 papers

Mixture-of-Experts (MoE) models are typically pre-trained with explicit load-balancing constraints to ensure statistically balanced expert routing. Despite this, we observe that even well-trained MoE models exhibit significantly imbalanced…

Machine Learning · Computer Science 2026-01-27 Xuan-Phi Nguyen , Shrey Pandit , Austin Xu , Caiming Xiong , Shafiq Joty

The sparse Mixture-of-Experts (MoE) architecture is increasingly favored for scaling Large Language Models (LLMs) efficiently, but it depends on heterogeneous compute and memory resources. These factors jointly affect system Cost, Accuracy,…

The sparse Mixture-of-Experts (MoE) architecture is increasingly favored for scaling Large Language Models (LLMs) efficiently, but it depends on heterogeneous compute and memory resources. These factors jointly affect system Cost, Accuracy,…

Mixture-of-Experts (MoE) Large Language Models (LLMs) face a trilemma of load imbalance, parameter redundancy, and communication overhead. We introduce a unified framework based on dynamic expert clustering and structured compression to…

Computation and Language · Computer Science 2026-02-06 Peijun Zhu , Ning Yang , Baoliang Tian , Jiayu Wei , Weihao Zhang , Haijun Zhang , Pin Lv

We propose Chain-of-Experts (CoE), a new Mixture-of-Experts (MoE) architecture that introduces sequential expert communication within each layer. Unlike traditional MoE models, where experts operate independently in parallel, CoE processes…

Machine Learning · Computer Science 2025-06-25 Zihan Wang , Rui Pan , Jiarui Yao , Robert Csordas , Linjie Li , Lu Yin , Jiajun Wu , Tong Zhang , Manling Li , Shiwei Liu

Continually expanding new languages for existing large language models (LLMs) is a promising yet challenging approach to building powerful multilingual LLMs. The biggest challenge is to make the model continuously learn new languages while…

Computation and Language · Computer Science 2025-05-29 Xue Zhang , Yunlong Liang , Fandong Meng , Songming Zhang , Yufeng Chen , Jinan Xu , Jie Zhou

The deployment of mixture-of-experts (MoE) large language models (LLMs) presents significant challenges due to their high memory demands. These challenges become even more pronounced in multi-tenant environments, where shared resources must…

Machine Learning · Computer Science 2025-05-13 HamidReza Imani , Jiaxin Peng , Peiman Mohseni , Abdolah Amirany , Tarek El-Ghazawi

Vertical Federated Learning (VFL) has emerged as a critical paradigm for collaborative model training in privacy-sensitive domains such as finance and healthcare. However, most existing VFL frameworks rely on the idealized assumption of…

Machine Learning · Computer Science 2026-04-22 Jon Irureta , Gorka Azkune , Jon Imaz , Aizea Lojo , Javier Fernandez-Marques

Mixture-of-Experts (MoE) has become the de facto architecture for hundred-billion-parameter language models, yet its advantages at sub-billion scales for on-device deployment remain largely unexplored. To close this gap, we present…

Prevailing LLM serving engines employ expert parallelism (EP) to implement multi-device inference of massive MoE models. However, the efficiency of expert parallel inference is largely bounded by inter-device communication, as EP embraces…

Machine Learning · Computer Science 2026-03-02 Yan Li , Zhenyu Zhang , Zhengang Wang , Pengfei Chen , Pengfei Zheng

Mixture-of-Experts (MoE) models enable efficient scaling of large language models (LLMs) by activating only a subset of experts per input. However, we observe that the commonly used auxiliary load balancing loss often leads to expert…

Computation and Language · Computer Science 2026-01-27 Hongcan Guo , Haolang Lu , Guoshun Nan , Bolun Chu , Jialin Zhuang , Yuan Yang , Wenhao Che , Xinye Cao , Sicong Leng , Qimei Cui , Xudong Jiang

The Mixture of Experts (MoE) for language models has been proven effective in augmenting the capacity of models by dynamically routing each input token to a specific subset of experts for processing. Despite the success, most existing…

Machine Learning · Computer Science 2024-07-26 Hao Zhao , Zihan Qiu , Huijia Wu , Zili Wang , Zhaofeng He , Jie Fu

The interpretability of Mixture-of-Experts (MoE) models, especially those with heterogeneous designs, remains underexplored. Existing attribution methods for dense models fail to capture dynamic routing-expert interactions in sparse MoE…

Computation and Language · Computer Science 2025-06-12 Junzhuo Li , Bo Wang , Xiuze Zhou , Peijie Jiang , Jia Liu , Xuming Hu

The Mixture of Experts (MoE) is an effective architecture for scaling large language models by leveraging sparse expert activation to balance performance and efficiency. However, under expert parallelism, MoE suffers from inference…

Machine Learning · Computer Science 2026-05-12 Shwai He , Weilin Cai , Jiayi Huang , Ang Li

Large Language Models (LLMs) have demonstrated impressive performance across various tasks, and their application in edge scenarios has attracted significant attention. However, sparse-activated Mixture-of-Experts (MoE) models, which are…

Artificial Intelligence · Computer Science 2025-05-08 Zhiyuan Fang , Zicong Hong , Yuegui Huang , Yufeng Lyu , Wuhui Chen , Yue Yu , Fan Yu , Zibin Zheng

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

In this work, we aim to simultaneously enhance the effectiveness and efficiency of Mixture-of-Experts (MoE) methods. To achieve this, we propose MoE++, a general and heterogeneous MoE framework that integrates both Feed-Forward…

Machine Learning · Computer Science 2024-10-11 Peng Jin , Bo Zhu , Li Yuan , Shuicheng Yan

Large Mixture of Experts (MoE) models could achieve state-of-the-art quality on various language tasks, including machine translation task, thanks to the efficient model scaling capability with expert parallelism. However, it has brought a…

Machine Learning · Computer Science 2023-10-05 Young Jin Kim , Raffy Fahim , Hany Hassan Awadalla

Mixture-of-Experts (MoE) architectures have become the key to scaling modern LLMs, yet little is understood about how their sparse routing dynamics respond to multilingual data. In this work, we analyze expert routing patterns using…

Computation and Language · Computer Science 2026-02-19 Lucas Bandarkar , Chenyuan Yang , Mohsen Fayyaz , Junlin Hu , Nanyun Peng

Sparse Mixture-of-Experts (MoE) models enable efficient scalability by activating only a small sub-set of experts per input, yet their massive parameter counts lead to substantial memory and energy inefficiency during inference. Analog…

Machine Learning · Computer Science 2026-03-04 Mohammed Nowaz Rabbani Chowdhury , Hsinyu Tsai , Geoffrey W. Burr , Kaoutar El Maghraoui , Liu Liu , Meng Wang