English
Related papers

Related papers: Mozart: Modularized and Efficient MoE Training on …

200 papers

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 large language models (MoE-LLMs) marks a significant step forward of language models, however, they encounter two critical challenges in practice: 1) expert parameters lead to considerable memory consumption and loading…

Machine Learning · Computer Science 2025-02-25 Wei Huang , Yue Liao , Jianhui Liu , Ruifei He , Haoru Tan , Shiming Zhang , Hongsheng Li , Si Liu , Xiaojuan Qi

Mixture of Experts (MoE) has become a mainstream architecture for building Large Language Models (LLMs) by reducing per-token computation while enabling model scaling. It can be viewed as partitioning a large Feed-Forward Network (FFN) at…

Machine Learning · Computer Science 2025-08-27 Weilin Cai , Le Qin , Shwai He , Junwei Cui , Ang Li , Jiayi Huang

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

Conventional large language models (LLMs) are equipped with dozens of GB to TB of model parameters, making inference highly energy-intensive and costly as all the weights need to be loaded to onboard processing elements during computation.…

Hardware Architecture · Computer Science 2025-07-28 Wei-Hsing Huang , Janak Sharda , Cheng-Jhih Shih , Yuyao Kong , Faaiq Waqar , Pin-Jun Chen , Yingyan , Lin , Shimeng Yu

Despite LLMs' excellent code creation capabilities, multilingual code generation remains extremely challenging. To address this, we intent to improve the multi-programming-lingual (MultiPL) performance of the base LLMs while retaining the…

Computation and Language · Computer Science 2025-09-09 Qing Wang , Xue Han , Jiahui Wang , Lehao Xing , Qian Hu , Lianlian Zhang , Chao Deng , Junlan Feng

The Mixture of Experts (MoE) architecture is a cornerstone of modern state-of-the-art (SOTA) large language models (LLMs). MoE models facilitate scalability by enabling sparse parameter activation. However, traditional MoE architecture uses…

Computation and Language · Computer Science 2025-08-12 Haoyuan Wu , Haoxing Chen , Xiaodong Chen , Zhanchao Zhou , Tieyuan Chen , Yihong Zhuang , Guoshan Lu , Zenan Huang , Junbo Zhao , Lin Liu , Zhenzhong Lan , Bei Yu , Jianguo Li

The Sparsely-Activated Mixture-of-Experts (MoE) has gained increasing popularity for scaling up large language models (LLMs) without exploding computational costs. Despite its success, the current design faces a challenge where all experts…

Machine Learning · Computer Science 2024-09-20 Manxi Sun , Wei Liu , Jian Luan , Pengzhi Gao , Bin Wang

The Mixture of Experts (MoE) architecture is an important method for scaling Large Language Models (LLMs). It increases model capacity while keeping computation cost low. However, the ultra-large MoE models still have hundreds of billions…

Artificial Intelligence · Computer Science 2025-10-01 Yixiao Chen , Yanyue Xie , Ruining Yang , Wei Jiang , Wei Wang , Yong He , Yue Chen , Pu Zhao , Yanzhi Wang

Large Vision-Language Models (LVLMs) have demonstrated remarkable performance across multi-modal tasks by scaling model size and training data. However, these dense LVLMs incur significant computational costs and motivate the exploration of…

Computer Vision and Pattern Recognition · Computer Science 2026-01-09 Dianyi Wang , Siyuan Wang , Zejun Li , Yikun Wang , Yitong Li , Duyu Tang , Xiaoyu Shen , Xuanjing Huang , Zhongyu Wei

The mixture of experts (MoE) model is a sparse variant of large language models (LLMs), designed to hold a better balance between intelligent capability and computational overhead. Despite its benefits, MoE is still too expensive to deploy…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-23 Haodong Wang , Qihua Zhou , Zicong Hong , Song Guo

The Mixtures-of-Experts (MoE) model is a widespread distributed and integrated learning method for large language models (LLM), which is favored due to its ability to sparsify and expand models efficiently. However, the performance of MoE…

Machine Learning · Computer Science 2024-05-24 Jing Li , Zhijie Sun , Xuan He , Li Zeng , Yi Lin , Entong Li , Binfan Zheng , Rongqian Zhao , Xin 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

Optimizing various wireless user tasks poses a significant challenge for networking systems because of the expanding range of user requirements. Despite advancements in Deep Reinforcement Learning (DRL), the need for customized optimization…

Networking and Internet Architecture · Computer Science 2024-02-16 Hongyang Du , Guangyuan Liu , Yijing Lin , Dusit Niyato , Jiawen Kang , Zehui Xiong , Dong In Kim

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

The Mixture-of-Experts (MoE) approach has demonstrated outstanding scalability in multi-task learning including low-level upstream tasks such as concurrent removal of multiple adverse weather effects. However, the conventional MoE…

Computer Vision and Pattern Recognition · Computer Science 2023-12-29 Rongyu Zhang , Yulin Luo , Jiaming Liu , Huanrui Yang , Zhen Dong , Denis Gudovskiy , Tomoyuki Okuno , Yohei Nakata , Kurt Keutzer , Yuan Du , Shanghang Zhang

Mixture-of-Experts (MoE) models enable scalable performance by activating large parameter sets sparsely, minimizing computational overhead. To mitigate the prohibitive cost of training MoEs from scratch, recent work employs upcycling,…

Machine Learning · Computer Science 2025-11-13 Qi Wang , Hanyang Peng , Yue Yu

Mixture of Experts (MoE) architectures increase large language model scalability, yet their performance depends on the router module that moves tokens to specialized experts. Bad routing can load imbalance and reduced accuracy. This project…

Machine Learning · Computer Science 2025-06-23 Daniel Fidel Harvey , George Weale , Berk Yilmaz

Previous work on Universal Transformers (UTs) has demonstrated the importance of parameter sharing across layers. By allowing recurrence in depth, UTs have advantages over standard Transformers in learning compositional generalizations, but…

Machine Learning · Computer Science 2024-10-15 Róbert Csordás , Kazuki Irie , Jürgen Schmidhuber , Christopher Potts , Christopher D. Manning

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