English
Related papers

Related papers: Optimizing Mixture of Experts using Dynamic Recomp…

200 papers

This paper introduces a theoretical framework for a Transformer-augmented, sectional Mixture-of-Experts (MoE) architecture that aims to enhance computational efficiency while preserving model scalability. Unlike conventional MoE models,…

Machine Learning · Computer Science 2025-03-27 Soham Sane

Continual learning enables vision-language models to accumulate knowledge and adapt to evolving tasks without retraining from scratch. However, in multi-domain task-incremental learning, large domain shifts intensify the…

Computer Vision and Pattern Recognition · Computer Science 2026-05-11 Mengxin Qin , Xiang Zhang , Xi Wang , Kun Wei , Xu Yang , Cheng Deng

Mixture-of-Experts (MoE) has emerged as a promising approach to scale up deep learning models due to its significant reduction in computational resources. However, the dynamic nature of MoE leads to load imbalance among experts, severely…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-16 Chenqi Zhao , Wenfei Wu , Linhai Song , Yuchen Xu , Yitao Yuan

Large language models are typically deployed as monolithic systems, requiring the full model even when applications need only a narrow subset of capabilities, e.g., code, math, or domain-specific knowledge. Mixture-of-Experts (MoEs)…

Computation and Language · Computer Science 2026-05-12 Ryan Wang , Akshita Bhagia , Sewon Min

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 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

The Mixture-of-Experts (MoE) architecture has emerged as a promising approach to mitigate the rising computational costs of large language models (LLMs) by selectively activating parameters. However, its high memory requirements and…

Artificial Intelligence · Computer Science 2026-04-14 Jehyeon Bang , Eunyeong Cho , Ranggi Hwang , Jinha Chung , Minsoo Rhu

Graph neural networks excel at graph representation learning but struggle with heterophilous data and long-range dependencies. And graph transformers address these issues through self-attention, yet face scalability and noise challenges on…

Machine Learning · Computer Science 2025-02-13 Xuanze Chen , Jiajun Zhou , Shanqing Yu , Qi Xuan

The Mixture-of-Experts (MoE) model has emerged as a prominent architecture in the field of Large Language Models (LLMs), providing a better balance between model performance and computational efficiency. However the General Matrix Multiply…

Computation and Language · Computer Science 2025-01-06 Yulei Qian , Fengcun Li , Xiangyang Ji , Xiaoyu Zhao , Jianchao Tan , Kefeng Zhang , Xunliang Cai

Large multimodal Mixture-of-Experts (MoEs) effectively scale the model size to boost performance while maintaining fixed active parameters. However, previous works primarily utilized full-precision experts during sparse up-cycling. Despite…

Computer Vision and Pattern Recognition · Computer Science 2026-01-08 Hongyu Wang , Jiayu Xu , Ruiping Wang , Yan Feng , Yitao Zhai , Peng Pei , Xunliang Cai , Xilin Chen

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) Transformer, the backbone architecture of multiple phenomenal language models, leverages sparsity by activating only a fraction of model parameters for each input token. The sparse structure, while allowing constant…

Expert parallelism is vital for effectively training Mixture-of-Experts (MoE) models, enabling different devices to host distinct experts, with each device processing different input data. However, during expert parallel training, dynamic…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-13 Xinyi Liu , Yujie Wang , Fangcheng Fu , Xuefeng Xiao , Huixia Li , Jiashi Li , Bin Cui

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

Mixture-of-Experts (MoE) has recently emerged as the mainstream architecture for efficiently scaling large language models while maintaining near-constant computational cost. Expert parallelism distributes parameters by partitioning experts…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-01 Adrian Zhao , Zhenkun Cai , Zhenyu Song , Lingfan Yu , Haozheng Fan , Jun Wu , Yida Wang , Nandita Vijaykumar

Mixture of Experts (MoE) models enable parameter-efficient scaling through sparse expert activations, yet optimizing their inference and memory costs remains challenging due to limited understanding of their specialization behavior. We…

Machine Learning · Computer Science 2026-03-09 Marmik Chaudhari , Idhant Gulati , Nishkal Hundia , Pranav Karra , Shivam Raval

The sparsely activated mixture of experts (MoE) model presents a promising alternative to traditional densely activated (dense) models, enhancing both quality and computational efficiency. However, training MoE models from scratch demands…

Computer Vision and Pattern Recognition · Computer Science 2024-06-10 Xingkui Zhu , Yiran Guan , Dingkang Liang , Yuchao Chen , Yuliang Liu , Xiang Bai

The ability to customize a trained Deep Neural Network (DNN) locally using user-specific data may greatly enhance user experiences, reduce development costs, and protect user's privacy. In this work, we propose to incorporate a novel…

Computer Vision and Pattern Recognition · Computer Science 2018-11-02 Boyu Zhang , Azadeh Davoodi , Yu-Hen Hu

We present MegaBlocks, a system for efficient Mixture-of-Experts (MoE) training on GPUs. Our system is motivated by the limitations of current frameworks, which restrict the dynamic routing in MoE layers to satisfy the constraints of…

Machine Learning · Computer Science 2022-11-30 Trevor Gale , Deepak Narayanan , Cliff Young , Matei Zaharia

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