English
Related papers

Related papers: Cross-Platform Fused MoE Dispatch in Triton: Porta…

200 papers

Mixture of Experts (MoE) models enhance neural network scalability by dynamically selecting relevant experts per input token, enabling larger model sizes while maintaining manageable computation costs. However, efficient training of…

The Mixture-of-Experts (MoE) architecture is crucial for scaling large language models, but its scalability is severely limited by inter-GPU communication bottlenecks in multi-GPU systems. Although overlapping communication with computation…

Hardware Architecture · Computer Science 2026-05-08 Zhuoshan Zhou , Chen Zhang , Shuyi Zhang , Qijun Zhang , Haibo Wang , Zhe Zhou , Zhipeng Tu , Guangyu Sun , Yijia Diao , Zhigang Ji , Jingwen Leng , Guanghui He , Minyi Guo

Mixture-of-Experts (MoE) has emerged as a powerful paradigm for scaling model capacity while preserving computational efficiency. Despite its notable success in large language models (LLMs), existing attempts to apply MoE to Diffusion…

Computer Vision and Pattern Recognition · Computer Science 2026-03-03 Yujie Wei , Shiwei Zhang , Hangjie Yuan , Yujin Han , Zhekai Chen , Jiayu Wang , Difan Zou , Xihui Liu , Yingya Zhang , Yu Liu , Hongming Shan

Mixture of Experts (MoE) models have emerged as the de facto architecture for scaling up language models without significantly increasing the computational cost. Recent MoE models demonstrate a clear trend towards high expert granularity…

Machine Learning · Computer Science 2026-03-30 Wentao Guo , Mayank Mishra , Xinle Cheng , Ion Stoica , Tri Dao

Mixture-of-Experts (MoE) model architecture has emerged as a promising solution for scaling transformer models efficiently, offering sparse activation that reduces computational costs while increasing model capacity. However, as MoE models…

Machine Learning · Computer Science 2025-02-11 Seokjin Go , Divya Mahajan

We present MegaBlocks, a system for efficient Mixture-of-Experts (MoE) training on GPUs. Our system is motivated by the limitations of current frameworks, which restrict the dynamic routing in MoE layers to satisfy the constraints of…

Machine Learning · Computer Science 2022-11-30 Trevor Gale , Deepak Narayanan , Cliff Young , Matei Zaharia

Sparsely activated Mixture-of-Experts (MoE) models are widely adopted to scale up model capacity without increasing the computation budget. However, vanilla TopK routers are trained in a discontinuous, non-differentiable way, limiting their…

Machine Learning · Computer Science 2025-02-28 Ziteng Wang , Jun Zhu , Jianfei Chen

The computational sparsity of Mixture-of-Experts (MoE) models enables sub-linear growth in compute cost as model size increases, thus offering a scalable path to training massive neural networks. However, existing implementations suffer…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-11 Osayamen Jonathan Aimuyo , Byungsoo Oh , Rachee Singh

Scaling Mixture-of-Experts (MoE) training introduces systems challenges absent in dense models. Because each token activates only a subset of experts, this sparsity allows total parameters to grow much faster than per-token computation,…

The Mixture of Experts (MoE) architecture has emerged as a key technique for scaling Large Language Models by activating only a subset of experts per query. Deploying MoE on consumer-grade edge hardware, however, is constrained by limited…

Artificial Intelligence · Computer Science 2026-05-05 Guoying Zhu , Meng Li , Haipeng Dai , Xuechen Liu , Weijun Wang , Keran Li , Jun xiao , Ligeng Chen , Wei Wang

Sparsely gated Mixture-of-Expert (MoE) has demonstrated its effectiveness in scaling up deep neural networks to an extreme scale. Despite that numerous efforts have been made to improve the performance of MoE from the model design or system…

Machine Learning · Computer Science 2023-02-21 Chang Chen , Min Li , Zhihua Wu , Dianhai Yu , Chao Yang

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

Mixture of experts (MoE) architectures have become a cornerstone for scaling up and are a key component in most large language models such as GPT-OSS, DeepSeek-V3, Llama-4, and Gemini-2.5. However, systematic research on MoE remains…

Computation and Language · Computer Science 2026-02-11 Nam V. Nguyen , Thong T. Doan , Luong Tran , Van Nguyen , Quang Pham

As giant dense models advance quality but require large amounts of GPU budgets for training, the sparsely gated Mixture-of-Experts (MoE), a kind of conditional computation architecture, is proposed to scale models while keeping their…

Distributed, Parallel, and Cluster Computing · Computer Science 2022-11-18 Xiaonan Nie , Pinxue Zhao , Xupeng Miao , Tong Zhao , Bin Cui

Mixture-of-Experts (MoE) activates only a subset of experts during inference, allowing the model to maintain low inference FLOPs and latency even as the parameter count scales up. However, since MoE dynamically selects the experts, all the…

Machine Learning · Computer Science 2025-05-27 Shibo Jie , Yehui Tang , Kai Han , Yitong Li , Duyu Tang , Zhi-Hong Deng , Yunhe Wang

In recent years, Mixture-of-Experts (MoE) has emerged as an effective approach for enhancing the capacity of deep neural network (DNN) with sub-linear computational costs. However, storing all experts on GPUs incurs significant memory…

Machine Learning · Computer Science 2025-03-11 Suraiya Tairin , Shohaib Mahmud , Haiying Shen , Anand Iyer

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

Emerging expert-specialized Mixture-of-Experts (MoE) architectures, such as DeepSeek-MoE, deliver strong model quality through fine-grained expert segmentation and large top-k routing. However, their scalability is limited by substantial…

Machine Learning · Computer Science 2025-08-20 Yueming Yuan , Ahan Gupta , Jianping Li , Sajal Dash , Feiyi Wang , Minjia Zhang

The Mixtures-of-Experts (MoE) model is a widespread distributed and integrated learning method for large language models (LLM), which is favored due to its ability to sparsify and expand models efficiently. However, the performance of MoE…

Machine Learning · Computer Science 2024-05-24 Jing Li , Zhijie Sun , Xuan He , Li Zeng , Yi Lin , Entong Li , Binfan Zheng , Rongqian Zhao , Xin Chen

The Mixture of Experts (MoE) architecture has become a fundamental building block in state-of-the-art large language models (LLMs), improving domain-specific expertise in LLMs and scaling model capacity without proportionally increasing…

Machine Learning · Computer Science 2026-05-13 Ankit Jyothish , Ali Jannesari , Aishwarya Sarkar , Joseph Zuber
‹ Prev 1 2 3 10 Next ›