English
Related papers

Related papers: Occult: Optimizing Collaborative Communication acr…

200 papers

Mixture-of-Experts models have become a dominant architecture for scaling Large Language Models by activating only a sparse subset of experts per token. However, latency-critical MoE inference faces a fundamental tension: while expert…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-04 Qianchao Zhu , Xucheng Ye , Yuliang Liu , Haodong Ouyang , Chengru Song

In recent years, Mixture-of-Experts (MoE) has emerged as an effective approach for enhancing the capacity of deep neural network (DNN) with sub-linear computational costs. However, storing all experts on GPUs incurs significant memory…

Machine Learning · Computer Science 2025-03-11 Suraiya Tairin , Shohaib Mahmud , Haiying Shen , Anand Iyer

The immense memory requirements of state-of-the-art Mixture-of-Experts (MoE) models present a significant challenge for inference, often exceeding the capacity of a single accelerator. While offloading experts to host memory is a common…

Machine Learning · Computer Science 2025-11-19 Wenfeng Wang , Jiacheng Liu , Xiaofeng Hou , Xinfeng Xia , Peng Tang , Mingxuan Zhang , Chao Li , Minyi Guo

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

Recent large language models (LLMs) have tended to leverage sparsity to reduce computations, employing the sparsely activated mixture-of-experts (MoE) technique. MoE introduces four modules, including token routing, token communication,…

Machine Learning · Computer Science 2025-01-22 Xinglin Pan , Wenxiang Lin , Lin Zhang , Shaohuai Shi , Zhenheng Tang , Rui Wang , Bo Li , Xiaowen Chu

Mixture-of-Experts (MoE) Large Language Models (LLMs) face a trilemma of load imbalance, parameter redundancy, and communication overhead. We introduce a unified framework based on dynamic expert clustering and structured compression to…

Computation and Language · Computer Science 2026-02-06 Peijun Zhu , Ning Yang , Baoliang Tian , Jiayu Wei , Weihao Zhang , Haijun Zhang , Pin Lv

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) models have gained popularity in achieving state-of-the-art performance in a wide range of tasks in computer vision and natural language processing. They effectively expand the model capacity while incurring a…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-06-21 Haiyang Huang , Newsha Ardalani , Anna Sun , Liu Ke , Hsien-Hsin S. Lee , Anjali Sridhar , Shruti Bhosale , Carole-Jean Wu , Benjamin Lee

Scaling large language models has driven remarkable advancements across various domains, yet the continual increase in model size presents significant challenges for real-world deployment. The Mixture of Experts (MoE) architecture offers a…

Machine Learning · Computer Science 2025-03-18 Shwai He , Daize Dong , Liang Ding , Ang Li

Mixture-of-Experts is a promising approach for edge AI with low-batch inference. Yet, on-device deployments often face limited on-chip memory and severe workload imbalance; the prevalent use of offloading further incurs off-chip memory…

Hardware Architecture · Computer Science 2026-03-31 Songchen Ma , Hongyi Li , Weihao Zhang , Yonghao Tan , Pingcheng Dong , Yu Liu , Lan Liu , Yuzhong Jiao , Xuejiao Liu , Luhong Liang , Kwang-Ting Cheng

The goal of this paper is to improve (upcycle) an existing large language model without the prohibitive requirements of continued pre-training of the full-model. The idea is to split the pre-training data into semantically relevant groups…

Machine Learning · Computer Science 2024-10-15 Ajinkya Tejankar , KL Navaneet , Ujjawal Panchal , Kossar Pourahmadi , Hamed Pirsiavash

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

Mixture of Experts (MoE) offers remarkable performance and computational efficiency by selectively activating subsets of model parameters. Traditionally, MoE models use homogeneous experts, each with identical capacity. However, varying…

Computation and Language · Computer Science 2024-08-21 An Wang , Xingwu Sun , Ruobing Xie , Shuaipeng Li , Jiaqi Zhu , Zhen Yang , Pinxue Zhao , J. N. Han , Zhanhui Kang , Di Wang , Naoaki Okazaki , Cheng-zhong Xu

Mixture-of-Experts (MoE) large language models (LLMs), which leverage dynamic routing and sparse activation to enhance efficiency and scalability, have achieved higher performance while reducing computational costs. However, these models…

Machine Learning · Computer Science 2025-05-08 Xing Hu , Zhixuan Chen , Dawei Yang , Zukang Xu , Chen Xu , Zhihang Yuan , Sifan Zhou , Jiangyong Yu

Mixture-of-Expert (MoE) based large language models (LLMs), such as the recent Mixtral and DeepSeek-MoE, have shown great promise in scaling model size without suffering from the quadratic growth of training cost of dense transformers. Like…

Machine Learning · Computer Science 2024-04-04 Longfei Yun , Yonghao Zhuang , Yao Fu , Eric P Xing , Hao Zhang

The Mixture-of-Expert (MoE) technique plays a crucial role in expanding the size of DNN model parameters. However, it faces the challenge of extended all-to-all communication latency during the training process. Existing methods attempt to…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-05-01 Chenyu Jiang , Ye Tian , Zhen Jia , Shuai Zheng , Chuan Wu , Yida Wang

Machine learning models often need to adapt to new data after deployment due to structured or unstructured real-world dynamics. The Continual Learning (CL) framework enables continuous model adaptation, but most existing approaches either…

Machine Learning · Computer Science 2026-03-25 Connor Mclaughlin , Nigel Lee , Lili Su

Mixture-of-Experts (MoE) architecture offers enhanced efficiency for Large Language Models (LLMs) with modularized computation, yet its inherent sparsity poses significant hardware deployment challenges, including memory locality issues,…

Hardware Architecture · Computer Science 2026-03-10 Shuqing Luo , Ye Han , Pingzhi Li , Jiayin Qin , Jie Peng , Yang , Zhao , Yu , Cao , Tianlong Chen

Frontier models increasingly adopt Mixture-of-Experts (MoE) architectures to achieve large-model performance at reduced cost. However, training MoE models on HPC platforms is hindered by large memory footprints, frequent large-scale…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-07 Sajal Dash , Feiyi Wang

Mixture-of-Experts (MoE) models are designed to enhance the efficiency of large language models (LLMs) without proportionally increasing the computational demands. However, their deployment on edge devices still faces significant challenges…

Machine Learning · Computer Science 2024-08-21 Shuzhang Zhong , Ling Liang , Yuan Wang , Runsheng Wang , Ru Huang , Meng Li