Lune

ICML2026Top-tier venue

Opportunistic Expert Activation: Batch-Aware Expert Routing for Faster Decode Without Retraining

Costin-Andrei Oncescu, Qingyang Wu, Wai Tong Chung, Tsai-chuan Wu, Bryan Gopal, Junxiong Wang, Tri Dao, Ben Athiwaratkun

2026Year
4Citations

Abstract

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 often enter a memory-bound regime even for moderate batch sizes because the average expert load grows more slowly than in an equivalent dense feedforward layer. Consequently, MoE latency is governed by the number of activated experts. We introduce a framework for dynamically\textbf{dynamically} re-routing token-to-expert mapping to lower this number (and thus, the decode latency) while preserving a comparable quality. Our best results use a batch-aware routing\textbf{batch-aware routing} that works by having tokens piggyback\textbf{piggyback} experts that have already been loaded into memory due to being crucial to other tokens within the same batch. At batch size 1616, OEA reduces MoE-layer decode latency by 3939\\% on Qwen3-30B while preserving standard-error-adjusted downstream accuracy, and by 1515\\% on Qwen3-235B with only small overall degradation on the long-generation benchmark suite.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 31ddd95b-0ae4-4818-9e00-04510bfc0f43

Builds on8

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines