English
Related papers

Related papers: Harder Tasks Need More Experts: Dynamic Routing in…

200 papers

To be effective, efficient, and diverse, deep learning models need to dynamically choose its architecture based on signals from a population of neurons. We hypothesize dynamic routing models can be improved with neural inhibition in those…

Machine Learning · Computer Science 2025-07-08 Will Y. Zou , Jennifer Y. Zhang

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…

Mixture-of-Experts (MoE) architectures scale large language models (LLMs) by activating only a subset of experts per token, but the standard TopK routing assigns the same fixed number of experts to all tokens, ignoring their varying…

Machine Learning · Computer Science 2026-03-30 Tiansheng Wen , Yifei Wang , Aosong Feng , Long Ma , Xinyang Liu , Yifan Wang , Lixuan Guo , Bo Chen , Stefanie Jegelka , Chenyu You

Despite recent advancements in offline multi-task reinforcement learning (MTRL) have harnessed the powerful capabilities of the Transformer architecture, most approaches focus on a limited number of tasks, with scaling to extremely massive…

Machine Learning · Computer Science 2025-06-02 Yilun Kong , Guozheng Ma , Qi Zhao , Haoyu Wang , Li Shen , Xueqian Wang , Dacheng Tao

Mixture-of-Experts (MoE) models embody the divide-and-conquer concept and are a promising approach for increasing model capacity, demonstrating excellent scalability across multiple domains. In this paper, we integrate the MoE structure…

Computer Vision and Pattern Recognition · Computer Science 2024-11-27 Xumeng Han , Longhui Wei , Zhiyang Dou , Zipeng Wang , Chenhui Qiang , Xin He , Yingfei Sun , Zhenjun Han , Qi Tian

Due to the inherent difficulty in modeling phonetic similarities across different languages, code-switching speech recognition presents a formidable challenge. This study proposes a Collaborative-MoE, a Mixture of Experts (MoE) model that…

Computation and Language · Computer Science 2024-09-06 Hukai Huang , Jiayan Lin , Kaidi Wang , Yishuang Li , Wenhao Guan , Lin Li , Qingyang Hong

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

Mixture-of-Experts layers achieve compute efficiency through weight sparsity: each token activates only a subset of experts. Data sparsity, where each expert processes only a subset of tokens, offers a complementary axis. Expert-choice…

Machine Learning · Computer Science 2026-01-23 Maciej Kilian , Oleg Mkrtchyan , Luke Zettlemoyer , Akshat Shrivastava , Armen Aghajanyan

Mixture-of-Experts (MoE) models improve transformer efficiency but lack a unified theoretical explanation, especially when both feed-forward and attention layers are allowed to specialize. To this end, we study the Mixture-of-Transformers…

Machine Learning · Computer Science 2025-11-03 Hongbo Li , Qinhang Wu , Sen Lin , Yingbin Liang , Ness B. Shroff

Efficiency, specialization, and adaptability to new data distributions are qualities that are hard to combine in current Large Language Models. The Mixture of Experts (MoE) architecture has been the focus of significant research because its…

Computation and Language · Computer Science 2024-08-29 Nikolas Gritsch , Qizhen Zhang , Acyr Locatelli , Sara Hooker , Ahmet Üstün

Non-stationary time series forecasting is challenged by evolving distribution shifts that static models struggle to capture. While Mixture-of-Experts (MoE) architectures offer a promising paradigm for decoupling complex drift patterns,…

Machine Learning · Computer Science 2026-05-21 Jiawen Zhu , Shuhan Liu , Di Weng , Yingcai Wu

Mixture-of-Experts (MoE) architectures have become standard in large language models, yet many of their core design choices - expert count, granularity, shared experts, load balancing, token dropping - have only been studied one or two at a…

Machine Learning · Computer Science 2026-05-13 Margaret Li , Sneha Kudugunta , Danielle Rothermel , Luke Zettlemoyer

Real-world model deployment across multiple domains requires multimodal models to operate under two complementary regimes: (1) multi-task pretraining, tasks are co-available at design time where related tasks could borrow representational…

Machine Learning · Computer Science 2026-05-12 Xing Han , Shravan Chaudhari , Tanvi Ranade , Rama Chellappa , Suchi Saria

Mixture of Experts (MoE) has become a key architectural paradigm for efficiently scaling Large Language Models (LLMs) by selectively activating a subset of parameters for each input token. However, standard MoE architectures face…

Machine Learning · Computer Science 2025-05-27 Zehua Liu , Han Wu , Ruifeng She , Xiaojin Fu , Xiongwei Han , Tao Zhong , Mingxuan Yuan

Supervised fine-tuning (SFT) is a milestone in aligning large language models with human instructions and adapting them to downstream tasks. In particular, Low-Rank Adaptation (LoRA) has gained widespread attention due to its parameter…

Computation and Language · Computer Science 2025-11-05 Jia-Chen Zhang , Yu-Jie Xiong , Xi-He Qiu , Chun-Ming Xia , Fei Dai , Zheng Zhou

We present a novel approach called Mixture of Mixture of Expert (MoMoE) that combines the strengths of Mixture-of-Experts (MoE) architectures with collaborative multi-agent frameworks. By modifying the LLaMA 3.1 8B architecture to…

Computational Engineering, Finance, and Science · Computer Science 2025-11-19 Peng Shu , Junhao Chen , Zhengliang Liu , Hanqi Jiang , Yi Pan , Khanh Nhu Nguyen , Zihao Wu , Huaqin Zhao , Yiwei Li , Enze Shi , ShaoChen Xu

Sparse Mixture-of-Experts (MoE) architectures are increasingly popular for frontier large language models (LLM) but they introduce training challenges due to routing complexity. Fully leveraging parameters of an MoE model requires all…

Mixture-of-Experts (MoE) architectures enable conditional computation by routing inputs to multiple expert subnetworks and are often motivated as a mechanism for scaling large language models. In this project, we instead study MoE behavior…

Machine Learning · Computer Science 2026-01-22 Adam Rokah , Daniel Veress , Caleb Caulk , Sourav Sharan

The Mixture-of-Experts (MoE) architecture is showing promising results in improving parameter sharing in multi-task learning (MTL) and in scaling high-capacity neural networks. State-of-the-art MoE models use a trainable sparse gate to…

Mixture-of-Experts (MoE) models can achieve promising results with outrageous large amount of parameters but constant computation cost, and thus it has become a trend in model scaling. Still it is a mystery how MoE layers bring quality…

Machine Learning · Computer Science 2021-08-10 An Yang , Junyang Lin , Rui Men , Chang Zhou , Le Jiang , Xianyan Jia , Ang Wang , Jie Zhang , Jiamang Wang , Yong Li , Di Zhang , Wei Lin , Lin Qu , Jingren Zhou , Hongxia Yang