English
Related papers

Related papers: FlashMoE: Fast Distributed MoE in a Single Kernel

200 papers

Recently, Mixture of Experts (MoE) based Transformer has shown promising results in many domains. This is largely due to the following advantages of this architecture: firstly, MoE based Transformer can increase model capacity without…

Sound · Computer Science 2021-05-10 Zhao You , Shulin Feng , Dan Su , Dong Yu

Multimodal learning has gained increasing importance across various fields, offering the ability to integrate data from diverse sources such as images, text, and personalized records, which are frequently observed in medical domains.…

Machine Learning · Computer Science 2024-11-01 Sukwon Yun , Inyoung Choi , Jie Peng , Yangfan Wu , Jingxuan Bao , Qiyiwen Zhang , Jiayi Xin , Qi Long , Tianlong Chen

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

Recent large language models such as Gemini-1.5, DeepSeek-V3, and Llama-4 increasingly adopt Mixture-of-Experts (MoE) architectures, which offer strong efficiency-performance trade-offs by activating only a fraction of the model per token.…

Computation and Language · Computer Science 2025-05-27 Hao Kang , Zichun Yu , Chenyan Xiong

The Sparsely-Activated Mixture-of-Experts (MoE) has gained increasing popularity for scaling up large language models (LLMs) without exploding computational costs. Despite its success, the current design faces a challenge where all experts…

Machine Learning · Computer Science 2024-09-20 Manxi Sun , Wei Liu , Jian Luan , Pengzhi Gao , Bin Wang

Federated learning (FL) is a collaborative machine learning approach that enables multiple clients to train models without sharing their private data. With the rise of deep learning, large-scale models have garnered significant attention…

Machine Learning · Computer Science 2025-09-09 Ziwei Zhan , Wenkuan Zhao , Yuanqing Li , Weijie Liu , Xiaoxi Zhang , Chee Wei Tan , Chuan Wu , Deke Guo , Xu Chen

Scaling the size of a model enhances its capabilities but significantly increases computation complexity. Mixture-of-Experts models (MoE) address the issue by allowing model size to scale up without substantially increasing training or…

Computation and Language · Computer Science 2024-08-30 Zhenpeng Su , Zijia Lin , Xue Bai , Xing Wu , Yizhe Xiong , Haoran Lian , Guangyuan Ma , Hui Chen , Guiguang Ding , Wei Zhou , Songlin Hu

Mixture-of-Experts is a promising approach for edge AI with low-batch inference. Yet, on-device deployments often face limited on-chip memory and severe workload imbalance; the prevalent use of offloading further incurs off-chip memory…

Hardware Architecture · Computer Science 2026-03-31 Songchen Ma , Hongyi Li , Weihao Zhang , Yonghao Tan , Pingcheng Dong , Yu Liu , Lan Liu , Yuzhong Jiao , Xuejiao Liu , Luhong Liang , Kwang-Ting Cheng

The Sparse Mixture of Experts (SMoE) has been widely employed to enhance the efficiency of training and inference for Transformer-based foundational models, yielding promising results.However, the performance of SMoE heavily depends on the…

Machine Learning · Computer Science 2025-03-11 Yongxin Guo , Zhenglin Cheng , Xiaoying Tang , Zhaopeng Tu , Tao Lin

Artificial intelligence (AI) has achieved astonishing successes in many domains, especially with the recent breakthroughs in the development of foundational large models. These large models, leveraging their extensive training data, provide…

Machine Learning · Computer Science 2026-01-27 Siyuan Mu , Sen Lin

Mixture-of-Experts (MoE) showcases tremendous potential to scale large language models (LLMs) with enhanced performance and reduced computational complexity. However, its sparsely activated architecture shifts feed-forward networks (FFNs)…

Recent advancements in Mixture of Experts (MoE) models have significantly increased their parameter scale as well as model performance. Extensive offloading techniques have been proposed to address the GPU memory limitations of MoE…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-03 Zhibin Wang , Zhonghui Zhang , Yuhang Zhou , Zibo Wang , Mo Zhou , Peng Jiang , Weilin Cai , Chengying Huan , Rong Gu , Sheng Zhong , Chen Tian

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

In the era of large language models, Mixture-of-Experts (MoE) is a promising architecture for managing computational costs when scaling up model parameters. However, conventional MoE architectures like GShard, which activate the top-$K$ out…

Mixture-of-Experts (MoE) is a promising way to scale up the learning capacity of large language models. It increases the number of parameters while keeping FLOPs nearly constant during inference through sparse activation. Yet, it still…

Machine Learning · Computer Science 2025-02-26 Pingzhi Li , Xiaolong Jin , Zhen Tan , Yu Cheng , Tianlong Chen

The immense memory requirements of state-of-the-art Mixture-of-Experts (MoE) models present a significant challenge for inference, often exceeding the capacity of a single accelerator. While offloading experts to host memory is a common…

Machine Learning · Computer Science 2025-11-19 Wenfeng Wang , Jiacheng Liu , Xiaofeng Hou , Xinfeng Xia , Peng Tang , Mingxuan Zhang , Chao Li , Minyi Guo

To deploy large Mixture-of-Experts (MoE) models cost-effectively, offloading-based single-GPU heterogeneous inference is crucial. While GPU-CPU architectures that offload cold experts are constrained by host memory bandwidth, emerging…

Hardware Architecture · Computer Science 2026-03-03 Yudong Pan , Yintao He , Tianhua Han , Lian Liu , Shixin Zhao , Zhirong Chen , Mengdi Wang , Cangyuan Li , Yinhe Han , Ying Wang

The proliferation of large language models (LLMs) has driven the adoption of Mixture-of-Experts (MoE) architectures as a promising solution to scale model capacity while controlling computational costs. However, deploying MoE models in…

Networking and Internet Architecture · Computer Science 2025-08-14 Muqing Li , Ning Li , Xin Yuan , Wenchao Xu , Quan Chen , Song Guo , Haijun Zhang

Sparse Mixture of Experts (SMoEs) models scale the capacity of models while maintaining constant computational overhead. Early designs typically relied on a fixed value of $k$, where $k$ represents either the number of experts selected per…

Computation and Language · Computer Science 2025-10-28 Giang Do , Hung Le , Truyen Tran

Traditional Mixture-of-Experts (MoE) networks benefit from utilizing multiple smaller expert models as opposed to a single large network. However, these experts typically operate independently, leaving a question open about whether…

Computation and Language · Computer Science 2026-01-06 Bo Lv , Chen Tang , Zifan Zheng , Bohao Yang , Kun Zhao , Ning Liao , Xiaoxing Wang , Feiyu Xiong , Zhiyu Li , Nayu Liu , Jingchi Jiang