English
Related papers

Related papers: $\infty$-MoE: Generalizing Mixture of Experts to I…

200 papers

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) has gained increasing popularity as a promising framework for scaling up large language models (LLMs). However, training MoE from scratch in a large-scale setting still suffers from data-hungry and instability…

Computation and Language · Computer Science 2024-06-25 Tong Zhu , Xiaoye Qu , Daize Dong , Jiacheng Ruan , Jingqi Tong , Conghui He , Yu Cheng

In large language models like the Generative Pre-trained Transformer, the Mixture of Experts paradigm has emerged as a powerful technique for enhancing model expressiveness and accuracy. However, deploying GPT MoE models for parallel…

Machine Learning · Computer Science 2024-01-18 Jinghan Yao , Quentin Anthony , Aamir Shafi , Hari Subramoni , Dhabaleswar K. , Panda

Mixture-of-Experts (MoE) language models can reduce computational costs by 2-4$\times$ compared to dense models without sacrificing performance, making them more efficient in computation-bounded scenarios. However, MoE models generally…

Machine Learning · Computer Science 2024-04-09 Bowen Pan , Yikang Shen , Haokun Liu , Mayank Mishra , Gaoyuan Zhang , Aude Oliva , Colin Raffel , Rameswar Panda

Mixture-of-Experts (MoE) presents a naturally compatible and scalable framework for multimodal learning, demonstrating strong adaptability across diverse modalities and tasks. Despite its growing success, a comprehensive and systematic…

Machine Learning · Computer Science 2026-05-28 Liangwei Nathan Zheng , Wei Emma Zhang , Olaf Maennel , Lin Yue , Weitong Chen

Sparse Mixture-of-Experts (MoE) models scale capacity by routing each token to a small subset of experts. However, their routers exhibit a fundamental trade-off: strong load balancing can suppress expert specialization, while aggressive…

Machine Learning · Computer Science 2026-05-12 Gleb Molodtsov , Alexander Miasnikov , Aleksandr Beznosikov

MoE-PEFT methods combine Mixture of Experts with parameter-efficient fine-tuning for multi-task adaptation, but require separate adapters per expert causing trainable parameters to scale linearly with expert count and limiting applicability…

Machine Learning · Computer Science 2026-04-06 Md Kowsher , Haris Mansoor , Nusrat Jahan Prottasha , Ozlem Garibay , Victor Zhu , Zhengping Ji , Chen Chen

Mixture of Experts (MoE) LLMs have recently gained attention for their ability to enhance performance by selectively engaging specialized subnetworks or "experts" for each input. However, deploying MoEs on memory-constrained devices remains…

Mixture of experts has emerged as the primary mechanism for making Large Language Models (LLMs) computationally efficient. However, in distributed settings, communicating token embeddings between experts is a significant bottleneck. We…

Machine Learning · Computer Science 2026-05-08 Muhammad Shahir Abdurrahman , Chun Deng , Azalia Mirhoseini , Philip Levis

Mixture-of-Experts (MoE) has become a practical architecture for scaling LLM capacity while keeping per-token compute modest, but deploying MoE models on a single, memory-limited GPU remains difficult because expert weights dominate the HBM…

Performance · Computer Science 2026-02-09 Kexin Chu , Dawei Xiang , Zixu Shen , Yiwei Yang , Zecheng Liu , Wei Zhang

An increasing number of LLMs employ Mixture-of-Experts (MoE) architectures where the feed-forward layer is replaced by a pool of experts and each token only activates a small subset of them. During autoregressive generation, these models…

Machine Learning · Computer Science 2025-11-05 Costin-Andrei Oncescu , Qingyang Wu , Wai Tong Chung , Robert Wu , Bryan Gopal , Junxiong Wang , Tri Dao , Ben Athiwaratkun

Mixture-of-Expert (MoE) presents a strong potential in enlarging the size of language model to trillions of parameters. However, training trillion-scale MoE requires algorithm and system co-design for a well-tuned high performance…

Machine Learning · Computer Science 2021-03-25 Jiaao He , Jiezhong Qiu , Aohan Zeng , Zhilin Yang , Jidong Zhai , Jie Tang

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

Pre-training has proven effective in addressing data scarcity and performance limitations in solving PDE problems with neural operators. However, challenges remain due to the heterogeneity of PDE datasets in equation types, which leads to…

Machine Learning · Computer Science 2025-11-03 Hong Wang , Haiyang Xin , Jie Wang , Xuanze Yang , Fei Zha , Huanshuo Dong , Yan Jiang

Large language models (LLMs) have garnered unprecedented advancements across diverse fields, ranging from natural language processing to computer vision and beyond. The prowess of LLMs is underpinned by their substantial model size,…

Machine Learning · Computer Science 2025-04-10 Weilin Cai , Juyong Jiang , Fan Wang , Jing Tang , Sunghun Kim , Jiayi Huang

Mixture-of-Experts (MoE) models offer computational efficiency during inference by activating only a subset of specialized experts for a given input. This enables efficient model scaling on multi-GPU systems that use expert parallelism…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-06-18 Zachary Doucet , Rishi Sharma , Martijn de Vos , Rafael Pires , Anne-Marie Kermarrec , Oana Balmau

Mixture-of-Experts (MoE) is a neural network architecture that adds sparsely activated expert blocks to a base model, increasing the number of parameters without impacting computational costs. However, current distributed deep learning…

Machine Learning · Computer Science 2023-05-16 Siddharth Singh , Olatunji Ruwase , Ammar Ahmad Awan , Samyam Rajbhandari , Yuxiong He , Abhinav Bhatele

A sparse Mixture-of-Experts (MoE) architecture has emerged as a highly scalable solution by conditionally activating sub-modules without a proportional increase in computational costs. However, improving expert specialization to enhance…

Machine Learning · Computer Science 2025-09-16 Sugyeong Eo , Jungjun Lee , Chanjun Park , Heuiseok Lim

Mixture of Experts (MoE) architectures have significantly increased computational efficiency in both research and real-world applications of large-scale machine learning models. However, their scalability and efficiency under memory…

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