A Persistent State for Auditable Mixture-of-Experts Routing
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
Mixture-of-Experts (MoE) models repeatedly route tokens to sparse subsets of experts, but conventional routers expose no routing-specific record of how cross-layer influences accumulate. We introduce Scratchpad-Augmented Mixture-of-Experts (SA-MoE), which gives each router access to a low-dimensional persistent state that is not provided to the experts. Learned layerwise writes update this state, and their realized post-update changes exactly decompose the state-mediated contribution to any later routing margin, forming a routing ledger. Across sparsely upcycled SmolLM2- and Gemma-based models and three independent training seeds per architecture, this pathway adds less than 1% analytical forward compute and is strongly used by trained routers: local removal of its router contribution changes the selected Top-2 expert set in 87.6% and 69.9% of decisions, respectively. Relative to a matched latest-write-only control, persistent accumulation increases long-horizon future-routing accessibility by 19.4 and 12.2 percentage points, with positive effects in every seed. More than 90% of absolute ledger contribution comes from non-recent writes in both families, and full-forward suppression of ledger-selected writes changes later routing and output distributions. The ledger is an exact provenance object for the persistent-state pathway, not a complete causal explanation of routing. Sensitivity-aware scores better predict full-forward intervention effects, and post-hoc methods recover related cross-layer attribution without architectural modification. SA-MoE instead makes one routing-specific computational history explicit and directly inspectable within the model's natural forward computation.
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 17 of its 24 references that are in the map. First posted on arXiv 28 Sep 2026.
Builds on
17 of 24 references.
Closest ancestors · 4
Same direction, newest first
- 2025Multilingual Routing in Mixture-of-Experts6 Oct
- 2025Decoding Knowledge Attribution in Mixture-of-Experts: A Framework of Basic-Refinement Collaboration and Efficiency Analysis30 May
- 2024Layerwise Recurrent Router for Mixture-of-Experts13 Aug
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 13
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
- 2022The Stack: 3 TB of permissively licensed source codecited by 354
- 2022Sparse Upcycling: Training Mixture-of-Experts from Dense Checkpointscited by 207
- 2025Drop-Upcycling: Training Sparse Mixture of Experts with Partial Re-initializationcited by 23
- 2025Mixture of Experts Made Intrinsically Interpretablecited by 22
1 more
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Share First, Route What Remains: A Unified Framework for Token-Adaptive MoE Computation
- 2026Evidence for Shared Routing Geometry and Dynamics in Sparse Mixture-of-Experts
- 2026Beyond Routing: Decoupling Expert Dispatch and Aggregation in Sparse Mixture-of-Experts
- 2026Path-Constrained Mixture-of-Experts
- 2026Hierarchical Copula-Gumbel-Top-K Routing: Two-Sided Dependence Control for Frozen Mixture-of-Experts at Fixed Per-Token Routing Laws
- 2026Routing-Free Mixture-of-Experts
- 2025Coupling Experts and Routers in Mixture-of-Experts via an Auxiliary Loss
- 2026Towards a Statistical Understanding of Mixture-of-Experts