English
Related papers

Related papers: Multi-Layer Scheduling for MoE-Based LLM Reasoning

200 papers

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

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-Experts (MoE) activates only a subset of experts during inference, allowing the model to maintain low inference FLOPs and latency even as the parameter count scales up. However, since MoE dynamically selects the experts, all the…

Machine Learning · Computer Science 2025-05-27 Shibo Jie , Yehui Tang , Kai Han , Yitong Li , Duyu Tang , Zhi-Hong Deng , Yunhe Wang

Mixture of Experts (MoE) LLMs, characterized by their sparse activation patterns, offer a promising approach to scaling language models while avoiding proportionally increasing the inference cost. However, their large parameter sizes…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-15 Yichao Yuan , Lin Ma , Nishil Talati

Mixture-of-Experts (MoE) model architecture has emerged as a promising solution for scaling transformer models efficiently, offering sparse activation that reduces computational costs while increasing model capacity. However, as MoE models…

Machine Learning · Computer Science 2025-02-11 Seokjin Go , Divya Mahajan

Large language models (LLMs) have been a disruptive innovation in recent years, and they play a crucial role in our daily lives due to their ability to understand and generate human-like text. Their capabilities include natural language…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-10-17 Akrit Mudvari , Yuang Jiang , Leandros Tassiulas

Augmented Large Language Models (LLMs) enhance the capabilities of standalone LLMs by integrating external data sources through API calls. In interactive LLM applications, efficient scheduling is crucial for maintaining low request…

Machine Learning · Computer Science 2024-10-29 Rana Shahout , Cong Liang , Shiji Xin , Qianru Lao , Yong Cui , Minlan Yu , Michael Mitzenmacher

Large-scale Mixture of Experts (MoE) Large Language Models (LLMs) have recently become the frontier open-weight models, achieving remarkable model capability similar to proprietary ones. But their random expert selection mechanism…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-13 Zhongkai Yu , Yue Guan , Zihao Yu , Chenyang Zhou , Zhengding Hu , Shuyi Pei , Yangwook Kang , Yufei Ding , Po-An Tsai

Mixture of Experts (MoE) models have enabled the scaling of Large Language Models (LLMs) and Vision Language Models (VLMs) by achieving massive parameter counts while maintaining computational efficiency. However, MoEs introduce several…

Efficient scheduling of LLM inference tasks is essential for achieving low latency and high throughput, particularly with the growing use of reasoning-capable LLMs. Traditional strategies like First-Come-First-Serve (FCFS) often suffer from…

Machine Learning · Computer Science 2025-10-13 Yiheng Tao , Yihe Zhang , Matthew T. Dearing , Xin Wang , Yuping Fan , Zhiling Lan

The deployment of mixture-of-experts (MoE) large language models (LLMs) presents significant challenges due to their high memory demands. These challenges become even more pronounced in multi-tenant environments, where shared resources must…

Machine Learning · Computer Science 2025-05-13 HamidReza Imani , Jiaxin Peng , Peiman Mohseni , Abdolah Amirany , Tarek El-Ghazawi

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

As a prominent data modality task, time series forecasting plays a pivotal role in diverse applications. With the remarkable advancements in Large Language Models (LLMs), the adoption of LLMs as the foundational architecture for time series…

Machine Learning · Computer Science 2025-07-10 Yiwen Liu , Chenyu Zhang , Junjie Song , Siqi Chen , Sun Yin , Zihan Wang , Lingming Zeng , Yuji Cao , Junming Jiao

As AI moves beyond text, large language models (LLMs) increasingly power vision, audio, and document understanding; however, their high inference costs hinder real-time, scalable deployment. Conversely, smaller open-source models offer cost…

Computation and Language · Computer Science 2025-11-11 Mayank Saini , Arit Kumar Bishwas

High-Performance Computing (HPC) job scheduling involves balancing conflicting objectives such as minimizing makespan, reducing wait times, optimizing resource use, and ensuring fairness. Traditional methods, including heuristic-based,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-05 Prachi Jadhav , Hongwei Jin , Ewa Deelman , Prasanna Balaprakash

Recent progress in Language Models (LMs) has dramatically advanced the field of natural language processing (NLP), excelling at tasks like text generation, summarization, and question answering. However, their inference remains…

Machine Learning · Computer Science 2025-06-10 Adarsh Prasad Behera , Jaya Prakash Champati , Roberto Morabito , Sasu Tarkoma , James Gross

Mixture-of-Experts (MoE) enables efficient scaling of large language models (LLMs) with sparsely activated experts during inference. To effectively deploy large MoE models on memory-constrained devices, many systems introduce *expert…

Machine Learning · Computer Science 2026-03-03 Jingcong Liang , Siyuan Wang , Miren Tian , Yitong Li , Duyu Tang , Zhongyu Wei

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 surged in popularity and are extensively used in commercial applications, where the efficiency of model serving is crucial for the user experience. Most current research focuses on optimizing individual…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-09-12 Yuhang Yao , Han Jin , Alay Dilipbhai Shah , Shanshan Han , Zijian Hu , Yide Ran , Dimitris Stripelis , Zhaozhuo Xu , Salman Avestimehr , Chaoyang He

The Mixture of Experts (MoE) architecture has become a fundamental building block in state-of-the-art large language models (LLMs), improving domain-specific expertise in LLMs and scaling model capacity without proportionally increasing…

Machine Learning · Computer Science 2026-05-13 Ankit Jyothish , Ali Jannesari , Aishwarya Sarkar , Joseph Zuber