English
Related papers

Related papers: Mixture of A Million Experts

200 papers

The mixture-of-experts (MoE) architecture scales model size with sublinear computational increase but suffers from memory-intensive inference due to KV caches and sparse expert activation. Recent disaggregated expert parallelism (DEP)…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-12-29 Xinglin Pan , Shaohuai Shi , Wenxiang Lin , Yuxin Wang , Zhenheng Tang , Wei Wang , Xiaowen Chu

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

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

Federated learning (FL) is a collaborative machine learning approach that enables multiple clients to train models without sharing their private data. With the rise of deep learning, large-scale models have garnered significant attention…

Machine Learning · Computer Science 2025-09-09 Ziwei Zhan , Wenkuan Zhao , Yuanqing Li , Weijie Liu , Xiaoxi Zhang , Chee Wei Tan , Chuan Wu , Deke Guo , Xu Chen

Large Language Models (LLMs) with Mixture-of-Experts (MoE) architectures are distinguished by their strong performance scaling with increasing parameters across a wide range of tasks, yet they also suffer from substantial computational and…

Computation and Language · Computer Science 2025-11-27 Yuzhuang Xu , Xu Han , Yuanchi Zhang , Yixuan Wang , Yijun Liu , Shiyu Ji , Qingfu Zhu , Wanxiang Che

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

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

Reinforcement learning (RL) has equipped LLM agents with a strong ability to solve complex tasks. However, existing RL methods normally use a \emph{single} policy network, causing \emph{simplicity bias} where simple tasks occupy most…

Artificial Intelligence · Computer Science 2026-05-20 Shengtian Yang , Yu Li , Shuo He , Yewen Li , Qingpeng Cai , Peng Jiang , Lei Feng

Mixture-of-Experts (MoE) models have gained popularity as a means of scaling the capacity of large language models (LLMs) while maintaining sparse activations and reduced per-token compute. However, in memory-constrained inference settings,…

Machine Learning · Computer Science 2026-03-23 Vivan Madan , Prajwal Singhania , Abhinav Bhatele , Tom Goldstein , Ashwinee Panda

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

The proliferation of large language models (LLMs) has led to the adoption of Mixture-of-Experts (MoE) architectures that dynamically leverage specialized subnetworks for improved efficiency and performance. Despite their benefits, MoE…

Computation and Language · Computer Science 2024-10-28 Ruisi Cai , Yeonju Ro , Geon-Woo Kim , Peihao Wang , Babak Ehteshami Bejnordi , Aditya Akella , Zhangyang Wang

Mixture-of-Experts (MoE) models offer dynamic computation, but are typically deployed as static full-capacity models, missing opportunities for deployment-specific specialization. We introduce PreMoE, a training-free framework that…

Machine Learning · Computer Science 2026-04-27 Zehua Pei , Ying Zhang , Hui-Ling Zhen , Tao Yuan , Xianzhi Yu , Zhenhua Dong , Sinno Jialin Pan , Mingxuan Yuan , Bei Yu

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…

Mixture-of-Experts (MoE) has become the dominant architecture for scaling large language models: frontier models routinely decouple total parameters from per-token computation through sparse expert routing. Scaling laws show that under…

Machine Learning · Computer Science 2026-05-12 Chaitanya Dwivedi , Binxuan Huang , Himanshu Gupta , Pratik Jayarao , Neeraj Varshney , Bing Yin

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

Sparse Mixture of Expert (SMoE) models have emerged as a scalable alternative to dense models in language modeling. These models use conditionally activated feedforward subnetworks in transformer blocks, allowing for a separation between…

Machine Learning · Computer Science 2024-09-04 Soumajyoti Sarkar , Leonard Lausen , Volkan Cevher , Sheng Zha , Thomas Brox , George Karypis

Large Language Models (LLMs) encounter significant challenges in continual learning due to catastrophic forgetting, where new information overwrites previously acquired knowledge. This limitation leads to substantial environmental and…

Computation and Language · Computer Science 2024-08-01 Min Jae Jung , JooHee Kim

The Mixture-of-Experts (MoE) architecture has become a predominant paradigm for scaling large language models (LLMs). Despite offering strong performance and computational efficiency, large MoE-based LLMs like DeepSeek-V3-0324 and…

Machine Learning · Computer Science 2025-08-08 Xiaodong Chen , Mingming Ha , Zhenzhong Lan , Jing Zhang , Jianguo Li

Mixture-of-Experts (MoE) architectures employ sparse activation to deliver faster training and inference with higher accuracy than dense LLMs. However, in production serving, MoE models require batch inference to optimize hardware…

Machine Learning · Computer Science 2026-02-10 Juntong Wu , Jialiang Cheng , Fuyu Lv , Ou Dan , Li Yuan

Scaling language models unlocks impressive capabilities, but the accompanying computational and memory demands make both training and deployment expensive. Existing efficiency efforts typically target either parameter sharing or adaptive…

Computation and Language · Computer Science 2025-10-28 Sangmin Bae , Yujin Kim , Reza Bayat , Sungnyun Kim , Jiyoun Ha , Tal Schuster , Adam Fisch , Hrayr Harutyunyan , Ziwei Ji , Aaron Courville , Se-Young Yun