English
Related papers

Related papers: Multi-Head Mixture-of-Experts

200 papers

We present ScatterMoE, an implementation of Sparse Mixture-of-Experts (SMoE) on GPUs. ScatterMoE builds upon existing implementations, and overcoming some of the limitations to improve inference and training speed, and memory footprint.…

Machine Learning · Computer Science 2024-10-07 Shawn Tan , Yikang Shen , Rameswar Panda , Aaron Courville

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

Large language models have transformed many applications but remain expensive to train. Sparse Mixture of Experts (MoE) addresses this through conditional computation, with Expert Parallel (EP) as the standard distributed training method.…

Machine Learning · Computer Science 2026-02-05 Chenwei Cui , Rockwell Jackson , Benjamin Joseph Herrera , Ana María Tárano , Hannah Kerner

Mixture of Experts (MoE) achieve parameter-efficient scaling through sparse expert routing, yet their internal representations remain poorly understood compared to dense models. We present a systematic comparison of MoE and dense model…

Machine Learning · Computer Science 2026-03-09 Marmik Chaudhari , Nishkal Hundia , Idhant Gulati

Mixture-of-Experts (MoE) architectures scale Large Language Models via expert specialization induced by conditional computation. In practice, however, expert specialization often fails: some experts become functionally similar, while others…

Mixture-of-Experts (MoE) architectures have emerged as a promising direction, offering efficiency and scalability by activating only a subset of parameters during inference. However, current research remains largely performance-centric,…

Machine Learning · Computer Science 2025-09-30 Jiahao Ying , Mingbao Lin , Qianru Sun , Yixin Cao

Mixture-of-Experts (MoE) architectures have emerged as a promising approach to scale Large Language Models (LLMs). MoE boosts the efficiency by activating a subset of experts per token. Recent works show that fine-grained experts…

Computation and Language · Computer Science 2025-10-21 Zheyue Tan , Zhiyuan Li , Tao Yuan , Dong Zhou , Weilin Liu , Yueqing Zhuang , Yadong Li , Guowei Niu , Cheng Qin , Zhuyu Yao , Congyi Liu , Haiyang Xu , Boxun Li , Guohao Dai , Bo Zhao , Yu Wang

We propose Chain-of-Experts (CoE), a new Mixture-of-Experts (MoE) architecture that introduces sequential expert communication within each layer. Unlike traditional MoE models, where experts operate independently in parallel, CoE processes…

Machine Learning · Computer Science 2025-06-25 Zihan Wang , Rui Pan , Jiarui Yao , Robert Csordas , Linjie Li , Lu Yin , Jiajun Wu , Tong Zhang , Manling Li , Shiwei Liu

Mixture of experts (MoE) is a popular technique to improve capacity of Large Language Models (LLMs) with conditionally-activated parallel experts. However, serving MoE models on memory-constrained devices is challenging due to the large…

Artificial Intelligence · Computer Science 2024-05-30 Rui Kong , Yuanchun Li , Qingtian Feng , Weijun Wang , Xiaozhou Ye , Ye Ouyang , Linghe Kong , Yunxin Liu

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

This paper presents a comprehensive review of the Mixture-of-Experts (MoE) architecture in large language models, highlighting its ability to significantly enhance model performance while maintaining minimal computational overhead. Through…

Machine Learning · Computer Science 2025-12-24 Danyang Zhang , Junhao Song , Ziqian Bi , Xinyuan Song , Yingfang Yuan , Tianyang Wang , Joe Yeong , Junfeng Hao

Sparse Mixture-of-Experts (MoE) is a neural architecture design that can be utilized to add learnable parameters to Large Language Models (LLMs) without increasing inference cost. Instruction tuning is a technique for training LLMs to…

Mixture-of-Experts (MoE) models enable efficient scaling of large language models (LLMs) by activating only a subset of experts per input. However, we observe that the commonly used auxiliary load balancing loss often leads to expert…

Computation and Language · Computer Science 2026-01-27 Hongcan Guo , Haolang Lu , Guoshun Nan , Bolun Chu , Jialin Zhuang , Yuan Yang , Wenhao Che , Xinye Cao , Sicong Leng , Qimei Cui , Xudong Jiang

Recent advances in large language models highlighted the excessive quadratic cost of self-attention. Despite the significant research efforts, subquadratic attention methods still suffer from inferior performance in practice. We hypothesize…

Machine Learning · Computer Science 2025-05-02 Piotr Piękos , Róbert Csordás , Jürgen Schmidhuber

To help the open-source community have a better understanding of Mixture-of-Experts (MoE) based large language models (LLMs), we train and release OpenMoE, a series of fully open-sourced and reproducible decoder-only MoE LLMs, ranging from…

Computation and Language · Computer Science 2024-03-28 Fuzhao Xue , Zian Zheng , Yao Fu , Jinjie Ni , Zangwei Zheng , Wangchunshu Zhou , Yang You

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

In this paper, we introduce a novel dynamic expert selection framework for Mixture of Experts (MoE) models, aiming to enhance computational efficiency and model performance by adjusting the number of activated experts based on input…

Machine Learning · Computer Science 2024-03-13 Quzhe Huang , Zhenwei An , Nan Zhuang , Mingxu Tao , Chen Zhang , Yang Jin , Kun Xu , Kun Xu , Liwei Chen , Songfang Huang , Yansong Feng

Mixture-of-Experts (MoE) architectures have shown strong multilingual capabilities, yet the internal mechanisms underlying performance gains and cross-language differences remain insufficiently understood. In this work, we conduct a…

Computation and Language · Computer Science 2026-01-21 Yuxin Chen , Zhengzhou Cai , Xiangtian Ji , Weixiang Zhao , An Zhang , Xiang Wang , Tat-Seng Chua

Real-world time series exhibit complex and evolving dynamics, making accurate forecasting extremely challenging. Recent multi-modal forecasting methods leverage textual information such as news reports to improve prediction, but most rely…

Machine Learning · Computer Science 2026-01-30 Lige Zhang , Ali Maatouk , Jialin Chen , Leandros Tassiulas , Rex Ying

While federated learning (FL) enables fine-tuning of large language models (LLMs) without compromising data privacy, the substantial size of an LLM renders on-device training impractical for resource-constrained clients, such as mobile…

Machine Learning · Computer Science 2026-01-05 Zihan Fang , Zheng Lin , Senkang Hu , Yanan Ma , Yihang Tao , Yiqin Deng , Xianhao Chen , Yuguang Fang