English
Related papers

Related papers: PROBE: Co-Balancing Computation and Communication …

200 papers

Large Language Models (LLMs) based on Mixture-of-Experts (MoE) are pivotal in industrial applications for their ability to scale performance efficiently. However, standard MoEs enforce uniform expert sizes,creating a rigidity that fails to…

Computation and Language · Computer Science 2026-04-29 Zhicheng Ma , Xiang Liu , Zhaoxiang Liu , Ning Wang , Yi Shen , Kai Wang , Shuming Shi , Shiguo Lian

Mixture-of-Experts (MoE) models have emerged as a promising direction for scaling vision architectures efficiently. Among them, Soft MoE improves training stability by assigning each token to all experts via continuous dispatch weights.…

Computer Vision and Pattern Recognition · Computer Science 2025-05-27 Chengxi Min , Wei Wang , Yahui Liu , Weixin Ye , Enver Sangineto , Qi Wang , Yao Zhao

The sparsely activated mixture-of-experts (MoE) transformer has become a common architecture for large language models (LLMs) due to its sparsity, which requires fewer computational demands while easily scaling the model size. In MoE…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-14 Wenxiang Lin , Xinglin Pan , Lin Zhang , Shaohuai Shi , Xuan Wang , Xiaowen Chu

In multi-GPU Mixture-of-Experts (MoE) network, experts are distributed across different GPUs, which creates load imbalance as each expert processes different number of tokens. Recent works improve MoE inference load balance by dynamically…

Machine Learning · Computer Science 2025-06-10 Haiyue Ma , Zhixu Du , Yiran Chen

Large language models (LLMs) have demonstrated remarkable capabilities across a wide range of natural language processing tasks. Exploiting the heterogeneous capabilities of edge LLMs is crucial for diverse emerging applications, as it…

Networking and Internet Architecture · Computer Science 2025-01-17 Lyudong Jin , Yanning Zhang , Yanhan Li , Shurong Wang , Howard H. Yang , Jian Wu , Meng Zhang

The Mixture of Experts (MoE) is a widely known neural architecture where an ensemble of specialized sub-models optimizes overall performance with a constant computational cost. However, conventional MoEs pose challenges at scale due to the…

Computation and Language · Computer Science 2023-09-12 Ted Zadouri , Ahmet Üstün , Arash Ahmadian , Beyza Ermiş , Acyr Locatelli , Sara Hooker

One of the primary challenges in optimizing large language models (LLMs) for long-context inference lies in the high memory consumption of the Key-Value (KV) cache. Existing approaches, such as quantization, have demonstrated promising…

Computer Vision and Pattern Recognition · Computer Science 2025-06-10 Wei Tao , Haocheng Lu , Xiaoyang Qu , Bin Zhang , Kai Lu , Jiguang Wan , Jianzong 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) architectures employ sparse activation to deliver faster training and inference with higher accuracy than dense LLMs. However, in production serving, MoE models require batch inference to optimize hardware…

Machine Learning · Computer Science 2026-02-10 Juntong Wu , Jialiang Cheng , Fuyu Lv , Ou Dan , Li Yuan

Mixture of experts (MoE) is a popular technique to improve capacity of Large Language Models (LLMs) with conditionally-activated parallel experts. However, serving MoE models on memory-constrained devices is challenging due to the large…

Artificial Intelligence · Computer Science 2024-05-30 Rui Kong , Yuanchun Li , Qingtian Feng , Weijun Wang , Xiaozhou Ye , Ye Ouyang , Linghe Kong , Yunxin Liu

Mixture-of-Experts (MoE) architectures enhance the efficiency of large language models by activating only a subset of experts per token. However, standard MoE employs a fixed Top-K routing strategy, leading to redundant computation and…

Artificial Intelligence · Computer Science 2026-05-15 Juntong Wu , Jialiang Cheng , Qishen Yin , Yue Dai , Yuliang Yan , Fuyu Lv , Ou Dan , Li Yuan

While modern internet services, such as chatbots, search engines, and online advertising, demand the use of large-scale deep neural networks (DNNs), distributed training and inference over heterogeneous computing systems are desired to…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-08-13 Dianhai Yu , Liang Shen , Hongxiang Hao , Weibao Gong , Huachao Wu , Jiang Bian , Lirong Dai , Haoyi Xiong

Reliable channel estimation (CE) is fundamental for robust communication in dynamic wireless environments, where models must generalize across varying conditions such as signal-to-noise ratios (SNRs), the number of resource blocks (RBs),…

Signal Processing · Electrical Eng. & Systems 2025-09-22 Tianyu Li , Yan Xin , Jianzhong , Zhang

Mixture-of-Experts (MoE) models typically fix the number of activated experts $k$ at both training and inference. However, real-world deployments often face heterogeneous hardware, fluctuating workloads, and diverse quality-latency…

Computation and Language · Computer Science 2026-05-12 Naibin Gu , Zhenyu Zhang , Yuchen Feng , Yilong Chen , Peng Fu , Zheng Lin , Shuohuan Wang , Yu Sun , Hua Wu , Weiping Wang , Haifeng Wang

By increasing model parameters but activating them sparsely when performing a task, the use of Mixture-of-Experts (MoE) architecture significantly improves the performance of Large Language Models (LLMs) without increasing the inference…

Computation and Language · Computer Science 2025-06-10 Zeliang Zhang , Xiaodong Liu , Hao Cheng , Chenliang Xu , Jianfeng Gao

Mixture-of-Experts (MoE) models enable scalable performance but face severe memory constraints on edge devices. Existing offloading strategies struggle with I/O bottlenecks due to the dynamic, low-information nature of autoregressive expert…

Machine Learning · Computer Science 2026-03-12 Shuhuai Li , Jianghao Lin , Dongdong Ge , Yinyu Ye

Mixture-of-Experts (MoE) layers activate a subset of model weights, dubbed experts, to improve model performance. MoE is particularly promising for deployment on process-in-memory (PIM) architectures, because PIM can naturally fit experts…

Hardware Architecture · Computer Science 2026-02-12 Hanyuan Gao , Xiaoxuan Yang

Deep learning on an edge device requires energy efficient operation due to ever diminishing power budget. Intentional low quality data during the data acquisition for longer battery life, and natural noise from the low cost sensor degrade…

Machine Learning · Computer Science 2019-04-30 Taesik Na , Minah Lee , Burhan A. Mudassar , Priyabrata Saha , Jong Hwan Ko , Saibal Mukhopadhyay

Fine-grained Mixture-of-Experts (MoE) models sparsely activate only a subset of experts per token, reducing activated computation while maintaining high model capacity. However, in memory-constrained inference scenarios, only a small set of…

Machine Learning · Computer Science 2026-05-27 Xiongwei Zhu , Xiaojian Liao , Tianyang Jiang , Yusen Zhang , Liang Wang , Limin Xiao

Large Language Models have revolutionized natural language processing, yet serving them efficiently in data centers remains challenging due to mixed workloads comprising latency-sensitive (LS) and best-effort (BE) jobs. Existing inference…

Machine Learning · Computer Science 2025-03-13 Mohammad Siavashi , Faezeh Keshmiri Dindarloo , Dejan Kostic , Marco Chiesa