English
Related papers

Related papers: MoNTA: Accelerating Mixture-of-Experts Training wi…

200 papers

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 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

We present MoE-MLA-RoPE, a novel architecture combination that combines Mixture of Experts (MoE) with Multi-head Latent Attention (MLA) and Rotary Position Embeddings (RoPE) for efficient language modeling. Our approach addresses the…

Artificial Intelligence · Computer Science 2025-08-05 Sushant Mehta , Raj Dandekar , Rajat Dandekar , Sreedath Panat

The sparse Mixture-of-Experts (MoE) model is powerful for large-scale pre-training and has achieved promising results due to its model capacity. However, with trillions of parameters, MoE is hard to be deployed on cloud or mobile…

Machine Learning · Computer Science 2022-06-03 Tianyu Chen , Shaohan Huang , Yuan Xie , Binxing Jiao , Daxin Jiang , Haoyi Zhou , Jianxin Li , Furu Wei

We introduce a Mixture of Raytraced Experts, a stacked Mixture of Experts (MoE) architecture which can dynamically select sequences of experts, producing computational graphs of variable width and depth. Existing MoE architectures generally…

Machine Learning · Computer Science 2025-07-17 Andrea Perin , Giacomo Lagomarsini , Claudio Gallicchio , Giuseppe Nuti

Mixture-of-Experts (MoE) models substantially improve performance by increasing the capacity of dense architectures. However, directly training MoE models requires considerable computational resources and introduces extra overhead in…

Computer Vision and Pattern Recognition · Computer Science 2026-03-31 Jiacheng Ruan , Daize Dong , Xiaoye Qu , Tong Zhu , Ting Liu , Yuzhuo Fu , Yu Cheng , Suncheng Xiang

Recently, mixture of experts (MoE) has become a popular paradigm for achieving the trade-off between modal capacity and efficiency of multi-modal large language models (MLLMs). Different from previous efforts, we are dedicated to exploring…

Multimedia · Computer Science 2025-02-13 Qiong Wu , Zhaoxi Ke , Yiyi Zhou , Xiaoshuai Sun , Rongrong Ji

The Mixture-of-Experts (MoE) architecture has become increasingly popular as a method to scale up large language models (LLMs). To save costs, heterogeneity-aware training solutions have been proposed to utilize GPU clusters made up of both…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-08 Yongji Wu , Xueshen Liu , Shuowei Jin , Ceyu Xu , Feng Qian , Z. Morley Mao , Matthew Lentz , Danyang Zhuo , Ion Stoica

Mixture-of-Experts (MoE) has emerged as a prominent architecture for scaling model size while maintaining computational efficiency. In MoE, each token in the input sequence activates a different subset of experts determined by a routing…

Computation and Language · Computer Science 2024-11-05 Chufan Shi , Cheng Yang , Xinyu Zhu , Jiahao Wang , Taiqiang Wu , Siheng Li , Deng Cai , Yujiu Yang , Yu Meng

Scaling large language models has driven remarkable advancements across various domains, yet the continual increase in model size presents significant challenges for real-world deployment. The Mixture of Experts (MoE) architecture offers a…

Machine Learning · Computer Science 2025-03-18 Shwai He , Daize Dong , Liang Ding , Ang Li

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) models can achieve promising results with outrageous large amount of parameters but constant computation cost, and thus it has become a trend in model scaling. Still it is a mystery how MoE layers bring quality…

Machine Learning · Computer Science 2021-08-10 An Yang , Junyang Lin , Rui Men , Chang Zhou , Le Jiang , Xianyan Jia , Ang Wang , Jie Zhang , Jiamang Wang , Yong Li , Di Zhang , Wei Lin , Lin Qu , Jingren Zhou , Hongxia Yang

Expert Parallelism (EP) permits Mixture of Experts (MoE) models to scale beyond a single GPU. To address load imbalance across GPUs in EP, existing approaches aim to balance the number of tokens each GPU processes. Surprisingly, we find…

Large Language Models (LLMs) with Mixture-of-Expert (MoE) architectures achieve superior model performance with reduced computation costs, but at the cost of high memory capacity and bandwidth requirements. Near-Memory Processing (NMP)…

Performance · Computer Science 2025-09-12 Haochen Huang , Shuzhang Zhong , Zhe Zhang , Shuangchen Li , Dimin Niu , Hongzhong Zheng , Runsheng Wang , Meng Li

In this paper, we aim to build a robust question answering system that can adapt to out-of-domain datasets. A single network may overfit to the superficial correlation in the training distribution, but with a meaningful number of expert…

Computation and Language · Computer Science 2022-04-21 Yu Qing Zhou , Xixuan Julie Liu , Yuanzhe Dong

Mixture-of-experts (MoE) is becoming popular due to its success in improving the model quality, especially in Transformers. By routing tokens with a sparse gate to a few experts (i.e., a small pieces of the full model), MoE can easily…

Machine Learning · Computer Science 2022-10-11 Xiaonan Nie , Xupeng Miao , Shijie Cao , Lingxiao Ma , Qibin Liu , Jilong Xue , Youshan Miao , Yi Liu , Zhi Yang , Bin Cui

Pre-training has proven effective in addressing data scarcity and performance limitations in solving PDE problems with neural operators. However, challenges remain due to the heterogeneity of PDE datasets in equation types, which leads to…

Machine Learning · Computer Science 2025-11-03 Hong Wang , Haiyang Xin , Jie Wang , Xuanze Yang , Fei Zha , Huanshuo Dong , Yan Jiang

The Mixture-of-Experts (MoE) model has emerged as a prominent architecture in the field of Large Language Models (LLMs), providing a better balance between model performance and computational efficiency. However the General Matrix Multiply…

Computation and Language · Computer Science 2025-01-06 Yulei Qian , Fengcun Li , Xiangyang Ji , Xiaoyu Zhao , Jianchao Tan , Kefeng Zhang , Xunliang Cai

Mixture of Experts (MoE) has become a key architectural paradigm for efficiently scaling Large Language Models (LLMs) by selectively activating a subset of parameters for each input token. However, standard MoE architectures face…

Machine Learning · Computer Science 2025-05-27 Zehua Liu , Han Wu , Ruifeng She , Xiaojin Fu , Xiongwei Han , Tao Zhong , Mingxuan Yuan

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