English
Related papers

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

200 papers

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

Large language models (LLMs) have become increasingly popular in various areas, traditional business gradually shifting from rule-based systems to LLM-based solutions. However, the inference of LLMs is resource-intensive or…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-06 Wanyi Zheng , Minxian Xu , Shengye Song , Kejiang Ye

Serverless computing has grown rapidly for serving Large Language Model (LLM) inference due to its pay-as-you-go pricing, fine-grained GPU usage, and rapid scaling. However, our analysis reveals that current serverless can effectively serve…

Machine Learning · Computer Science 2025-05-21 Yifan Sui , Hao Wang , Hanfei Yu , Yitao Hu , Jianxun Li , Hao Wang

Mixture-of-Experts (MoE) inference requires large-scale token exchange across devices, making dispatch and combine major bottlenecks in both prefill and decode. Beyond network transfer, routing-driven layout transformation, temporary relay,…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-11 Tianlun Hu , Tiancheng Hu , Shengsheng Litang , Sheng Wang , Xiaoming Bao , Yuxing Li , Wei Wang , Zhongzhe Hu , Lijun Li , Hongwei Sun , Jingbin Zhou

Fine-tuning pre-trained large language models (LLMs) in a distributed manner poses significant challenges on resource-constrained edge networks. To address this challenge, we propose SflLLM, a novel framework that integrates split federated…

Machine Learning · Computer Science 2025-07-03 Kai Zhao , Zhaohui Yang , Ye Hu , Mingzhe Chen , Chen Zhu , Zhaoyang Zhang

Large language models (LLMs) have emerged due to their capability to generate high-quality content across diverse contexts. To reduce their explosively increasing demands for computing resources, a mixture of experts (MoE) has emerged. The…

Hardware Architecture · Computer Science 2025-11-04 Sungmin Yun , Kwanhee Kyung , Juhwan Cho , Jaewan Choi , Jongmin Kim , Byeongho Kim , Sukhan Lee , Kyomin Sohn , Jung Ho Ahn

Parameter-efficient fine-tuning (PEFT) has emerged as a crucial paradigm for adapting large language models (LLMs) under constrained computational budgets. However, standard PEFT methods often struggle in multi-task fine-tuning settings,…

Machine Learning · Computer Science 2026-04-03 Juyong Jiang , Fan Wang , Hong Qi , Sunghun Kim , Jing Tang

The Mixture-of-Experts (MoE) architecture has demonstrated significant advantages in the era of Large Language Models (LLMs), offering enhanced capabilities with reduced inference costs. However, deploying MoE-based LLMs on…

Machine Learning · Computer Science 2024-11-07 Peng Tang , Jiacheng Liu , Xiaofeng Hou , Yifei Pu , Jing Wang , Pheng-Ann Heng , Chao Li , Minyi Guo

Large language model (LLM)-based multi-agent systems have demonstrated remarkable promise for tackling complex tasks by breaking them down into subtasks that are iteratively planned, executed, observed, and refined. Despite their…

Multiagent Systems · Computer Science 2025-07-15 Enhao Zhang , Erkang Zhu , Gagan Bansal , Adam Fourney , Hussein Mozannar , Jack Gerrits

Large language models (LLMs) power many modern applications, but serving them at scale remains costly and resource-intensive. Current server-centric systems overlook consumer-grade GPUs at the edge. We introduce SpecEdge, an edge-assisted…

Computation and Language · Computer Science 2025-11-19 Jinwoo Park , Seunggeun Cho , Dongsu Han

This paper introduces MoxE, a novel architecture that synergistically combines the Extended Long Short-Term Memory (xLSTM) with the Mixture of Experts (MoE) framework to address critical scalability and efficiency challenges in large…

Computation and Language · Computer Science 2025-05-06 Abdoul Majid O. Thiombiano , Brahim Hnich , Ali Ben Mrad , Mohamed Wiem Mkaouer

Mixture-of-Experts (MoE) models enable scalable neural networks through conditional computation, offering enhanced effectiveness and efficiency for next-generation wireless communications. However, deploying MoE with federated learning (FL)…

Machine Learning · Computer Science 2026-05-19 Boyang Zhang , Xiaobing Chen , Songyang Zhang , Shuai Zhang , Xiangwei Zhou , Jian Zhang , Mingxuan Sun

Emerging computation-intensive applications impose stringent latency requirements on resource-constrained mobile devices. Mobile Edge Computing (MEC) addresses this challenge through task offloading. However, designing effective policies…

Machine Learning · Computer Science 2026-04-09 Ning Yang , Chuangxin Cheng , Haijun Zhang

In the era of Large Language Models (LLMs), Mixture-of-Experts (MoE) architectures offer a promising approach to managing computational costs while scaling up model parameters. Conventional MoE-based LLMs typically employ static Top-K…

Computation and Language · Computer Science 2024-10-16 Tongtian Yue , Longteng Guo , Jie Cheng , Xuange Gao , Jing Liu

Large Language Models (LLMs) such as GPT-4 and Llama3 can already comprehend complex commands and process diverse tasks. This advancement facilitates their application in controlling drones and robots for various tasks. However, existing…

Robotics · Computer Science 2024-12-30 Neiwen Ling , Guojun Chen , Lin Zhong

Large Language Models (LLMs) demonstrate substantial potential across a diverse array of domains via request serving. However, as trends continue to push for expanding context sizes, the autoregressive nature of LLMs results in highly…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-07-08 Bin Lin , Chen Zhang , Tao Peng , Hanyu Zhao , Wencong Xiao , Minmin Sun , Anmin Liu , Zhipeng Zhang , Lanbo Li , Xiafei Qiu , Shen Li , Zhigang Ji , Tao Xie , Yong Li , Wei Lin

Nowadays, many companies possess various types of AI accelerators, forming heterogeneous clusters. Efficiently leveraging these clusters for high-throughput large language model (LLM) inference services can significantly reduce costs and…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-23 Yi Xiong , Jinqi Huang , Wenjie Huang , Xuebing Yu , Entong Li , Zhixiong Ning , Jinhua Zhou , Li Zeng , Xin Chen

A pivotal advancement in the progress of large language models (LLMs) is the emergence of the Mixture-of-Experts (MoE) LLMs. Compared to traditional LLMs, MoE LLMs can achieve higher performance with fewer parameters, but it is still hard…

Computation and Language · Computer Science 2024-05-31 Xudong Lu , Qi Liu , Yuhui Xu , Aojun Zhou , Siyuan Huang , Bo Zhang , Junchi Yan , Hongsheng Li

Mixture-of-Experts architectures have become the standard for scaling large language models due to their superior parameter efficiency. To accommodate the growing number of experts in practice, modern inference systems commonly adopt expert…

Cryptography and Security · Computer Science 2026-05-26 Ruixuan Huang , Qingyue Wang , Hantao Huang , Yudong Gao , Dong Chen , Shuai Wang , Wei Wang

Recent advances in code generation models have unlocked unprecedented opportunities for automating feature engineering, yet their adoption in real-world ML teams remains constrained by critical challenges: (i) the scarcity of datasets…

Machine Learning · Computer Science 2026-01-19 Himanshu Thakur , Anusha Kamath , Anurag Muthyala , Dhwani Sanmukhani , Smruthi Mukund , Jay Katukuri