Share First, Route What Remains: A Unified Framework for Token-Adaptive MoE Computation
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
Mixture-of-experts (MoE) models have recently moved beyond routing a fixed number of complete experts. Shared-expert designs preserve reusable knowledge, fine-grained methods vary computation within experts, and dynamic routers adapt the number of active experts. Yet these decisions are usually made independently, overlooking a basic dependency: extracting reusable computation changes both what remains and how much expert capacity the remainder needs. We study this dependency by decomposing sparsely upcycled feed-forward experts into key-value channels. Co-activated experts align at a subset of value positions; removing these positions changes expert preference; and greater shared coverage is associated with lower residual expert demand. These observations lead to one principle: share first, then route what remains. We instantiate it in UniF-MoE, a unified framework for token-adaptive MoE computation. Each expert is partitioned into aligned blocks. A shared-demand score sets the shared block count and pathway weight, key prototypes select the shared content, and the complementary demand determines the residual expert count through cumulative routing mass. A Gram regularizer separates and normalizes router embeddings, promoting diverse routing directions, sparse expert overlap, and a simple routing geometry. Experiments on DomainBed and GLUE show that this unified design improves predictive performance over representative static and dynamic MoEs while reducing activated computation, inference latency, and memory. Code is available at https://github.com/existence0420/UniF-MoE.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 28 of its 36 references that are in the map. First posted on arXiv 11 Aug 2026.
Builds on
28 of 36 references.
Closest ancestors · 3
Same direction, newest first
- 2025How Many Experts Are Enough? Towards Optimal Semantic Specialization for Mixture-of-Experts21 Dec
- 2024Dynamic Mixture of Experts: An Auto-Tuning Approach for Efficient Transformer Models23 May
- 2024Statistical Advantages of Perturbing Cosine Router in Mixture of Experts23 May
Other references · 25
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2018Moment Matching for Multi-Source Domain Adaptationcited by 1,327
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2020In Search of Lost Domain Generalizationcited by 1,042
13 more
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2018Neural Network Acceptability Judgmentscited by 952
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2018Recognition in Terra Incognitacited by 655
- 2022Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpointscited by 207
- 2022Sparse Mixture-of-Experts are Domain Generalizable Learnerscited by 85
- 2025Advancing Expert Specialization for Better MoEcited by 37
- 2024Mixture of Nested Experts: Adaptive Processing of Visual Tokenscited by 25
- 2023Unlocking Emergent Modularity in Large Language Modelscited by 17
- 2023Rethinking Domain Generalization: Discriminability and Generalizabilitycited by 16
- 2024LFME: A Simple Framework for Learning from Multiple Experts in Domain Generalizationcited by 8
- 2026MoSE: Mixture of Slimmable Experts for Efficient and Adaptive Language Modelscited by 3
- 2026Alloc-MoE: Budget-Aware Expert Activation Allocation for Efficient Mixture-of-Experts Inferencecited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Path-Constrained Mixture-of-Experts
- 2026Towards a Statistical Understanding of Mixture-of-Experts
- 2026Beyond Routing: Decoupling Expert Dispatch and Aggregation in Sparse Mixture-of-Experts
- 2026Hierarchical Mixture-of-Experts with Two-Stage Optimization
- 2026A Persistent State for Auditable Mixture-of-Experts Routing
- 2026MoRA: MoE Pruning via Router Bias Learning and Expert Approximation
- 2026Routing-Free Mixture-of-Experts
- 2026Elbow-Based MoE Routing: A Training-Free Inference Time Plugin for Expert Selection