English
Related papers

Related papers: Fine-grained MoE Load Balancing with Linear Progra…

200 papers

The surgence of Mixture of Experts (MoE) in Large Language Models promises a small price of execution cost for a much larger model parameter count and learning capacity, because only a small fraction of parameters are activated for each…

By increasing model parameters but activating them sparsely when performing a task, the use of Mixture-of-Experts (MoE) architecture significantly improves the performance of Large Language Models (LLMs) without increasing the inference…

Computation and Language · Computer Science 2025-06-10 Zeliang Zhang , Xiaodong Liu , Hao Cheng , Chenliang Xu , Jianfeng Gao

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 Experts (MoE), an ensemble of specialized models equipped with a router that dynamically distributes each input to appropriate experts, has achieved successful results in the field of machine learning. However, theoretical…

Machine Learning · Computer Science 2025-08-19 Ryotaro Kawata , Kohsei Matsutani , Yuri Kinoshita , Naoki Nishikawa , Taiji Suzuki

We propose a novel adaptive Mixture-of-Experts (MoE) framework for time series forecasting that enhances expert specialization by incorporating expert-specific loss information directly into the training process. Notably, the overall…

Machine Learning · Statistics 2026-05-12 Btissame El Mahtout , Florian Ziel

For pre-training of MoE (Mixture-of-Experts) models, one of the main issues is unbalanced expert loads, which may cause routing collapse or increased computational overhead. Existing methods contain the Loss-Controlled method and the…

Machine Learning · Computer Science 2025-03-21 Yuan Sun

Mixture-of-experts (MoE) has been extensively employed to scale large language models to trillion-plus parameters while maintaining a fixed computational cost. The development of large MoE models in the distributed scenario encounters the…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-03-05 Shulai Zhang , Ningxin Zheng , Haibin Lin , Ziheng Jiang , Wenlei Bao , Chengquan Jiang , Qi Hou , Weihao Cui , Size Zheng , Li-Wen Chang , Quan Chen , Xin Liu

Large Language Models (LLMs) have achieved impressive results across various tasks, yet their high computational demands pose deployment challenges, especially on consumer-grade hardware. Mixture of Experts (MoE) models provide an efficient…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-12-19 En-Ming Huang , Li-Shang Lin , Chun-Yi Lee

Mixture-of-Experts (MoE) has emerged as an effective approach to reduce the computational overhead of Transformer architectures by sparsely activating a subset of parameters for each token while preserving high model capacity. This paradigm…

Computer Vision and Pattern Recognition · Computer Science 2026-03-31 Dohwan Ko , Jinyoung Park , Seoung Choi , Sanghyeok Lee , Seohyun Lee , Hyunwoo J. Kim

Merging various task-specific Transformer-based models trained on different tasks into a single unified model can execute all the tasks concurrently. Previous methods, exemplified by task arithmetic, have been proven to be both effective…

Machine Learning · Computer Science 2024-06-10 Anke Tang , Li Shen , Yong Luo , Nan Yin , Lefei Zhang , Dacheng Tao

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

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

Mixture of Experts (MoE) has emerged as a promising paradigm for scaling model capacity while preserving computational efficiency, particularly in large-scale machine learning architectures such as large language models (LLMs). Recent…

Networking and Internet Architecture · Computer Science 2026-01-29 Yunting Xu , Jiacheng Wang , Ruichen Zhang , Changyuan Zhao , Dusit Niyato , Jiawen Kang , Zehui Xiong , Bo Qian , Haibo Zhou , Shiwen Mao , Abbas Jamalipour , Xuemin Shen , Dong In Kim

Supervised fine-tuning (SFT) is a milestone in aligning large language models with human instructions and adapting them to downstream tasks. In particular, Low-Rank Adaptation (LoRA) has gained widespread attention due to its parameter…

Computation and Language · Computer Science 2025-11-05 Jia-Chen Zhang , Yu-Jie Xiong , Xi-He Qiu , Chun-Ming Xia , Fei Dai , Zheng Zhou

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

The advent of Large Language Models (LLMs) has ushered in a new era of artificial intelligence, with the potential to transform various sectors through automation and insightful analysis. The Mixture of Experts (MoE) architecture has been…

Machine Learning · Computer Science 2024-10-22 Xurui Li , Juanjuan Yao

Load imbalance is a long-standing challenge in Mixture-of-Experts (MoE) training and is exacerbated in reinforcement learning (RL) for LLMs, where hot experts can shift frequently across micro-batches. Existing MoE training systems rely on…

Machine Learning · Computer Science 2026-05-12 Chao Jin , Xinming Wei , Yinmin Zhong , Chengxu Yang , Bingyang Wu , Ruidong Zhu , Zili Zhang , Yuliang Liu , Xin Jin

Mixture-of-Experts (MoE) based Large Language Models (LLMs) have demonstrated impressive performance and computational efficiency. However, their deployment is often constrained by substantial memory demands, primarily due to the need to…

Machine Learning · Computer Science 2026-03-16 Jiawei Hao , Zhiwei Hao , Jianyuan Guo , Li Shen , Yong Luo , Han Hu , Dan Zeng

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

The Mixture-of-Experts (MoE) architecture has emerged as a promising approach to mitigate the rising computational costs of large language models (LLMs) by selectively activating parameters. However, its high memory requirements and…

Artificial Intelligence · Computer Science 2026-04-14 Jehyeon Bang , Eunyeong Cho , Ranggi Hwang , Jinha Chung , Minsoo Rhu