Task-Conditioned Routing Signatures in Sparse Mixture-of-Experts Transformers
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
Sparse Mixture-of-Experts (MoE) architectures enable efficient scaling of large language models through conditional computation, yet the routing mechanisms responsible for expert selection remain poorly understood. In this work, we introduce routing signatures, a vector representation summarizing expert activation patterns across layers for a given prompt, and use them to study whether MoE routing exhibits task-conditioned structure. Using OLMoE-1B-7B-0125-Instruct as an empirical testbed, we show that prompts from the same task category induce highly similar routing signatures, while prompts from different categories exhibit substantially lower similarity. Within-category routing similarity (0.8435 +/- 0.0879) significantly exceeds across-category similarity (0.6225 +/- 0.1687), corresponding to Cohen's d = 1.44. A logistic regression classifier trained solely on routing signatures achieves 92.5% +/- 6.1% cross-validated accuracy on four-way task classification. To ensure statistical validity, we introduce permutation and load-balancing baselines and show that the observed separation is not explained by sparsity or balancing constraints alone. We further analyze layer-wise signal strength and low-dimensional projections of routing signatures, finding that task structure becomes increasingly apparent in deeper layers. These results suggest that routing in sparse transformers is not merely a balancing mechanism, but a measurable task-sensitive component of conditional computation. We release MOE-XRAY, a lightweight toolkit for routing telemetry and analysis.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 3 of its 3 references that are in the map. First posted on arXiv 11 Mar 2026.
Builds on
3 of 3 references.
Closest ancestors · 1
Same direction, newest first
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 2
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 291
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Input Domain Aware MoE: Decoupling Routing Decisions from Task Optimization in Mixture of Experts
- 2026Mixture of Experts for Low-Resource LLMs
- 2026Safety-Oriented Routing Analysis of Mixtral MoE Under Benign and Harmful Prompts
- 2026Routing by Analogy: kNN-Augmented Expert Assignment for Mixture-of-Experts
- 2026Equifinality in Mixture of Experts: Routing Topology Does Not Determine Language Modeling Quality
- 2026It Takes a MAESTRO To Prune Bad Experts
- 2026Path-Constrained Mixture-of-Experts
- 2026TEXAS: Task-Expert-Aware Supervision for Downstream Mixture-of-Experts LLM Adaptation