English
Related papers

Related papers: Expert Selections In MoE Models Reveal (Almost) As…

200 papers

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 (MoE) architectures have emerged as a key strategy for scaling large language models (LLMs) efficiently. However, current MoE systems suffer from severe load imbalance, where only a small subset of experts is consistently…

Machine Learning · Computer Science 2025-06-27 Jiajie Yang

We present Bayesian Mixture of Experts (Bayesian-MoE), a post-hoc uncertainty estimation framework for fine-tuned large language models (LLMs) based on Mixture-of-Experts architectures. Our method applies a structured Laplace approximation…

Machine Learning · Computer Science 2025-11-13 Maryam Dialameh , Hossein Rajabzadeh , Weiwei Zhang , Walid Ahmed , Hyock Ju Kwon

The sparsely gated mixture of experts (MoE) architecture sends different inputs to different subnetworks, i.e., experts, through trainable routers. MoE reduces the training computation significantly for large models, but its deployment can…

Large language models (LLMs) with the Mixture-of-Experts (MoE) architecture achieve high cost-efficiency by selectively activating a subset of the parameters. Despite the inference efficiency of MoE LLMs, the training of extensive experts…

Computation and Language · Computer Science 2025-06-12 Yuchen Feng , Bowen Shen , Naibin Gu , Jiaxuan Zhao , Peng Fu , Zheng Lin , Weiping Wang

Recent advances in reinforcement learning (RL) have substantially improved the training of large-scale language models, leading to significant gains in generation quality and reasoning ability. However, most existing research focuses on…

Machine Learning · Computer Science 2026-01-13 Di Zhang , Xun Wu , Shaohan Huang , Lingjie Jiang , Yaru Hao , Li Dong , Zewen Chi , Zhifang Sui , Furu Wei

Mixture of Experts (MoEs) are now ubiquitous in large language models, yet the mechanisms behind their "expert specialization" remain poorly understood. We show that, since MoE routers are linear maps, hidden state similarity is both…

Artificial Intelligence · Computer Science 2026-04-14 Xi Wang , Soufiane Hayou , Eric Nalisnick

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

Mixture-of-Experts (MoE) shines brightly in large language models (LLMs) and demonstrates outstanding performance in plentiful natural language processing tasks. However, existing methods transforming LLMs from dense to MoE face significant…

Computation and Language · Computer Science 2024-10-04 Tingfeng Hui , Zhenyu Zhang , Shuohuan Wang , Yu Sun , Hua Wu , Sen Su

Scaling large language models (LLMs) significantly improves performance but comes with prohibitive computational costs. Mixture-of-Experts (MoE) models offer an efficient alternative, increasing capacity without a proportional rise in…

Machine Learning · Computer Science 2024-12-16 Aditya Vavre , Ethan He , Dennis Liu , Zijie Yan , June Yang , Nima Tajbakhsh , Ashwath Aithal

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

Diffusion policies have emerged as a powerful framework for robotic visuomotor control, yet they often lack the robustness to recover from subtask failures in long-horizon, multi-stage tasks and their learned representations of observations…

Robotics · Computer Science 2025-11-10 Baiye Cheng , Tianhai Liang , Suning Huang , Maanping Shao , Feihong Zhang , Botian Xu , Zhengrong Xue , Huazhe Xu

The scaling of large language models (LLMs) has revolutionized their capabilities in various tasks, yet this growth must be matched with efficient computational strategies. The Mixture-of-Experts (MoE) architecture stands out for its…

Computation and Language · Computer Science 2025-03-20 Zihan Qiu , Zeyu Huang , Shuang Cheng , Yizhi Zhou , Zili Wang , Ivan Titov , Jie Fu

Mixture-of-experts (MoE) models enable scalable transformer architectures by activating only a subset of experts per token. Recent evidence suggests that performance improves with increasingly granular experts, i.e., many small experts…

Machine Learning · Computer Science 2026-05-07 Klaus-Rudolf Kladny , Maximilian Mordig , Bernhard Schölkopf , Michael Muehlebach

Robustifying convolutional neural networks (CNNs) against adversarial attacks remains challenging and often requires resource-intensive countermeasures. We explore the use of sparse mixture-of-experts (MoE) layers to improve robustness by…

Computer Vision and Pattern Recognition · Computer Science 2025-09-08 Svetlana Pavlitska , Haixi Fan , Konstantin Ditschuneit , J. Marius Zöllner

Mixture of Experts (MoE) pretraining is more scalable than dense Transformer pretraining, because MoEs learn to route inputs to a sparse set of their feedforward parameters. However, this means that MoEs only receive a sparse backward…

Machine Learning · Computer Science 2025-11-05 Ashwinee Panda , Vatsal Baherwani , Zain Sarwar , Benjamin Therien , Sambit Sahu , Tom Goldstein , Supriyo Chakraborty

Mixture-of-Experts (MoE) architectures face challenges such as high memory consumption and redundancy in experts. Pruning MoE can reduce network weights while maintaining model performance. Motivated by the recent observation of emergent…

Computation and Language · Computer Science 2024-10-17 Yanyue Xie , Zhi Zhang , Ding Zhou , Cong Xie , Ziang Song , Xin Liu , Yanzhi Wang , Xue Lin , An Xu

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

Sparsely-activated Mixture-of-Experts (SMoE) models offer efficient pre-training and low latency but their large parameter counts create significant memory overhead, motivating research into expert compression. Contrary to recent findings…

Machine Learning · Computer Science 2026-05-14 Mike Lasby , Ivan Lazarevich , Nish Sinnadurai , Sean Lie , Yani Ioannou , Vithursan Thangarasa

The advancement of transformer neural networks has significantly elevated the capabilities of sentence similarity models, but they still struggle with highly discriminative tasks and may produce sub-optimal representations of important…

Machine Learning · Computer Science 2024-12-19 Logan Hallee , Rohan Kapur , Arjun Patel , Jason P. Gleghorn , Bohdan Khomtchouk