English
Related papers

Related papers: Accelerating Mixture-of-Experts Inference by Hidin…

200 papers

Diffusion-based stylization has advanced significantly, yet existing methods are limited to color-driven transformations, neglecting complex semantics and material details. We introduce StyleExpert, a semantic-aware framework based on the…

Computer Vision and Pattern Recognition · Computer Science 2026-03-31 Shihao Zhu , Ziheng Ouyang , Yijia Kang , Qilong Wang , Mi Zhou , Bo Li , Ming-Ming Cheng , Qibin Hou

Expert Parallelism (EP) permits Mixture of Experts (MoE) models to scale beyond a single GPU. To address load imbalance across GPUs in EP, existing approaches aim to balance the number of tokens each GPU processes. Surprisingly, we find…

Speculative decoding is a prominent technique to speed up the inference of a large target language model based on predictions of an auxiliary draft model. While effective, in application-specific settings, it often involves fine-tuning both…

Computation and Language · Computer Science 2024-02-20 Nikhil Bhendawade , Irina Belousova , Qichen Fu , Henry Mason , Mohammad Rastegari , Mahyar Najibi

Mixture-of-Experts (MoE) has become the de facto architecture for hundred-billion-parameter language models, yet its advantages at sub-billion scales for on-device deployment remain largely unexplored. To close this gap, we present…

The computational sparsity of Mixture-of-Experts (MoE) models enables sub-linear growth in compute cost as model size increases, thus offering a scalable path to training massive neural networks. However, existing implementations suffer…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-11 Osayamen Jonathan Aimuyo , Byungsoo Oh , Rachee Singh

RL post-training of frontier language models is increasingly bottlenecked by autoregressive rollout generation, making rollout acceleration a central systems challenge. Many existing efficiency methods improve throughput by changing the…

We propose Chain-of-Experts (CoE), a new Mixture-of-Experts (MoE) architecture that introduces sequential expert communication within each layer. Unlike traditional MoE models, where experts operate independently in parallel, CoE processes…

Machine Learning · Computer Science 2025-06-25 Zihan Wang , Rui Pan , Jiarui Yao , Robert Csordas , Linjie Li , Lu Yin , Jiajun Wu , Tong Zhang , Manling Li , Shiwei Liu

Expert parallelism has emerged as a key strategy for distributing the computational workload of sparsely-gated mixture-of-experts (MoE) models across multiple devices, enabling the processing of increasingly large-scale models. However, the…

Machine Learning · Computer Science 2025-05-30 Weilin Cai , Juyong Jiang , Le Qin , Junwei Cui , Sunghun Kim , Jiayi Huang

The sparsely activated mixture of experts (MoE) model presents a promising alternative to traditional densely activated (dense) models, enhancing both quality and computational efficiency. However, training MoE models from scratch demands…

Computer Vision and Pattern Recognition · Computer Science 2024-06-10 Xingkui Zhu , Yiran Guan , Dingkang Liang , Yuchao Chen , Yuliang Liu , Xiang Bai

In a distributed mixture-of-experts (MoE) system, a server collaborates with multiple specialized expert clients to perform inference. The server extracts features from input data and dynamically selects experts based on their areas of…

Machine Learning · Computer Science 2025-04-02 Qiuchen Song , Shusen Jing , Shuai Zhang , Songyang Zhang , Chuan Huang

Mixture-of-Experts (MoE) Multimodal large language models (MLLMs) excel at vision-language tasks, but they suffer from high computational inefficiency. To reduce inference overhead, expert skipping methods have been proposed to deactivate…

Computer Vision and Pattern Recognition · Computer Science 2026-02-24 Yushi Huang , Zining Wang , Zhihang Yuan , Yifu Ding , Ruihao Gong , Jinyang Guo , Xianglong Liu , Jun Zhang

Mixture-of-Experts (MoE) is a promising way to scale up the learning capacity of large language models. It increases the number of parameters while keeping FLOPs nearly constant during inference through sparse activation. Yet, it still…

Machine Learning · Computer Science 2025-02-26 Pingzhi Li , Xiaolong Jin , Zhen Tan , Yu Cheng , Tianlong Chen

Sparsely gated Mixture of Experts (MoE) models have been shown to be a compute-efficient method to scale model capacity for multilingual machine translation. However, for low-resource tasks, MoE models severely over-fit. We show effective…

Computation and Language · Computer Science 2022-12-19 Maha Elbayad , Anna Sun , Shruti Bhosale

The Mixture-of-Experts (MoE) paradigm has emerged as a promising solution to scale up model capacity while maintaining inference efficiency. However, deploying MoE models across heterogeneous end-cloud environments poses new challenges in…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-11 Zheming Yang , Yunqing Hu , Sheng Sun , Wen Ji

The increasing demand for deploying large Mixture-of-Experts (MoE) models in resource-constrained environments necessitates efficient approaches to address their high memory and computational requirements challenges. Moreover, given that…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-09-10 HamidReza Imani , Abdolah Amirany , Tarek El-Ghazawi

Mixture of experts (MoE) model is a statistical machine learning design that aggregates multiple expert networks using a softmax gating function in order to form a more intricate and expressive model. Despite being commonly used in several…

Machine Learning · Statistics 2024-06-25 Huy Nguyen , Nhat Ho , Alessandro Rinaldo

Sparse Mixture of Experts (MoE) architectures have emerged as a promising approach for scaling Transformer models. While initial works primarily incorporated MoE into feed-forward network (FFN) layers, recent studies have explored extending…

Machine Learning · Computer Science 2025-10-24 Yuanhang Yang , Chaozheng Wang , Jing Li

We propose a simple yet effective approach to enhance the performance of feed-forward 3D reconstruction models. Existing methods often struggle near depth discontinuities, where standard regression losses encourage spatial averaging and…

Computer Vision and Pattern Recognition · Computer Science 2026-01-27 Zichen Wang , Ang Cao , Liam J. Wang , Jeong Joon Park

Sparse Mixture-of-Experts (MoE) models offer a powerful way to scale model size without increasing compute, as per-token FLOPs depend only on k active experts rather than the total pool of E experts. Yet, this asymmetry creates an MoE…

Machine Learning · Computer Science 2026-05-15 Linghao Jin , Chufan Shi , Huijuan Wang , Nuan Wen , Zhengzhong Liu , Eric Xing , Xuezhe Ma

Speculative Decoding is a widely used technique to speed up inference for Large Language Models (LLMs) without sacrificing quality. When performing inference, speculative decoding uses a smaller draft model to generate speculative tokens…

Machine Learning · Computer Science 2025-02-06 Minghao Yan , Saurabh Agarwal , Shivaram Venkataraman