English
Related papers

Related papers: FastMoE: A Fast Mixture-of-Expert Training System

200 papers

Mixture-of-Experts (MoE) models have become a widely-adopted solution to continue scaling model sizes without a corresponding linear increase in compute. During MoE model training, each input token is dynamically routed to a subset of…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-20 Athinagoras Skiadopoulos , Mark Zhao , Swapnil Gandhi , Thomas Norrie , Shrijeet Mukherjee , Christos Kozyrakis

Mixture-of-Experts (MoE) models face deployment challenges due to their large parameter counts and computational demands. We explore quantization for MoE models and highlight two key insights: 1) linear blocks exhibit varying quantization…

Machine Learning · Computer Science 2025-05-12 Haojie Duanmu , Xiuhong Li , Zhihang Yuan , Size Zheng , Jiangfei Duan , Xingcheng Zhang , Dahua Lin

Sparse models, including sparse Mixture-of-Experts (MoE) models, have emerged as an effective approach for scaling Transformer models. However, they often suffer from computational inefficiency since a significant number of parameters are…

Machine Learning · Computer Science 2024-05-27 Yuanhang Yang , Shiyi Qi , Wenchao Gu , Chaozheng Wang , Cuiyun Gao , Zenglin Xu

Mixture-of-experts (MoE) architecture has been proven a powerful method for diverse tasks in training deep models in many applications. However, current MoE implementations are task agnostic, treating all tokens from different tasks in the…

Computation and Language · Computer Science 2023-10-26 Hai Pham , Young Jin Kim , Subhabrata Mukherjee , David P. Woodruff , Barnabas Poczos , Hany Hassan Awadalla

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…

End-to-end models with large capacity have significantly improved multilingual automatic speech recognition, but their computation cost poses challenges for on-device applications. We propose a streaming truly multilingual Conformer…

Computation and Language · Computer Science 2023-05-26 Ke Hu , Bo Li , Tara N. Sainath , Yu Zhang , Francoise Beaufays

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

We propose the Mixture of Frozen Experts (MoFE) architecture, which integrates Parameter-efficient Fine-tuning (PEFT) and the Mixture of Experts (MoE) architecture to enhance both training efficiency and model scalability. By freezing the…

Computation and Language · Computer Science 2025-03-11 Jean Seo , Jaeyoon Kim , Hyopil Shin

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

Mixture-of-Experts (MoE) model architectures can significantly reduce the number of activated parameters per token, enabling computationally efficient training and inference. However, their large overall parameter counts and model sizes…

Machine Learning · Computer Science 2026-02-13 Arian Raje , Anupam Nayak , Gauri Joshi

The promising applications of large language models are often limited by the constrained GPU memory capacity available on edge devices. Mixture-of-Experts (MoE) models help address this issue by activating only a subset of the model's…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-03 Xiaoniu Song , Zihang Zhong , Rong Chen , Haibo Chen

Sparse Mixtures of Experts (SMoE) scales model capacity without significant increases in training and inference costs, but exhibits the following two issues: (1) Low expert activation, where only a small subset of experts are activated for…

Computation and Language · Computer Science 2024-04-24 Xun Wu , Shaohan Huang , Wenhui Wang , Furu Wei

Mixture-of-Experts (MoE) layers have emerged as an important tool in scaling up modern neural networks by decoupling total trainable parameters from activated parameters in the forward pass for each token. However, sparse MoEs add…

Machine Learning · Computer Science 2026-05-22 Tianze Jiang , Blake Bordelon , Cengiz Pehlevan , Boris Hanin

Mixture-of-Experts (MoE) models have gained popularity in achieving state-of-the-art performance in a wide range of tasks in computer vision and natural language processing. They effectively expand the model capacity while incurring a…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-06-21 Haiyang Huang , Newsha Ardalani , Anna Sun , Liu Ke , Hsien-Hsin S. Lee , Anjali Sridhar , Shruti Bhosale , Carole-Jean Wu , Benjamin Lee

Multi-Head Mixture-of-Experts (MH-MoE) demonstrates superior performance by using the multi-head mechanism to collectively attend to information from various representation spaces within different experts. In this paper, we present a novel…

Computation and Language · Computer Science 2024-12-02 Shaohan Huang , Xun Wu , Shuming Ma , Furu Wei

Mixture-of-Experts (MoE) based Large Language Models (LLMs) have demonstrated impressive performance and computational efficiency. However, their deployment is often constrained by substantial memory demands, primarily due to the need to…

Machine Learning · Computer Science 2026-03-16 Jiawei Hao , Zhiwei Hao , Jianyuan Guo , Li Shen , Yong Luo , Han Hu , Dan Zeng

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

Mixture-of-Experts (MoE) architectures have emerged as a powerful paradigm for scaling neural networks while maintaining computational efficiency. However, standard MoE implementations rely on two rigid design assumptions: (1) fixed Top-K…

Machine Learning · Computer Science 2026-03-03 Gökdeniz Gülmez

The Mixture of Experts (MoE) architecture is an important method for scaling Large Language Models (LLMs). It increases model capacity while keeping computation cost low. However, the ultra-large MoE models still have hundreds of billions…

Artificial Intelligence · Computer Science 2025-10-01 Yixiao Chen , Yanyue Xie , Ruining Yang , Wei Jiang , Wei Wang , Yong He , Yue Chen , Pu Zhao , Yanzhi Wang

Modern applications increasingly involve many heterogeneous input streams, such as clinical sensors, wearable device data, imaging, and text, each with distinct measurement models, sampling rates, and noise characteristics. We define this…

Machine Learning · Computer Science 2026-03-03 Xing Han , Hsing-Huan Chung , Joydeep Ghosh , Paul Pu Liang , Suchi Saria