English
Related papers

Related papers: SwitchHead: Accelerating Transformers with Mixture…

200 papers

Mixture-of-Experts (MoE) Multimodal large language models (MLLMs) excel at vision-language tasks, but they suffer from high computational inefficiency. To reduce inference overhead, expert skipping methods have been proposed to deactivate…

Computer Vision and Pattern Recognition · Computer Science 2026-02-24 Yushi Huang , Zining Wang , Zhihang Yuan , Yifu Ding , Ruihao Gong , Jinyang Guo , Xianglong Liu , Jun Zhang

Mixture-of-Experts (MoE) enhances model performance while maintaining computational efficiency, making it well-suited for large-scale applications. Conventional mixture-of-experts (MoE) architectures suffer from suboptimal coordination…

Machine Learning · Computer Science 2025-09-24 Yujiao Yang , Jing Lian , Linhui Li

Mixture-of-Expert (MoE) models have obtained state-of-the-art performance in Neural Machine Translation (NMT) tasks. Existing works in MoE mostly consider a homogeneous design where the same number of experts of the same size are placed…

In this work, we upgrade the multi-head attention mechanism, the core of the Transformer model, to improve efficiency while maintaining or surpassing the previous accuracy level. We show that multi-head attention can be expressed in the…

Computer Vision and Pattern Recognition · Computer Science 2025-12-02 Peng Jin , Bo Zhu , Li Yuan , Shuicheng Yan

In recent years, Mixture-of-Experts (MoE) has emerged as an effective approach for enhancing the capacity of deep neural network (DNN) with sub-linear computational costs. However, storing all experts on GPUs incurs significant memory…

Machine Learning · Computer Science 2025-03-11 Suraiya Tairin , Shohaib Mahmud , Haiying Shen , Anand Iyer

Mixture-of-Experts (MoE) Large Language Models (LLMs) efficiently scale-up the model while keeping relatively low inference cost. As MoE models only activate part of the experts, related work has proposed expert prediction and caching…

Computation and Language · Computer Science 2025-11-17 Shien Zhu , Samuel Bohl , Robin Oester , Gustavo Alonso

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) models based on Transformer architecture are pushing the boundaries of language and vision tasks. The allure of these models lies in their ability to substantially increase the parameter count without a…

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

How to reduce compute and memory requirements of neural networks (NNs) without sacrificing performance? Many recent works use sparse Mixtures of Experts (MoEs) to build resource-efficient large language models (LMs). Here we introduce…

Machine Learning · Computer Science 2023-11-22 Róbert Csordás , Kazuki Irie , Jürgen Schmidhuber

In this paper, we explore a strategy that uses Mixture-of-Experts (MoE) to streamline, rather than augment, vision transformers. Each expert in an MoE layer is a SwiGLU feedforward network, where V and W2 are shared across the layer. No…

Computer Vision and Pattern Recognition · Computer Science 2024-07-26 Jen Hong Tan

The Mixture of Experts (MoE) models are an emerging class of sparsely activated deep learning models that have sublinear compute costs with respect to their parameters. In contrast with dense models, the sparse architecture of MoE offers…

The computational sparsity of Mixture-of-Experts (MoE) models enables sub-linear growth in compute cost as model size increases, thus offering a scalable path to training massive neural networks. However, existing implementations suffer…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-11 Osayamen Jonathan Aimuyo , Byungsoo Oh , Rachee Singh

Scale has opened new frontiers in natural language processing -- but at a high cost. In response, Mixture-of-Experts (MoE) and Switch Transformers have been proposed as an energy efficient path to even larger and more capable language…

Computation and Language · Computer Science 2022-05-03 Barret Zoph , Irwan Bello , Sameer Kumar , Nan Du , Yanping Huang , Jeff Dean , Noam Shazeer , William Fedus

Mixture-of-Experts (MoE) models have recently demonstrated exceptional performance across a diverse range of applications. The principle of sparse activation in MoE models facilitates an offloading strategy, wherein active experts are…

Computation and Language · Computer Science 2025-10-15 Yushu Zhao , Yubin Qin , Yang Wang , Xiaolong Yang , Huiming Han , Shaojun Wei , Yang Hu , Shouyi Yin

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

Large Mixture of Experts (MoE) models could achieve state-of-the-art quality on various language tasks, including machine translation task, thanks to the efficient model scaling capability with expert parallelism. However, it has brought a…

Machine Learning · Computer Science 2023-10-05 Young Jin Kim , Raffy Fahim , Hany Hassan Awadalla

Mixture-of-experts (MoE) architectures have expanded from language modeling to automatic speech recognition (ASR). Traditional MoE methods, such as the Switch Transformer, route experts independently within each layer. Our analysis reveals…

Computation and Language · Computer Science 2025-11-06 Zijin Gu , Tatiana Likhomanenko , Navdeep Jaitly

Mixture-of-Experts models, now popular for scaling capacity at fixed inference speed, switch experts at nearly every token. Once a model outgrows available GPU memory, this churn can render optimizations like offloading and pre-fetching…

Machine Learning · Computer Science 2026-04-23 Zeyu Shen , Peter Henderson

Scaling up the number of parameters of language models has proven to be an effective approach to improve performance. For dense models, increasing model size proportionally increases the model's computation footprint. In this work, we seek…

Computation and Language · Computer Science 2023-11-21 Cicero Nogueira dos Santos , James Lee-Thorp , Isaac Noble , Chung-Ching Chang , David Uthus