English
Related papers

Related papers: Towards Greater Leverage: Scaling Laws for Efficie…

200 papers

An increasing number of LLMs employ Mixture-of-Experts (MoE) architectures where the feed-forward layer is replaced by a pool of experts and each token only activates a small subset of them. During autoregressive generation, these models…

Machine Learning · Computer Science 2025-11-05 Costin-Andrei Oncescu , Qingyang Wu , Wai Tong Chung , Robert Wu , Bryan Gopal , Junxiong Wang , Tri Dao , Ben Athiwaratkun

Mixture-of-Experts (MoE) architectures have emerged as a key strategy for scaling large language models (LLMs) efficiently. However, current MoE systems suffer from severe load imbalance, where only a small subset of experts is consistently…

Machine Learning · Computer Science 2025-06-27 Jiajie Yang

Large Language Models (LLMs) based on Mixture-of-Experts (MoE) are pivotal in industrial applications for their ability to scale performance efficiently. However, standard MoEs enforce uniform expert sizes,creating a rigidity that fails to…

Computation and Language · Computer Science 2026-04-29 Zhicheng Ma , Xiang Liu , Zhaoxiang Liu , Ning Wang , Yi Shen , Kai Wang , Shuming Shi , Shiguo Lian

The feedforward (FFW) layers in standard transformer architectures incur a linear increase in computational costs and activation memory as the hidden layer width grows. Sparse mixture-of-experts (MoE) architectures have emerged as a viable…

Machine Learning · Computer Science 2024-07-08 Xu Owen He

Mixture of Experts (MoE) LLMs have recently gained attention for their ability to enhance performance by selectively engaging specialized subnetworks or "experts" for each input. However, deploying MoEs on memory-constrained devices remains…

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

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

Mixture-of-Experts (MoE) is now the dominant architecture for frontier language models, yet it requires all expert parameters to be loaded in memory, making it less preferable for memory-constrained deployment. Existing compression methods…

Computation and Language · Computer Science 2026-05-28 Junhyuck Kim , Jihun Yun , Haechan Kim , Gyeongman Kim , Joonghyun Bae , Jaewoong Cho

Large language models (LLMs) encounter significant adaptation challenges in diverse multitask finetuning. Mixture-of-experts (MoE) provides a promising solution with a dynamic architecture, enabling effective task decoupling. However,…

Machine Learning · Computer Science 2025-05-28 Rongyu Zhang , Yijiang Liu , Huanrui Yang , Shenli Zheng , Dan Wang , Yuan Du , Li Du , Shanghang Zhang

In the era of Large Language Models (LLMs), the Mixture of Experts (MoE) architecture has emerged as an effective approach for training extremely large models with improved computational efficiency. This success builds upon extensive prior…

Computation and Language · Computer Science 2026-04-08 Giang Do , Hung Le , Truyen Tran

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

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

Mixture-of-Experts (MoE) shines brightly in large language models (LLMs) and demonstrates outstanding performance in plentiful natural language processing tasks. However, existing methods transforming LLMs from dense to MoE face significant…

Computation and Language · Computer Science 2024-10-04 Tingfeng Hui , Zhenyu Zhang , Shuohuan Wang , Yu Sun , Hua Wu , Sen Su

Expert parallelism is vital for effectively training Mixture-of-Experts (MoE) models, enabling different devices to host distinct experts, with each device processing different input data. However, during expert parallel training, dynamic…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-13 Xinyi Liu , Yujie Wang , Fangcheng Fu , Xuefeng Xiao , Huixia Li , Jiashi Li , Bin Cui

Sparsely Mixture of Experts (MoE) has received great interest due to its promising scaling capability with affordable computational overhead. MoE converts dense layers into sparse experts, and utilizes a gated routing network to make…

Computation and Language · Computer Science 2022-07-20 Yuan Xie , Shaohan Huang , Tianyu Chen , Furu Wei

Mixture-of-Experts (MoE) architectures enable efficient scaling of large language models by activating only a subset of parameters per input. However, existing MoE models suffer from two critical limitations: (1) inefficient token-to-expert…

Computation and Language · Computer Science 2025-10-10 Jing Li , Zhijie Sun , Dachao Lin , Xuan He , Binfan Zheng , Yi Lin , Rongqian Zhao , Xin Chen

Large language models (LLM) have been attracting much attention from the community recently, due to their remarkable performance in all kinds of downstream tasks. According to the well-known scaling law, scaling up a dense LLM enhances its…

Machine Learning · Computer Science 2025-02-19 Zhenpeng Su , Xing Wu , Zijia Lin , Yizhe Xiong , Minxuan Lv , Guangyuan Ma , Hui Chen , Songlin Hu , Guiguang Ding

Speculative decoding accelerates Large Language Model (LLM) inference by verifying multiple drafted tokens in parallel. However, for Mixture-of-Experts (MoE) models, this parallelism introduces a severe bottleneck: large draft trees…

Machine Learning · Computer Science 2026-02-19 Bradley McDanel , Steven Li , Sruthikesh Surineni , Harshit Khaitan

Mixture-of-experts networks (MoEs) have demonstrated remarkable efficiency in modern deep learning. Despite their empirical success, the theoretical foundations underlying their ability to model complex tasks remain poorly understood. In…

Machine Learning · Computer Science 2026-02-19 Mingze Wang , Weinan E

The emergence of large-scale Mixture of Experts (MoE) models represents a significant advancement in artificial intelligence, offering enhanced model capacity and computational efficiency through conditional computation. However, deploying…

Machine Learning · Computer Science 2025-01-23 Jiacheng Liu , Peng Tang , Wenfeng Wang , Yuhang Ren , Xiaofeng Hou , Pheng-Ann Heng , Minyi Guo , Chao Li