English
Related papers

Related papers: Scattered Mixture-of-Experts Implementation

200 papers

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

The expansion of large language models is increasingly limited by the constrained memory capacity of modern GPUs. To mitigate this, Mixture-of-Experts (MoE) architectures activate only a small portion of parameters during inference,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-31 Zixu Shen , Kexin Chu , Yifan Zhang , Dawei Xiang , Runxin Wu , Wei Zhang

It is widely acknowledged that the performance of Transformer models is logarithmically related to their number of parameters and computational complexity. While approaches like Mixture of Experts (MoE) decouple parameter count from…

Machine Learning · Computer Science 2025-02-07 Zihao Huang , Qiyang Min , Hongzhi Huang , Defa Zhu , Yutao Zeng , Ran Guo , Xun Zhou

Mixture-of-Experts (MoE) models promise efficient scaling of large language models (LLMs) by activating only a small subset of experts per token, but their parallelized inference pipelines make elastic serving challenging. Existing…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-06 Gursimran Singh , Timothy Yu , Haley Li , Cheng Chen , Hanieh Sadri , Qintao Zhang , Yu Zhang , Ying Xiong , Yong Zhang , Zhenan Fan

The rapid advancement of large language models (LLMs) has led to architectures with billions to trillions of parameters, posing significant deployment challenges due to their substantial demands on memory, processing power, and energy…

Machine Learning · Computer Science 2024-07-02 Enshu Liu , Junyi Zhu , Zinan Lin , Xuefei Ning , Matthew B. Blaschko , Shengen Yan , Guohao Dai , Huazhong Yang , Yu Wang

Mixture-of-Experts (MoE) has been gaining popularity due to its successful adaptation to large language models (LLMs). In this work, we introduce Privacy-preserving Collaborative Mixture-of-Experts (PC-MoE), which leverages the sparsity of…

Machine Learning · Computer Science 2025-06-05 Ze Yu Zhang , Bolin Ding , Bryan Kian Hsiang Low

Real-world model deployment across multiple domains requires multimodal models to operate under two complementary regimes: (1) multi-task pretraining, tasks are co-available at design time where related tasks could borrow representational…

Machine Learning · Computer Science 2026-05-12 Xing Han , Shravan Chaudhari , Tanvi Ranade , Rama Chellappa , Suchi Saria

Mixture of Experts (MoEs) have become a central component of many state-of-the-art open-source and proprietary large language models. Despite their widespread adoption, it remains unclear how close existing MoE architectures are to optimal…

Sparse mixture of experts (SMoE) offers an appealing solution to scale up the model complexity beyond the mean of increasing the network's depth or width. However, effective training of SMoE has proven to be challenging due to the…

Sparse Mixture-of-Experts (MoE) architectures route each token through a subset of experts at each layer independently. We propose viewing MoE computation through the lens of \emph{expert paths} -- the sequence of expert selections a token…

Machine Learning · Computer Science 2026-04-07 Zijin Gu , Tatiana Likhomanenko , Vimal Thilak , Jason Ramapuram , Navdeep Jaitly

Mixture-of-Experts (MoE) layers activate a subset of model weights, dubbed experts, to improve model performance. MoE is particularly promising for deployment on process-in-memory (PIM) architectures, because PIM can naturally fit experts…

Hardware Architecture · Computer Science 2026-02-12 Hanyuan Gao , Xiaoxuan Yang

Mixture-of-Experts (MoE), a conditional computation architecture, achieved promising performance by scaling local module (i.e. feed-forward network) of transformer. However, scaling the cross-token module (i.e. self-attention) is…

Machine Learning · Computer Science 2022-01-17 Yuxuan Lou , Fuzhao Xue , Zangwei Zheng , Yang You

Sparse Mixture of Expert (MoE) models are popular foundational architectures at large scale, however, under-explored at smaller sizes. Here, we show how to enable Compact Sparse Mixture of Experts (CoSMoEs) for on-device inference.…

Machine Learning · Computer Science 2025-03-04 Patrick Huber , Akshat Shrivastava , Ernie Chang , Chinnadhurai Sankar , Ahmed Aly , Adithya Sagar

Mixture-of-Experts (MoE) is a neural network architecture that adds sparsely activated expert blocks to a base model, increasing the number of parameters without impacting computational costs. However, current distributed deep learning…

Machine Learning · Computer Science 2023-05-16 Siddharth Singh , Olatunji Ruwase , Ammar Ahmad Awan , Samyam Rajbhandari , Yuxiong He , Abhinav Bhatele

Mixture-of-Experts (MoE) architectures scale language models by activating only a subset of specialized expert networks for each input token, thereby reducing the number of floating-point operations. However, the growing size of modern MoE…

Machine Learning · Computer Science 2025-11-14 Yun Wang , Lingyun Yang , Senhao Yu , Yixiao Wang , Ruixing Li , Zhixiang Wei , James Yen , Zhengwei Qi

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

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) model becomes an important choice of large language models nowadays because of its scalability with sublinear computational complexity for training and inference. However, existing MoE models suffer from two…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-04-25 Xin Chen , Hengheng Zhang , Xiaotao Gu , Kaifeng Bi , Lingxi Xie , Qi Tian

Mixture-of-Experts (MoE) models have become a dominant paradigm for scaling large language models, but their rapidly growing parameter sizes introduce a fundamental inefficiency during inference: most expert weights remain idle in GPU…

Machine Learning · Computer Science 2026-05-01 Qingxiu Liu , Cyril Y. He , Hanser Jiang , Zion Wang , Alan Zhao , Patrick P. C. Lee

Mixture of Experts (MoE) has become a mainstream architecture for building Large Language Models (LLMs) by reducing per-token computation while enabling model scaling. It can be viewed as partitioning a large Feed-Forward Network (FFN) at…

Machine Learning · Computer Science 2025-08-27 Weilin Cai , Le Qin , Shwai He , Junwei Cui , Ang Li , Jiayi Huang