A Persistent State for Auditable Mixture-of-Experts Routing
Abstract
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.
1 Introduction
Mixture-of-Experts (MoE) Transformers increase model capacity while keeping per-token computation sparse by routing each token to a small subset of experts (Shazeer et al., 2017; Lepikhin et al., 2021; Fedus et al., 2022). Because routing is repeated throughout the network, understanding why a later router favors one expert over another requires reasoning about influences that may accumulate across depth. Standard MoEs expose no routing-specific record of that history: routing is computed from the same general hidden stream used by the rest of the model. Cross-layer routing influences can be reconstructed post hoc, and prior architectures have introduced explicit memory between routers, but these approaches do not make write-level routing provenance a first-class object of the forward computation.
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. A learned write updates this state after each MoE layer. The realized post-update state changes form a routing ledger: for any later router margin, the state-mediated contribution decomposes exactly into contributions from earlier realized writes. We do not claim that telescoping a recorded recurrent trajectory is unique to SA-MoE; the same algebra can be applied to other recorded recurrent states. The architectural distinction is that SA-MoE defines learned write events as the native provenance object, with a direct source-level intervention semantics.
We test whether this provenance object is useful rather than merely algebraically available. Across sparsely upcycled SmolLM2- and Gemma-based models and three independent training seeds per architecture, trained routers strongly use the state pathway. Relative to a matched latest-write-only control, persistent accumulation increases long-horizon future-routing accessibility by 19.4 and 12.2 percentage points, respectively, and more than 90% of the ledger’s absolute contribution comes from non-recent writes in both families. Ledger-selected writes also affect later routing and output distributions under full-forward suppression. At the same time, sensitivity-aware gradientwrite can provide stronger intervention rankings, and post-hoc attribution on matched ordinary MoEs recovers related cross-layer influences without architectural modification.
Our contribution is therefore an MoE routing design in which one cross-layer pathway carries an explicit, inspectable computational history. SA-MoE couples a compact routing-specific state with write-level provenance, retrospective routing queries, and direct intervention on the recorded source events. This provides a complementary operating point to recurrent routing memory, hidden-stream probing, and post-hoc attribution rather than replacing them.
2 Background and Related Work
2.1 Sparse MoE Routing and Cross-Layer Memory
Sparse MoEs activate only a small subset of experts for each token (Shazeer et al., 2017; Lepikhin et al., 2021; Fedus et al., 2022), making routing a repeated discrete allocation decision across model depth. Cross-layer routing memory itself is not new. RMoE (Qiu et al., 2025) uses a GRU-based recurrent router that carries routing-specific information between successive MoE layers. Moreover, if a recurrent trajectory is recorded, any fixed linear target readout admits the identity
including when the recurrence generating is nonlinear. Exact telescoping of realized state differences is therefore not unique to SA-MoE.
SA-MoE instead makes the update itself an explicit audit interface. A learned write proposal produces a realized post-clamp write that is stored as the source-level provenance object and can subsequently be queried or suppressed at its source. Our latest-write-only control tests persistence against replacement while retaining this state-and-write pathway. Memory-Aware Routing (Hou et al., 2026) addresses a different form of memory, maintaining expert-level routing preferences to encourage stable specialization rather than a per-token state propagated across model depth.
2.2 Architecture-Level MoE Interpretability
Other work modifies MoE computation to make internal structure more interpretable. MoE-X (Yang et al., 2025) relates experts to a wide sparse MLP and encourages sparse expert computation, while MONET (Park et al., 2025) incorporates sparse dictionary-learning ideas into MoE training to promote more interpretable expert structure. These approaches target semantic or feature-level interpretability. SA-MoE does not require experts or individual state coordinates to have semantic meanings; its audit object is instead the computational provenance of one routing-specific pathway.
2.3 Post-Hoc Attribution and Causal Validation
A complementary line of work reconstructs routing influences after training. Li et al. (2025) propose cross-level knowledge attribution for ordinary MoE computation, while Li et al. (2026) recursively decompose later router inputs into contributions from earlier model components and identify cross-layer routing influences that can persist across depth. We therefore do not claim that cross-layer routing attribution requires a dedicated routing state. The distinction is where the audit object originates: these post-hoc methods reconstruct contributions from ordinary model computation, whereas SA-MoE places a routing-specific state and explicit write events inside the natural forward computation. We evaluate the method of Li et al. (2026) directly on matched Ordinary MoEs rather than treating it only as related work.
Attribution also differs from causal sensitivity. Activation replacement, ablation, and patching are widely used to test whether identified components affect model behavior (Meng et al., 2022; Conmy et al., 2023), including recent analyses that intervene directly on MoE routing (Bandarkar et al., 2026; Xu et al., 2026). Our full-forward intervention follows the same general logic: a selected realized write is removed at its source and all subsequent computation is allowed to recompute. Gradient-based rankings provide a complementary sensitivity-aware comparison. Accordingly, our experiments distinguish exact forward provenance, post-hoc attribution, and intervention sensitivity rather than treating them as the same objective.
3 Scratchpad-Augmented Mixture-of-Experts
We study a sparse MoE architecture in which routers receive a routing-specific cross-layer state, while the experts themselves do not. At MoE layer , the ordinary token representation is and the routing state is . The router computes expert logits
| (1) |
after which Top- routing selects the routed experts. As in an ordinary sparse MoE, the selected routed experts and a shared expert contribute to the next hidden representation . The key difference is that only the router reads ; the experts operate on alone.
A learned write head updates the routing state after each MoE layer. Let denote the write proposal produced from the layer’s shared representation. In the persistent architecture, the state evolves as
| (2) |
We define the realized write
| (3) |
which is the post-clamp state change actually retained by the model.
Exact routing ledger.
Consider a target router at layer and an audited margin between experts and arising from the state pathway alone:
| (4) |
Because the persistent state is the accumulated sum of realized writes,
| (5) |
the same margin decomposes exactly as
| (6) |
We call this exact write-level decomposition the routing ledger. It is exact for the persistent-state contribution to the audited router margin. Other influences on routing, including those mediated through the ordinary hidden stream , remain outside this decomposition.
Controls.
We compare Persistent SA-MoE against two matched controls:
Ordinary MoE removes the routing-state pathway entirely, so routing depends only on .
Latest-write-only keeps the same state width and write-head capacity, but removes cross-layer accumulation by replacing the state rather than accumulating it. Thus the router at layer receives only the most recently retained write-derived state. Because this recurrence is a replacement rather than an accumulation, its native audit object is not a telescoping sum of earlier state differences. Instead, its direct native attribution is structural: the currently retained state is attributed to the latest retained write.
Figure 1 summarizes the architecture and the resulting exact ledger for the persistent pathway.
4 Experimental Setup
Model families and training.
We study two sparsely upcycled (Nakamura et al., 2025; Komatsuzaki et al., 2023) model families based on SmolLM2-135M (Ben Allal et al., 2025) and Gemma-270M (Gemma Team, 2025). For each family, we train three matched sparse architectures: an Ordinary MoE baseline, a latest-write-only control, and Persistent SA-MoE. All three variants use the same sparse backbone, with 16 routed experts, one shared expert, Top-2 routing, and matched continued-pretraining budgets of approximately 30B packed input tokens. The two state-augmented variants use a 128-dimensional routing state. Training data are sampled from FineWeb-Edu (Penedo et al., 2024), The Stack Dedup Python (Kocetkov et al., 2023), and FineMath (Ben Allal et al., 2025) with probabilities , , and . We train three independent seeds (42, 43, and 44) for each architecture in each family. Full optimization, initialization, and implementation details are given in Appendix A.
Capability evaluation.
Before evaluating auditability, we first test whether adding the routing-state pathway materially changes model quality. We report validation perplexity and a four-task zero-shot downstream suite consisting of HellaSwag (Zellers et al., 2019), ARC-Easy (Clark et al., 2018), PIQA (Bisk et al., 2020), and WinoGrande (Sakaguchi et al., 2020). For the downstream suite, we report the unweighted macro average of the task primary metrics. We evaluate family-level capability preservation using frozen non-inferiority thresholds: at most a relative increase in perplexity and at most a percentage point decrease in downstream macro score relative to the matched Ordinary MoE. Appendix B gives the exact prompts, scoring rules, and task-specific metrics.
State use and persistence.
We first ask whether trained routers actually use the routing state. For this analysis, we locally intervene on the state contribution to a router while holding the ordinary hidden representation fixed, and measure the resulting change in the selected Top-2 expert set. We then compare Persistent SA-MoE against the latest-write-only control to test whether cross-layer accumulation provides additional value beyond a same-capacity non-persistent state pathway.
Future-routing accessibility.
To test whether the routing state compactly exposes future-routing information, we train probes to predict a later router’s Top-1 expert from either the native routing state or representations derived from the ordinary hidden stream. The main text compares three representation families: the native 128-dimensional routing state, a full-hidden linear probe, and a supervised 128-dimensional hidden subspace. We evaluate probe-training budgets of , , and examples. Appendix B.6 gives the full probe grids, fit selection rules, and supporting PCA and random-projection controls.
Attribution and intervention protocol.
Our confirmatory intervention study uses 24 fixed final-evaluation sequences per model family, spanning prose, code, and mathematics, and 96 target routing decisions per family. For each target decision at layer , we audit the Top-2 membership boundary between the clean second-selected expert and the clean best-excluded expert . For Persistent SA-MoE, the ledger score of source layer is the absolute write contribution from Eq. 6. We compare this ranking with recency, largest-write-norm, and gradientwrite baselines. We also apply the post-hoc cross-layer attribution method of Li et al. (2026) to matched Ordinary MoEs. Because SA-MoE writes and Ordinary-MoE residual components are distinct audit objects, we evaluate attribution quality within each object and do not compare raw intervention magnitudes across architectures. For latest-write-only, the native direct score is defined over its structural latest-state audit object rather than over a telescoping sum of earlier state differences.
We then perform full-forward source suppression. In Persistent SA-MoE, suppressing a source write restores the post-source state to its pre-write value and allows the rest of the model to recompute naturally. In latest-write-only, suppression restores the previous retained state. We measure the resulting absolute change in the audited target margin, the divergence of later Top-2 routing decisions, and the KL divergence between the clean and intervened output distributions at the selected token. Appendix D gives the full deterministic selection procedure, ranking definitions, and control rules.
Uncertainty.
For architecture-level claims, the training seed is the replication unit. We form paired seed-level contrasts and report three-seed means with 95% Student- intervals (, degrees of freedom). For within-checkpoint intervention quantities, uncertainty is estimated with paired sequence-level bootstrap resamples; these bootstrap samples are not treated as additional training replicates.
| Family | Architecture | Params | Active/token | FLOPs/token | PPL (%) | Macro (pp) |
|---|---|---|---|---|---|---|
| SmolLM2 | Ordinary MoE | 1.409B | 294.0M | 729.6M | Ref. | Ref. |
| Latest-write-only | 1.411B | 296.3M | 734.1M | [] | [] | |
| Persistent | 1.411B | 296.3M | 734.1M | [] | [] | |
| Gemma | Ordinary MoE | 1.401B | 409.8M | 876.2M | Ref. | Ref. |
| Latest-write-only | 1.402B | 411.4M | 879.2M | [] | [] | |
| Persistent | 1.402B | 411.4M | 879.2M | [] | [] |
5 Results
5.1 Model Quality and Architectural Cost
We first ask whether adding the routing-state pathway materially changes model quality or architectural cost. Table 1 reports paired architecture contrasts across the three training seeds. Persistent SA-MoE adds less than active parameters and analytical forward FLOPs per token in both model families.
On SmolLM2, Persistent changes validation perplexity by (95% CI ) and the four-task macro score by pp ( pp) relative to Ordinary MoE. Latest-write-only also remains within both frozen non-inferiority bounds. Thus, both state-augmented variants satisfy the pre-specified SmolLM2 capability criterion.
On Gemma, the perplexity criterion is supported for both variants: Persistent changes perplexity by (), while latest-write-only changes it by (). The downstream macro contrasts are pp ( pp) for Persistent and pp ( pp) for latest-write-only. Because both intervals cross the pre-specified pp boundary, the complete Gemma capability criterion is inconclusive rather than supported.
These results establish a small architectural overhead and no detected perplexity degradation beyond the frozen tolerance in either family, while leaving downstream capability preservation on Gemma unresolved.
Matched single-GPU measurements show implementation-dependent wall-clock effects with opposite directions across the two families; we therefore report these measurements descriptively in Appendix A.7 rather than treating them as evidence of a general speedup or slowdown.
5.2 Trained Routers Use the Persistent State
The routing ledger is only useful if trained routers actually rely on the state pathway. We therefore measure direct router dependence with a local same-state intervention. For a natural forward-pass pair , we attenuate only the state’s contribution to the router logits,
| (7) |
The hidden representation and state are held fixed, and later model computation is not rerun. This therefore measures the direct dependence of the current routing decision on the explicit state rather than a full-trajectory intervention.
Figure 2a shows a graded response as the state contribution is removed. At full suppression (), the selected Top-2 expert set changes for a three-seed mean of 87.6% of SmolLM2 routing decisions and 69.9% of Gemma decisions. The corresponding seed-level intervals are and . Thus, in both model families, the trained routers make substantial use of the explicit routing state.
5.3 Persistence Improves Long-Horizon Routing Accessibility
State use alone does not establish that cross-layer accumulation matters: a router could rely on a state pathway whose useful information is effectively local. We therefore compare Persistent SA-MoE with the matched latest-write-only control.
Our primary persistence endpoint measures how well the native routing state predicts routing four or eight MoE layers ahead. At the -example probe budget, we average transition-only source-cluster accuracy over the four pre-specified horizon-4 and horizon-8 source–target pairs within each family. Figure 2b reports the paired Persistent-minus-latest-write-only contrasts across training seeds.
Persistent improves long-horizon routing accessibility by pp on SmolLM2 (95% CI pp) and pp on Gemma ( pp). The contrast is positive in all three training seeds for both families. Thus, persistence changes the information carried by the native routing state in a way that remains visible many layers later.
The intervention-derived persistence contrasts in Figure 2b are less uniform: their three-seed means favor Persistent, but their confidence intervals cross zero. We therefore treat long-horizon accessibility as the replicated evidence for an effect of persistent accumulation, rather than claiming that persistence improves every downstream intervention metric.
The retained history is also distributed across depth rather than dominated by the most recent update. Across training seeds, non-recent writes account for 91.7% of absolute ledger contribution on SmolLM2 and 90.3% on Gemma, with effective source supports of 9.49 and 5.82 writes, respectively. These quantities describe the distribution of absolute forward contribution; they do not measure signed net support or downstream causal importance. Full definitions and per-seed results are given in Appendix C.
5.4 Compact Access to Future Routing
We next ask whether the persistent state provides a compact representation of future routing, rather than information that is uniquely available to SA-MoE. Figure 3 compares the native 128-dimensional routing state with a supervised 128-dimensional representation learned from the same Persistent model’s ordinary hidden stream, across probe-training budgets from to examples.
The Persistent native state substantially outperforms the matched latest-write-only state at every probe budget in both model families. At examples, the Persistent-minus-latest-write-only gap is pp on SmolLM2 (95% CI pp) and pp on Gemma ( pp). This is consistent with the long-horizon persistence result in Section 5.3, while using the full frozen probe aggregation rather than only the pre-specified long-horizon endpoint.
The ordinary hidden stream nevertheless contains comparable routing information. At the largest training budget, the supervised hidden subspace reaches 83.4% accuracy on SmolLM2 and 84.1% on Gemma, compared with 81.2% and 77.4% for the Persistent native state. Thus, SA-MoE does not make future-routing information uniquely available. Its architectural distinction is that a compact routing-specific representation is already present in the model’s native forward state, without first learning a separate post-hoc representation.
5.5 Attribution Quality and Downstream Consequences
Exact forward provenance need not coincide with the source whose removal has the largest downstream effect. We therefore compare clean-trajectory attribution scores with exhaustive full-forward source interventions. In the frozen seed-42 analysis, ledger scores correlate with intervention effects more strongly than largest-write-norm: within-target Spearman correlations are versus on SmolLM2 and versus on Gemma. Gradientwrite is stronger still, reaching and , respectively. The same ordering holds for top-source accuracy and normalized oracle regret (Appendix D). This distinction is expected: the ledger records exact forward contribution within the state pathway, whereas gradientwrite additionally incorporates target-dependent downstream sensitivity.
Useful cross-layer attribution is also available without architectural modification. On matched Ordinary MoEs, the Li et al. (2026) post-hoc attribution achieves three-seed mean Spearman correlations of on SmolLM2 and on Gemma; gradientcomponent reaches and . These values are not directly comparable with the SA-MoE numbers because state writes and Ordinary-MoE residual components are different audit objects. Operationally, if the required forward-pass quantities are retained, both ledger and Li-style scores can be materialized without a target-dependent backward pass, while gradient-based scores require target-dependent autograd. Phase-separated timing, memory, retention, and reconstruction measurements are reported in Appendix D.8.
Finally, ledger-selected writes have consequences beyond the scalar margin used to rank them (Figure 4). Across training seeds, suppressing the Persistent ledger-selected write changes the audited margin by on SmolLM2 and on Gemma, produces downstream Top-2 divergences of and , and yields selected-token output KL divergences of and , respectively. Latest-write-only interventions also produce downstream effects, and the Persistent-minus-latest-write-only contrasts are not uniformly resolved across training seeds. These results show that ledger-selected writes are computationally consequential, but do not imply that exact provenance is an optimal causal ranking or that persistence uniformly increases every intervention effect.
6 Discussion and Limitations
The contribution of SA-MoE is not that a recorded recurrent state admits a telescoping decomposition. For any recurrent trajectory and fixed linear target readout , ; this observation also applies to nonlinear recurrent routers such as RMoE. SA-MoE instead makes routing provenance an explicit interface of the forward computation: a dedicated routing state is updated through a learned write path, its realized post-clamp writes form the native audit object, and individual writes have a direct source-level intervention semantics. Empirically, trained routers use this pathway, persistence increases long-horizon routing accessibility relative to replacement, and the resulting history is distributed across earlier writes.
Exact provenance should also be distinguished from causal importance. The ledger exactly decomposes the state-mediated component of a realized routing margin, but it does not cover influences carried through the ordinary hidden stream or guarantee the strongest intervention ranking. Gradientwrite can rank consequential sources more strongly because it incorporates downstream sensitivity, while Li et al. (2026)’s post-hoc method recovers related cross-layer attribution in ordinary MoEs without architectural modification. Likewise, sufficiently trained hidden-state representations can match or exceed the native state’s future-routing accuracy. The value of the ledger is therefore its architecture-native routing-specific audit object, not exclusive access to routing information or universal superiority over post-hoc analysis.
Our latest-write-only control also does not hold state geometry fixed. Its replacement state is bounded by the write to , whereas Persistent can accumulate toward the clamp. The comparison therefore tests accumulation versus replacement under the state distributions produced by those mechanisms; differences in state magnitude and saturation cannot be separated from accumulation itself. State-geometry diagnostics and additional control details are reported in the appendix.
Finally, the experiments cover two sparsely upcycled model families and three training seeds per architecture. This provides replication of the principal effects but leaves wide uncertainty for some intervention contrasts and does not establish behavior at larger MoE scales. More fundamentally, the ledger audits only the persistent routing-state pathway: it does not assign semantic meaning to state coordinates, provide a complete causal explanation of expert selection, or imply improved generation quality or safety.
7 Conclusion
We introduced SA-MoE, an MoE architecture that makes one cross-layer routing pathway explicitly auditable. A compact persistent state is updated by learned write events, and the realized writes exactly decompose the state-mediated contribution to later routing margins. Across two sparsely upcycled model families and three training seeds, trained routers use this pathway, persistence increases long-horizon routing accessibility relative to a matched latest-write-only control, and the resulting ledger is distributed across earlier writes. Full-forward suppression further shows that ledger-selected writes can affect subsequent routing and model outputs.
The routing ledger is not a complete explanation of expert selection, nor is exact provenance equivalent to optimal causal ranking. Post-hoc attribution can recover related cross-layer influences without architectural modification, and sensitivity-aware methods can identify stronger intervention targets. SA-MoE instead provides a complementary design point: routing-specific provenance is represented directly in the model computation as a compact state, an explicit sequence of source events, and a native object for retrospective query and intervention.
AI use statement
We used generative AI tools to assist with experimental methodology, interpretation and presentation of results, code generation and implementation, and manuscript preparation. For the implementation work, generative AI was used to help write, modify, and debug research code used in model training, evaluation, probing, intervention analysis, and result processing. We also used generative AI to draft and edit manuscript text, improve organization and readability, check consistency between claims and reported results, identify relevant literature, assist with literature searches and reference formatting, and suggest titles, captions, and section structure.
All AI-assisted work was reviewed by the authors. AI-assisted code was inspected, tested, and validated against expected behavior and experimental outputs before being used for reported results. Numerical claims and interpretations were checked against the underlying experimental results and figures, and suggested citations and descriptions of prior work were verified against the original sources before inclusion. Generative AI was not treated as a source of experimental evidence, and the authors made the final decisions about the methodology, implementation, analysis, claims, and manuscript content. We take responsibility for the final content of this work, including text, code, claims, and artifacts produced with the aid of generative AI.
Ethics statement
This work studies model architecture and interpretability and does not involve human participants, user studies, or the collection of new personal or sensitive data. Our experiments use existing language models and research datasets comprising web text, code, and mathematical content. These source models and datasets may inherit biases, undesirable content, or other limitations from their underlying corpora. SA-MoE is intended as a mechanism for auditing one routing-specific computational pathway; the resulting provenance should not be interpreted as a guarantee of complete interpretability, model safety, or reliable behavior in deployment. We therefore avoid making claims of improved safety or generation quality from the intervention results reported in this work.
Reproducibility statement
Section 3 defines the SA-MoE architecture, realized routing writes, and routing-ledger equations. Section 4 specifies the model families, matched controls, evaluation roles, intervention design, and statistical treatment used for the main results. Appendix A provides dense-source revisions, MoE construction, initialization, continued-pretraining settings, parameter and compute accounting, runtime measurements, and artifact identities. Appendix B specifies validation and capability evaluation, local state suppression, and the complete future-routing probe protocol. Appendix C defines the Persistent ledger, numerical reconstruction checks, distributed-history metrics, the distinct latest-write-only audit object, and architecture-specific suppression semantics. Appendix D gives deterministic target selection, source-ranking methods, full-forward intervention metrics, ordinary-MoE post-hoc attribution, and audit-cost measurements. Finally, Appendix E reports the individual results for training seeds 42, 43, and 44 and explicitly identifies analyses that instead use a fixed checkpoint or diagnostic pilot. These specifications cover the procedures and statistical units underlying the results in Sections 5.1–5.5.
References
- Multilingual routing in mixture-of-experts. In International Conference on Learning Representations, Cited by: §2.3.
- SmolLM2: when smol goes big – data-centric training of a small language model. External Links: 2502.02737, Link Cited by: §A.5, §4.
- PIQA: reasoning about physical commonsense in natural language. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 34, pp. 7432–7439. External Links: Document, Link Cited by: §4.
- Think you have solved question answering? try ARC, the AI2 reasoning challenge. arXiv preprint arXiv:1803.05457. External Links: Link Cited by: §4.
- Towards automated circuit discovery for mechanistic interpretability. In Advances in Neural Information Processing Systems, Vol. 36, pp. 16318–16352. External Links: Document, Link Cited by: §2.3.
- Switch transformers: scaling to trillion parameter models with simple and efficient sparsity. Journal of Machine Learning Research 23 (120), pp. 1–39. External Links: Link Cited by: §1, §2.1.
- Gemma 3. External Links: Link Cited by: §4.
- From pseudo-balancing to true specialization: memory-aware routing for mixture-of-experts. In Findings of the Association for Computational Linguistics: ACL 2026, Cited by: §2.1.
- The stack: 3 TB of permissively licensed source code. Transactions on Machine Learning Research. External Links: Link Cited by: §A.5, §4.
- Sparse upcycling: training mixture-of-experts from dense checkpoints. In The Eleventh International Conference on Learning Representations, External Links: Link Cited by: §4.
- GShard: scaling giant models with conditional computation and automatic sharding. In International Conference on Learning Representations, External Links: Link Cited by: §1, §2.1.
- Decoding knowledge attribution in mixture-of-experts: a framework of basic-refinement collaboration and efficiency analysis. In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), Vienna, Austria, pp. 22431–22446. External Links: Document, Link Cited by: §2.3.
- Understanding cross-layer contributions to mixture-of-experts routing in LLMs. In The Fourteenth International Conference on Learning Representations, External Links: Link Cited by: §D.7, §2.3, §4, §5.5, §6.
- Locating and editing factual associations in GPT. In Advances in Neural Information Processing Systems, Vol. 35, pp. 17359–17372. External Links: Document, Link Cited by: §2.3.
- Drop-upcycling: training sparse mixture of experts with partial re-initialization. In The Thirteenth International Conference on Learning Representations, External Links: Link Cited by: §4.
- Monet: mixture of monosemantic experts for transformers. In The Thirteenth International Conference on Learning Representations, External Links: Link Cited by: §2.2.
- The FineWeb datasets: decanting the web for the finest text data at scale. Advances in Neural Information Processing Systems 37, pp. 30811–30849. External Links: Document, Link Cited by: §A.5, §4.
- Layerwise recurrent router for mixture-of-experts. In International Conference on Learning Representations, External Links: Link Cited by: §2.1.
- WinoGrande: an adversarial winograd schema challenge at scale. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 34, pp. 8732–8740. External Links: Document, Link Cited by: §4.
- Outrageously large neural networks: the sparsely-gated mixture-of-experts layer. In International Conference on Learning Representations, External Links: Link Cited by: §1, §2.1.
- Seeing but not thinking: routing distraction in multimodal mixture-of-experts. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pp. 31164–31178. External Links: Document Cited by: §2.3.
- Mixture of experts made intrinsically interpretable. In Proceedings of the 42nd International Conference on Machine Learning, Vol. 267, pp. 71231–71248. External Links: Link Cited by: §2.2.
- HellaSwag: can a machine really finish your sentence?. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, pp. 4791–4800. External Links: Document, Link Cited by: §4.
- ST-moe: designing stable and transferable sparse expert models. External Links: 2202.08906, Link Cited by: §A.5.
Appendix A Model, Training, and Reproducibility
This appendix gives the model construction, initialization, optimization, analytical cost, and runtime details underlying the experiments in Section 4. Evaluation and probe specifications are given in Appendix B; ledger-specific definitions and interventions are given in Appendices C and D.
A.1 Dense Source Models and Tokenizer Provenance
We sparse-upcycle two pretrained dense language models: HuggingFaceTB/SmolLM2-135M and google/gemma-3-270m. The local model and tokenizer bytes used by the reported experiments are reproducible from the following public repository revisions:
| Family | Repository | Reproducible revision |
|---|---|---|
| SmolLM2 | HuggingFaceTB/SmolLM2-135M | 93efa2f097d58c2a74874c7e644dbc9b0cee75a2 |
| Gemma | google/gemma-3-270m | 9b0cfec892e2bc2afd938c98eabe4e4a7b1e0ca1 |
For each family, the tokenizer is taken from the same repository revision as the model; no separate tokenizer repository or tokenizer revision is used. Verification against these revisions reproduces the frozen local model and tokenizer bytes. We therefore describe them as byte-compatible reproducible revisions, rather than as recovered records of the original download event. The accompanying artifact manifests record the corresponding model, tokenizer, configuration, and source-code SHA-256 hashes.
A.2 MoE Architecture
Every dense feed-forward block is replaced with an MoE block containing 16 routed experts and one always-active shared expert. All experts retain the MLP geometry of the corresponding dense source layer. Each token executes the shared expert and the Top-2 routed experts.
| SmolLM2 | Gemma | |
| Dense source | SmolLM2-135M | Gemma-3-270M |
| Transformer layers | 30 | 18 |
| Hidden size | 576 | 640 |
| Expert intermediate size | 1536 | 2048 |
| Routed experts / layer | 16 | 16 |
| Shared experts / layer | 1 | 1 |
| Routed experts / token | 2 | 2 |
| Active expert paths / token | 3 | 3 |
| Routing-state width | 128 | 128 |
| State clamp | ||
| Sequence length | 2048 | 2048 |
| Capacity factor | none | none |
| Token dropping | none | none |
For router logits , the implementation first computes a softmax over all 16 routed experts, selects the Top-2 entries, and renormalizes their probabilities:
| (8) |
The two selected routed-expert outputs are weighted by and summed; the always-active shared-expert output is then added to this routed output. There is no capacity clipping or dropped-token branch.
Ordinary MoE routes from only. Persistent and latest-write-only route from the concatenation of and . The implemented router is a bias-free linear map; equivalently, the bias term in the general notation of Section 3 is for the models reported here.
A.3 Sparse Upcycling and Initialization
Shared and routed experts are initialized as deep copies of the pretrained dense MLP at the corresponding layer. We then partially reinitialize each routed expert independently. For every routed-expert weight tensor with at least two dimensions, each entry is selected independently with probability ; selected entries are replaced by draws from a Gaussian whose mean and standard deviation equal those of the original tensor. Routed-expert biases are not reinitialized. The shared expert remains an unchanged copy of the pretrained dense MLP.
Router weights use Kaiming-uniform initialization with parameter . Routers contain no bias. The state write head is initialized to produce an exact zero write: both its weights and bias are initialized to zero. The training seed is set before sparse upcycling and data construction.
The principal experiments use three independent training seeds, , , and , for every model family and architecture variant. Architecture comparisons are paired by seed. The same construction algorithm and dense source checkpoint are used for matched variants, but we do not claim that historical initialized parameter tensors were retained byte-for-byte across variants.
A.4 Persistent and Latest-Write-Only State Implementations
At the beginning of each model forward pass, the routing state is initialized to zero in the hidden-stream dtype. It is then threaded through successive MoE layers for each token. The write head reads the always-active shared-expert representation and computes
| (9) |
The main text suppresses the write-head bias from the notation; the implementation includes it and initializes it to zero.
For Persistent,
| (10) |
For latest-write-only,
| (11) |
There is no detached or separately retained accumulator in the latest-write-only model. Because is produced by a , its native replacement state lies in even though the common outer clamp is . The resulting state-geometry difference from Persistent is discussed further in Appendix C.5.
The model state is stored in the forward dtype; the reported training runs use bf16. Attribution calculations that require state differences or router-weight dot products convert the relevant quantities to float32, as detailed in Appendix C.
A.5 Continued-Pretraining Recipe
Training examples are sampled from FineWeb-Edu (Penedo et al., 2024), The Stack Dedup Python (Kocetkov et al., 2023), and FineMath-4+ (Ben Allal et al., 2025) with probabilities , , and , respectively. These are sampling probabilities before packing and are not asserted to equal the realized proportions of the packed token stream.
All reported runs use 30,000 optimizer steps and a global batch of 512 length-2048 sequences, corresponding to packed input tokens per optimizer step and a nominal total of
| (12) |
input tokens per training run.
| SmolLM2 | Gemma | |
| Precision | bf16 | bf16 |
| Optimizer steps | 30,000 | 30,000 |
| Sequence length | 2048 | 2048 |
| World size | 16 | 16 |
| Global sequences / step | 512 | 512 |
| Tokens / optimizer step | 1,048,576 | 1,048,576 |
| Ordinary/Persistent microbatch / GPU | 8 | 2 |
| Ordinary/Persistent grad. accumulation | 4 | 16 |
| Latest-write-only microbatch / GPU | 8 | 4 |
| Latest-write-only grad. accumulation | 4 | 8 |
| Optimizer | AdamW | AdamW |
| Base peak learning rate | ||
| Router/write-head learning rate | ||
| Shared-expert learning rate | ||
| Shared-expert frozen period | 3,000 steps | 3,000 steps |
| Warmup | 1,500 steps | 1,500 steps |
| Post-warmup schedule | cosine to peak | cosine to peak |
| Weight decay | ||
| Global gradient-norm clip | ||
| Load-balance coefficient | ||
| Router z-loss coefficient |
Routed experts and remaining base-model parameters use the base learning rate. The shared expert uses and is held at zero learning rate for the first 3,000 optimizer steps. Router and write-head parameters use . After a 1,500-step linear warmup, each parameter group follows a cosine decay to times its peak learning rate. Weight decay is applied to eligible matrix weights but not to biases or normalization parameters.
Let denote the number of routed experts, the fraction of hard Top-2 assignments to expert , and its mean pre-Top-2 softmax probability. The load-balancing term is
| (13) |
where both selected Top-2 slots contribute to . Each MoE layer also uses the router z-loss (Zoph et al., 2022)
| (14) |
Auxiliary losses are averaged across MoE layers and added to the language-model cross-entropy objective.
The Gemma latest-write-only runs use microbatch 4 with gradient accumulation 8 rather than the microbatch-2, accumulation-16 shape used by the Ordinary/Persistent Gemma runs. This changes the per-device execution shape but preserves the world size, global batch, sequences per optimizer step, and nominal token budget.
A.6 Parameter Counts and Analytical Compute
Table 4 reports the analytical model-size and active-compute quantities used in Section 5.1. Active parameters count all non-routed parameters, the always-active shared expert, and the two selected routed experts. Forward FLOPs count the reported analytical matrix-multiplication work per token and should not be interpreted as measured wall-clock latency.
| Family | Architecture | Params. | Active/token | Fwd. FLOPs/token |
|---|---|---|---|---|
| SmolLM2 | Ordinary | 1.4088B | 294.044M | 729.575M |
| Latest-write-only | 1.4111B | 296.321M | 734.122M | |
| Persistent | 1.4111B | 296.321M | 734.122M | |
| Gemma | Ordinary | 1.4007B | 409.840M | 876.192M |
| Latest-write-only | 1.4023B | 411.354M | 879.215M | |
| Persistent | 1.4023B | 411.354M | 879.215M |
Relative to Ordinary MoE, Persistent therefore increases the analytical forward FLOP count by approximately for SmolLM2 and for Gemma. Persistent and latest-write-only have identical parameter and analytical active-compute counts within a family.
A.7 Common-Hardware Runtime Measurements
We additionally measured full-sequence runtime on a common NVIDIA GH200 120GB GPU. These measurements are descriptive implementation diagnostics rather than training-seed replications. Inference measurements are full length-2048 forward passes rather than autoregressive cached decoding. The training measurement is a single-GPU forward–backward pass and excludes optimizer updates and distributed communication.
| Family | Architecture | Inf. B1 | Inf. B16 | Train B4 |
|---|---|---|---|---|
| SmolLM2 | Ordinary | 117.3 | 192.2 | 446.8 |
| Latest-write-only | 151.4 | 210.7 | 545.8 | |
| Persistent | 140.9 | 204.2 | 524.4 | |
| Gemma | Ordinary | 94.3 | 191.8 | 318.4 |
| Latest-write-only | 82.2 | 176.9 | 326.6 | |
| Persistent | 82.5 | 182.3 | 316.4 |
The direction of the measured runtime difference is not consistent across the two model families. We therefore use the analytical FLOP counts as the architecture-level compute accounting and report the wall-clock measurements only as implementation-specific descriptive results.
A.8 Source Freezes and Artifact Identity
The canonical Persistent and Ordinary-MoE implementation freeze is tied to Git commit 984f88b7f1aac9373a55d221154308cf82e1a2b8, with an empty tracked-diff hash in the corresponding freeze manifest. The latest-write-only runs are accompanied by launch manifests that bind the model source, baseline source, configuration, upcycling code, trainer, software environment, and run configuration to cryptographic hashes.
Result and configuration artifacts use SHA-256 identities over their exact serialized bytes. The training-seed registry additionally records a structured content hash computed from compact canonical JSON with sorted keys, comma/colon separators, UTF-8 encoding, and no trailing newline; the human-readable on-disk registry is stored as pretty, sorted JSON with a trailing newline. These identities separate the logical checkpoint record from filenames or storage locations.
The main replicated architecture-level results use training seeds , , and , paired by seed within each model family. Analyses that use a fixed checkpoint rather than all three training realizations are labeled explicitly at the point of use in the later appendices.
Appendix B Evaluation Protocols and Future-Routing Probes
This appendix specifies the evaluation pools, capability metrics, local state-suppression analysis, and future-routing probes used in Sections 4 and 5. Training and model-construction details are given in Appendix A; ledger and intervention definitions are given in Appendices C and D.
B.1 Analysis Roles and Evaluation Pools
We separate analysis construction from final evaluation. Construction data are used to choose layer sets, representations, controls, and aggregation rules; where model selection is required, a separate calibration role is used. Corresponding final-evaluation examples are not inspected until those choices are frozen. Throughout the paper, “held out” refers to this analysis-role separation. It does not imply that the underlying documents were absent from the pretraining corpus of the original dense source model.
The language-model and local routing-state analyses use a balanced three-domain final-evaluation pool. For each model family, it contains 96 sequences of length 2048: 32 prose sequences from the local FineWeb-Edu sample, 32 code sequences from The Stack Dedup Python, and 32 mathematics sequences from FineMath-4+. Stable sequence identities are retained across matched architecture variants so that paired comparisons operate on the same source material.
The full-forward attribution experiment uses a smaller, independently fixed subset of this evaluation material. Its sequence and target-selection procedure is specified in Appendix D.
B.2 Validation Perplexity
The validation-perplexity quantity reported in Section 5.1 is computed on the 96-sequence balanced-domain pool described above; it is not perplexity on a standard external validation corpus.
Packed examples contain pre-aligned next-token labels. No additional label shift is applied during evaluation. Each length-2048 packed sequence contains one aligned label for each of its 2048 input positions, including the final input position, so each domain contributes exactly
valid labels and the full evaluation contains
valid labels per model evaluation.
Let denote the set of valid token positions and the next-token negative log likelihood at position . We compute
| (15) |
Thus the point estimate is token weighted, both within and across domains. We do not compare absolute perplexities between SmolLM2 and Gemma because the two model families use different tokenizers.
For each architecture contrast and training seed, the same 96 source sequences are evaluated under both matched variants. Relative perplexity is formed as
| (16) |
where is the training seed. Cross-seed inference is performed on ; the mean and Student- interval are transformed back to the ratio scale. Percentage changes reported in the main text are . Training seed, rather than token count or bootstrap replicate, is the replication unit for these architecture-level claims.
B.3 Downstream Capability Evaluation
We evaluate HellaSwag, ARC-Easy, PIQA, and WinoGrande zero-shot (num_fewshot=0), without a system instruction, chat template, or free-form generation. Request construction follows the corresponding EleutherAI lm-evaluation-harness task definitions at Git commit b954108c9baaaa934b4ad842033b31a97ee30816. Our local request construction and per-example metric implementation were checked against these pinned task definitions before the canonical evaluations.
For HellaSwag, ARC-Easy, and PIQA, each candidate answer is scored by its conditional continuation log likelihood. Raw accuracy selects the candidate with largest log likelihood. For normalized accuracy, the conditional log likelihood is divided by the Python character length of the corresponding task choice before selection. Ties are resolved by the lowest candidate index.
WinoGrande uses its task-specific harness construction rather than treating the candidate word itself as an independently scored continuation. Each candidate is inserted into the task prefix and the conditional likelihood of the remaining suffix is scored; the higher-scoring completed construction is selected.
The primary task metrics are
- •
normalized accuracy for HellaSwag;
- •
normalized accuracy for ARC-Easy;
- •
normalized accuracy for PIQA; and
- •
raw accuracy for WinoGrande.
The reported macro score is their unweighted arithmetic mean:
| (17) |
The evaluated splits and complete example counts are shown in Table 6. Inputs are limited to 2048 tokens. When truncation is required, context is left-truncated while retaining the complete scored continuation and at least one context token.
| Task | Split | Examples | Primary metric |
|---|---|---|---|
| HellaSwag | validation | 10,042 | normalized accuracy |
| ARC-Easy | test | 2,376 | normalized accuracy |
| PIQA | validation | 1,838 | normalized accuracy |
| WinoGrande | winogrande_xl validation | 1,267 | raw accuracy |
Table 7 gives the three-seed mean task-level changes relative to the matched Ordinary MoE. Values are absolute percentage-point changes in the corresponding primary metric. Full seed-by-seed values and Student- intervals are given in Appendix E.
| Family | Architecture | HellaSwag | ARC-Easy | PIQA | WinoGrande | Macro |
|---|---|---|---|---|---|---|
| SmolLM2 | Latest-write-only | |||||
| Persistent | ||||||
| Gemma | Latest-write-only | |||||
| Persistent |
B.4 Local Routing-State Suppression
The state-use experiment in Section 5.2 is a local same-state counterfactual rather than a full-forward intervention. For each natural Persistent forward pass, we retain the hidden representation and routing state entering a router and recompute only that router under
| (18) |
As noted in Appendix A.2, the implemented routers are bias-free, so in the reported models.
Neither nor is recomputed after attenuation, and the modified expert selection is not propagated through later model layers. The primary metric is
| (19) |
where Top-2 is treated as an unordered selected-expert set. The no-op endpoint is zero by definition, and removes the direct state term from the current router while leaving the naturally produced hidden representation and state fixed.
The analysis uses the 96-sequence final-evaluation pool from Appendix B.1. We first reduce the routing decisions within a checkpoint to one checkpoint-level rate. Cross-checkpoint summaries then treat training seeds 42, 43, and 44 as the three replicates and report Student- 95% intervals with two degrees of freedom. The full-suppression Persistent seed-level rates are
for SmolLM2 and
for Gemma, giving the main-text means of and , respectively.
B.5 Independent Future-Routing Probes
For each frozen source–target layer pair, a probe predicts the clean Top-1 expert index of the target router from a representation captured at the same token position at an earlier source layer. The source representation is the actual representation available at the source router: the general hidden stream for hidden-state probes and the 128-dimensional native routing state for native-state probes.
Training, calibration, and final-evaluation roles are disjoint. The frozen probe-training budgets are
association-role examples. Calibration and final evaluation each contain 98,304 examples per source layer, balanced as 32,768 examples from each of the prose, code, and mathematics domains. Each domain contribution comprises 64 selected token positions from each of 512 sequences. The final-evaluation role is not loaded until representation construction, hyperparameter selection, and fitted-model identities are frozen.
The frozen source and target layers are shown in Table 8.
| Family / source | Horizon 1 | Horizon 4 | Horizon 8 |
|---|---|---|---|
| SmolLM2, source 7 | |||
| SmolLM2, source 14 | |||
| Gemma, source 4 | |||
| Gemma, source 8 |
B.6 Probe Representations and Fit Selection
We evaluate the native 128-dimensional routing state and five representations of the model’s ordinary hidden stream:
- 1.
a full-hidden multinomial linear predictor;
- 2.
a one-hidden-layer MLP with a 128-unit ReLU bottleneck;
- 3.
a supervised 128-dimensional hidden subspace;
- 4.
a 128-dimensional PCA representation; and
- 5.
a 128-dimensional random projection, averaged over five independently fixed projections.
Unless explicitly stated otherwise, a hidden-state representation is taken from the same architectural variant being evaluated. In particular, the “Persistent hidden subspace-128” curve compared with the Persistent native state in Figure 3 is learned from the Persistent model’s ordinary hidden stream; it is not an Ordinary-MoE baseline.
Captured hidden and native-state features are stored from the natural forward pass and converted to float32 before fitting. A StandardScaler is fit on probe-training examples only and then applied to calibration and final examples. Learned projected representations receive a second scaler fit only on their projected training features.
For the linear predictors, the candidate regularization values are
Fitting uses scikit-learn 1.9.0 multinomial logistic regression with solver=lbfgs, max_iter=1000, and tol=1e-4. Calibration macro-F1 selects the candidate; ties are resolved in favor of smaller .
For the nonlinear bottleneck, the candidate pairs are
The classifier has one 128-unit ReLU hidden layer and uses Adam with batch size 256, max_iter=1000, tol=1e-4, n_iter_no_change=20, early_stopping=False, and deterministic shuffling. Calibration macro-F1 again selects the candidate. The frozen candidate order implements the tie rule of larger first and then smaller learning rate.
PCA uses randomized SVD with 128 components and a deterministic seed. The supervised hidden subspace is constructed from the row space of the fitted full-hidden linear classifier: the classifier coefficient matrix is converted to float64, decomposed by SVD, and deterministically orthogonally completed to 128 dimensions where necessary. The resulting 128-dimensional representation is then used as the input to the final probe. Random-projection results average five independently fixed 128-dimensional projections.
All frozen source-layer fits used for the reported analyses converged before final evaluation. Hyperparameter selection uses calibration data only; final evaluation is not used to choose a representation or fitting candidate.
B.7 Transition-Only Diagnostic
Figure 3 reports a diagnostic subset called transition-only. This term refers only to a transition in numeric expert-index sets. Let
be the clean Top-2 numeric expert indices at source and target layers for one example. After sorting each pair, the example is retained exactly when
| (20) |
Top-2 order is therefore ignored.
Because experts at different layers are distinct modules, equality of a numeric index across layers does not imply persistence of a physical or semantic expert identity. Conversely, the mask in Equation 20 should not be interpreted as a semantic expert-change label. It is a fixed numeric-index diagnostic intended to remove examples on which the source and target Top-2 index sets happen to be identical.
For retained examples, transition-only token accuracy is ordinary Top-1 classification accuracy:
| (21) |
where is the mask defined above, is the clean target Top-1 numeric expert index, and is the probe prediction. The metric is undefined only if is empty; no reported frozen pair has an empty final-evaluation mask.
B.8 Probe Aggregation and the Two Persistence Summaries
Two related but intentionally different probe aggregates appear in the main paper. We define them separately here to avoid conflating their numerical values.
Long-horizon persistence endpoint.
The pre-specified persistence endpoint in Section 5.3 and Figure 2b uses only horizons 4 and 8 at the -example probe-training budget. Thus, for each family,
containing four source–target pairs.
For training seed , let and denote the final transition-only native-state accuracy for Persistent and latest-write-only on pair . The seed-level endpoint is
| (22) |
Training seeds are then treated as the replication unit.
The resulting seed-level contrasts, in percentage points, are
| Family | Seed 42 | Seed 43 | Seed 44 | Mean [95% CI] |
|---|---|---|---|---|
| SmolLM2 | 21.97 | 19.33 | 16.99 | 19.43 |
| Gemma | 9.68 | 15.09 | 11.75 | 12.17 |
These are the persistence values reported in Section 5.3. In particular, the previously reported 21.97 pp and 9.68 pp values are the seed-42 entries, not the three-seed means.
Figure 3 learning-curve aggregate.
The probe-learning curves instead use the equal-pair mean over all six frozen source–target pairs in Table 8, including horizons 1, 4, and 8. For representation , budget , and training seed ,
| (23) |
Any frozen probe-fit reduction is performed within checkpoint before this equal-pair mean; the three training checkpoints remain the replication units for the Student- intervals shown in the figure.
At the -example budget, the corresponding Persistent-minus-LWO native-state contrasts are
These differ from Equation 22 because the Figure 3 aggregate additionally includes the two horizon-1 pairs.
| Family | Representation | 10k | 100k | 1M |
|---|---|---|---|---|
| SmolLM2 | Persistent native state | 79.28 | 81.06 | 81.25 |
| Latest-write-only native state | 56.87 | 58.55 | 58.73 | |
| P hidden-128 | 77.76 | 82.21 | 83.37 | |
| Gemma | Persistent native state | 74.95 | 77.15 | 77.41 |
| Latest-write-only native state | 64.76 | 66.77 | 66.99 | |
| P hidden-128 | 78.19 | 82.70 | 84.10 |
Thus the native Persistent state retains a large advantage over the matched latest-write-only state at each probe budget, while the supervised representation of the Persistent model’s ordinary hidden stream can match or exceed the native state with sufficient probe training. The latter comparison is why we interpret the native state as providing compact, directly available routing information rather than information that is unique to the routing-state pathway.
B.9 Additional Representation Controls
The full representation study additionally evaluates the full hidden-state linear predictor, 128-unit nonlinear bottleneck, PCA-128, and five-replicate random-projection-128 controls defined in Appendix B.6. These controls address different possible explanations for the performance of the supervised hidden subspace: unrestricted linear accessibility, nonlinear probe capacity, unsupervised low-dimensional variance, and arbitrary low-dimensional projection.
The representation comparison is not parameter matched to the native routing state: a learned predictor applied to a hidden representation may contain more fitted parameters than the model’s native 128-dimensional state pathway. Accordingly, we use these probes to characterize accessibility rather than to claim an equal-resource competition between representations. The complete per-seed numerical probe record is reported in Appendix E.
Appendix C Ledger Construction and Persistence Diagnostics
This appendix specifies the native Persistent routing ledger, its numerical reconstruction checks, the distributional summaries used in Section 5.3, and the distinct direct-state audit object used for latest-write-only. Full-forward target selection, attribution rankings, intervention outcomes, and audit-cost comparisons are given in Appendix D.
C.1 Native Persistent Ledger and Numerical Reconstruction
The multi-write routing ledger is defined only for Persistent. For consecutive natural states entering successive routers, the realized source write is
| (24) |
This is the state change that remains after the source layer’s write proposal, accumulation with the incoming state, and elementwise clamp. We never substitute the pre-clamp proposal for .
Consider a target layer , with clean Top-1 expert , clean second-selected expert , and clean best-excluded expert . Let denote the columns of the target router acting on the 128-dimensional state. We retain two margin decompositions:
| (25) | ||||
| (26) |
The second quantity is the contribution to the clean Top-2 membership boundary used throughout the attribution analyses.
Because ,
| (27) | ||||
| (28) |
These equalities concern the direct state-mediated component of the target router logits. Influences transmitted through the ordinary hidden stream are outside this decomposition.
The model forward pass uses bf16. Natural states and router logits are captured at the actual router hook and converted to float32 for attribution; the relevant router weights are likewise copied to float32. The attribution cache stores float32 realized-write vectors, target logits, and contribution tensors. Reconstruction is therefore exact algebraically but need not be bit-exact after finite-precision capture, subtraction, and summation.
Table 10 reports the frozen numerical validation on the construction checkpoints. The acceptance tolerance was absolute error. Both families remain more than an order of magnitude below this threshold.
| Family | Top-1 margin | Top-2 boundary |
|---|---|---|
| SmolLM2 | ||
| Gemma |
C.2 Distributed-History Metrics
The distributed-history analysis uses the selected-set contribution from Equation 26. For brevity, write and define the absolute attribution mass for target as
| (29) |
For , define normalized absolute source weights
| (30) |
Let denote the most recent eligible source before target . The non-recent absolute contribution share is
| (31) |
Thus “non-recent” means every eligible source except the immediately preceding source; it does not denote a depth tercile or a fixed number of layers.
We quantify concentration with inverse Herfindahl concentration:
| (32) |
We refer to as the effective source support. It equals one when all absolute mass is assigned to one source and increases as the mass becomes distributed across sources. This is an inverse-concentration quantity, not an entropy-based effective support. We additionally record the largest absolute source share
| (33) |
If , the normalized quantities in Equations 30–32 are undefined and are omitted from the corresponding Persistent summary. All frozen Persistent targets used for the reported exact-provenance analysis have nonzero attribution mass, so this rule does not remove any reported Persistent target.
The direct-provenance analysis uses five frozen target layers per family:
Each frozen target-layer summary uses the same 96-sequence evaluation pool described in Appendix B.1. Target-layer summaries are averaged with equal weight within a checkpoint. Cross-checkpoint summaries then treat training seeds 42, 43, and 44 as the replication units and use Student- 95% intervals with two degrees of freedom.
Persistent assigns most absolute direct contribution away from the latest write. The non-recent share is on SmolLM2 (95% CI ) and on Gemma (). The corresponding effective supports are writes () and writes (). The largest individual source accounts for only () of absolute mass on SmolLM2 and () on Gemma.
These quantities use absolute contribution and therefore do not measure net signed support. As a complementary signed summary, the direct state-mediated Top-2-boundary contribution, , has three-seed means on SmolLM2 and on Gemma, with 95% intervals and , respectively. Large positive and negative source terms may therefore cancel even when absolute provenance is distributed. The distribution statistics should not be interpreted as downstream causal effects; those are tested separately in Appendix D.
C.3 Latest-Write-Only Direct-State Audit Object
Latest-write-only does not use the Persistent multi-write ledger as its native audit object. Its recurrence is
| (34) |
so the state entering a later router directly contains only the most recent replacement value.
For target layer , let be the state entering the target router and let be the clean second-selected and best-excluded experts. The direct latest-state score assigned to the immediately preceding eligible source is
| (35) |
Every earlier eligible source receives direct native score zero because its replacement value has already been overwritten. The state and router-weight slice are converted to float32 before this dot product. Source-score ties are resolved by descending score and then ascending source-layer index.
The zero scores in this construction have a narrow interpretation: they mean that an earlier source has zero direct contribution through the state currently entering the target router. They do not imply zero indirect influence through the ordinary hidden stream or through consequences of earlier routing decisions.
This distinction is important for Figure 5. If one instead mechanically formed
under the latest-write-only recurrence, earlier differences would generally be nonzero and would telescope through cancellation. That is not the audit object used for the latest-write-only comparison. Under the implemented structural direct-state definition, the complete direct absolute mass belongs to the most recent replacement source; hence non-recent share is zero, effective source support is one, and largest-source share is one by construction.
Persistent and latest-write-only should therefore be understood as exposing different native direct-state provenance structures: Persistent retains a sequence of realized additive state changes, whereas latest-write-only retains one direct replacement source.
C.4 Source-Suppression Semantics
The full-forward source intervention uses the same hook operation for both state architectures but has a different recurrence-level interpretation.
For a selected token and source layer , a pre-hook first saves the state entering the source block. The source block is then allowed to run naturally, including its normal routing decision, expert computation, write proposal, and state update. A post-hook verifies the natural state transition against the clean recorded transition and then replaces the selected token’s post-source state by the saved pre-source value:
| (36) |
All subsequent model computation is then rerun naturally from this modified trajectory. Later hidden states, routers, expert selections, write proposals, states, and output logits are not held fixed.
For Persistent, the natural transition is
| (37) |
whereas the intervention uses
| (38) |
Thus the intervention removes the realized post-clamp state transition at that source. This formulation is preferable to setting the proposal to zero because saturation can make the realized write differ from the proposal.
For latest-write-only, the natural transition is
| (39) |
while the same hook gives
| (40) |
This intervention therefore skips the current replacement and restores the previous state; it does not set the current replacement value to zero. At the next latest-write-only update the explicit state is again replaced by the newly recomputed proposal. Indirect consequences can nevertheless persist, because the intervened source may alter subsequent routing, hidden representations, and therefore later write proposals.
For the scalar audited boundary effect, the identities of the clean and experts are held fixed so that the same clean margin is evaluated before and after intervention. Expert selections and other downstream quantities themselves are allowed to change. Detailed target selection and downstream metrics are specified in Appendix D.
C.5 State Geometry and Scope of the Latest-Write-Only Control
Persistent and latest-write-only match the routing-state width, write-head architecture, router input width, parameter count, and analytical active compute, but they do not induce the same state geometry. Because
the latest-write-only state is intrinsically confined to coordinate wise; its common outer clamp is therefore inactive under exact arithmetic. Persistent instead accumulates successive proposals and can reach the clamp.
Consequently, the comparison does not hold state magnitude, variance, or saturation frequency fixed. It tests persistent accumulation against replacement under the state distributions learned by those two mechanisms. We therefore do not interpret the latest-write-only comparison as isolating accumulation while controlling away every associated geometric or optimization difference.
An older fixed-checkpoint SmolLM2 diagnostic illustrates the magnitude of this difference. At source layer 14, the mean state norm was approximately for Persistent and for latest-write-only, while approximately of Persistent state coordinates were at the clamp boundary. These values are single-checkpoint diagnostics rather than three-seed estimates and are not used in the replicated statistical claims of Section 5. Probe preprocessing, including train-only standardization of native state features, is specified in Appendix B.6.
Appendix D Attribution, Full-Forward Intervention, and Audit Cost
This appendix specifies the fixed target-selection procedure, source-ranking methods, full-forward intervention metrics, exhaustive ranking diagnostics, ordinary-MoE post-hoc comparison, and audit-cost measurements underlying the attribution results in Section 5. Persistent-ledger construction and the latest-write-only direct-state audit object are defined separately in Appendix C.
D.1 Target-Sequence and Decision Selection
The full-forward intervention study uses 24 fixed final-evaluation sequences and 96 target routing decisions per model family. Sequence and position selection are deterministic and outcome independent.
For each of the prose, code, and mathematics domains, the 32 candidate final-evaluation sequences from Appendix B.1 are ranked by
| (41) |
using salt routing-ledger-confirmatory-e4-sequences-v1. Here architecture denotes the model-family identifier, smollm or gemma, rather than the Persistent, latest-write-only, or Ordinary architectural variant. Fields are UTF-8 serialized and separated by a single NUL byte. The first eight sequences in ascending hash order are retained in each domain, giving 24 sequences per family.
The fixed target layers are
Within every selected sequence–target-layer pair, candidate positions are those with a valid aligned next-token label. They are ranked by
| (42) |
using salt routing-ledger-confirmatory-e4-events-v1. The first two positions are retained. Each sequence therefore contributes four target decisions, for
targets per family.
The resulting sequence and token identities are reused across matched architectural variants. Selection does not use routing margins, ledger contributions, source rankings, intervention effects, output changes, or any other downstream outcome.
D.2 Audited Routing Boundary
For every retained target decision , let and be the clean first- and second-selected experts and let be the clean highest-logit expert excluded from the Top-2 set. The primary audited quantity is the clean Top-2 membership boundary
| (43) |
The identities and are fixed from the clean trajectory when this scalar margin is evaluated after intervention. Routing decisions themselves are not frozen.
Every MoE layer is an eligible source. This yields 1,776 source interventions over the 96 SmolLM2 targets and 1,344 over the 96 Gemma targets in the exhaustive SA-MoE boundary audit.
D.3 SA-MoE Source-Ranking Methods
All source rankings are computed from the clean trajectory before any intervention outcome is observed.
Routing ledger.
The primary ledger score is the magnitude of the exact Persistent selected-set-margin contribution:
| (44) |
Largest write norm.
This control ranks sources by
| (45) |
It tests whether the ledger merely selects unusually large state changes.
Recency.
The recency baseline ranks later eligible source layers above earlier ones; its top-ranked source is the greatest .
Gradient write.
The sensitivity-aware score is
| (46) |
The gradient is evaluated on the clean trajectory with respect to the post-source state at the same token position. The target expert identities are the fixed clean and . Thus this is a local clean-trajectory sensitivity score; the realized discrete routing configuration is not itself replaced by intervention outcomes during ranking.
Matched controls.
For comparisons against a ledger-selected source, three additional controls exclude that source. Random selection uses a deterministic SHA-256 ordering over model family, sample identity, token position, target layer, and candidate source. Norm matching selects the non-ledger source whose realized write norm is closest to that of the ledger-selected write. Distance matching selects deterministically from the same normalized source-depth tercile as the ledger-selected source. The retained target layers contain enough eligible sources that every reported depth tercile contains non-ledger candidates.
Unless otherwise specified, deterministic score ties are broken by ascending source-layer index.
The ledger-control advantage used in the replicated persistence summary is
| (47) |
where is the absolute fixed-boundary intervention effect defined below. The checkpoint-level statistic averages over the fixed target set.
D.4 Full-Forward Source Intervention and Outcome Metrics
State-source suppression follows the architecture-specific semantics in Appendix C.4. In both Persistent and latest-write-only, the selected token’s natural post-source state is replaced by its pre-source state. The model then continues normally from the modified trajectory. Later hidden representations, writes, routing decisions, experts, and output logits may all change.
For target and source , define the primary absolute fixed-boundary effect
| (48) |
We also retain the corresponding signed difference.
For clean and intervened Top-2 expert sets and , the routing-set distance is the Jaccard distance
| (49) |
Target Top-2 Jaccard distance applies Equation 49 at the audited target router. Target Top-1 change is the indicator that the clean and intervened Top-1 target experts differ. The reported downstream Top-2 divergence averages the corresponding Top-2 Jaccard distance over routing decisions strictly downstream of the audited target for the selected token.
At the selected token’s language-model output, let and denote the clean and intervened softmax distributions. Output KL is reported in the clean-to-intervention direction,
| (50) |
using natural logarithms. Output Top-1 change records whether the most likely output token changes.
For the aligned next-token label , the NLL effect is
| (51) |
so positive values indicate increased loss after intervention.
Within a checkpoint, target-level quantities are first grouped by their 24 source sequences. Confidence intervals use 2,000 paired sequence-level bootstrap resamples; multiple target layers and positions from one source sequence are not treated as independent replicates. For architecture-level comparisons across checkpoints, training seed is the replication unit and 95% Student- intervals use and two degrees of freedom.
D.5 Exhaustive Source-Ranking Metrics
The exhaustive seed-42 audit intervenes on every eligible source for every fixed target. Let
| (52) |
denote the oracle fixed-boundary effect, and let be the top-ranked source under method .
Raw oracle regret is
| (53) |
and normalized oracle regret is
| (54) |
Normalized regret is defined for targets with positive oracle effect. A zero-oracle target would be excluded from this normalized quantity and reported separately rather than used as a numerical denominator.
Top-source accuracy is one when the method-selected source belongs to the set of oracle-maximal sources and zero otherwise; hence ties among genuinely oracle-maximal sources all count as correct. Within-target Spearman is the rank correlation between the complete clean-trajectory source-score vector and the complete intervention-effect vector . It is reported only when both rankings are defined; a constant or unavailable score vector contributes no Spearman value rather than an artificial zero correlation.
D.6 Fixed-Checkpoint SA-MoE Ranking Diagnostic
Table 11 reports the exhaustive Persistent seed-42 diagnostic. These are sequence-bootstrap summaries within one trained checkpoint, not three-seed architecture-level estimates.
| Family | Method | Spearman | Top-source acc. | Normalized regret |
|---|---|---|---|---|
| SmolLM2 | Random | — | ||
| Recency | ||||
| Largest write norm | ||||
| Ledger | ||||
| Gradient write | ||||
| Gemma | Random | — | ||
| Recency | ||||
| Largest write norm | ||||
| Ledger | ||||
| Gradient write |
The ledger is therefore informative about full-forward suppression effects on its audited boundary, but exact forward contribution is not the same objective as intervention sensitivity. In this fixed-checkpoint diagnostic, gradientwrite provides the stronger ranking under all three displayed metrics. We consequently use the ledger as a provenance score, not as a claim of optimal causal-source ranking.
D.7 Ordinary-MoE Post-Hoc Attribution
We separately apply the cross-layer routing-attribution method of Li et al. (2026) to the matched Ordinary MoEs. This analysis uses ordinary-MoE residual components as its audit objects rather than SA-MoE state writes. Raw component-intervention magnitudes are therefore not pooled with or directly compared against SA-MoE write-intervention magnitudes.
The implementation was checked against the authors’ public reference code at commit fe2366a914f1da426ab7ec4d903015e229beaa72. The upstream decomposition helpers were directly parity-tested. The maximum absolute discrepancies were for the default difference breakdown and for the RMSNorm/TAM helper. The Gemma adapter uses Gemma’s effective RMSNorm scale, including the implementation’s convention.
We retain two Li-style scores. Li variance uses the reference expert-score variance criterion. Li Top-2 boundary uses the same decomposed ordinary-MoE components but adapts the final score to the fixed clean -versus- routing boundary used in our intervention benchmark. We additionally evaluate largest-component norm, recency, deterministic random ranking, and a gradientcomponent sensitivity score.
As a numerical validation of the decomposition, the maximum reconstructed router-input error is on SmolLM2 and on Gemma in the frozen validation run; maximum router-logit reconstruction error is zero for both.
Unlike the single-checkpoint SA-MoE exhaustive diagnostic above, the current ordinary-MoE ranking summaries are available for training seeds 42, 43, and 44. Table 12 therefore reports three-seed means with Student- intervals. The column labeled “regret” preserves the ordinary-MoE post-hoc artifact’s reported oracle-regret statistic; it should not be numerically pooled with SA-MoE intervention magnitudes.
| Family | Method | Spearman | Top-source acc. | Regret |
|---|---|---|---|---|
| SmolLM2 | Random | |||
| Recency | ||||
| Largest component norm | ||||
| Li variance | ||||
| Li Top-2 boundary | ||||
| Gradient component | ||||
| Gemma | Random | |||
| Recency | ||||
| Largest component norm | ||||
| Li variance | ||||
| Li Top-2 boundary | ||||
| Gradient component |
These results confirm that cross-layer routing attribution does not require an explicit routing state. Li-style post-hoc attribution recovers meaningful source rankings in the Ordinary MoEs, while gradient-based sensitivity is again competitive or stronger on the intervention-ranking objective. The distinction claimed for SA-MoE is therefore the availability and semantics of its routing-specific forward audit object, not exclusive access to cross-layer routing information.
D.8 Audit Cost Under a Common Capture Policy
Audit cost depends on what is retained during the clean forward pass. We therefore do not equate “post-hoc” with mandatory model replay.
For SA-MoE, retaining every bf16 realized 128-dimensional write from a length-2048 forward pass requires 15.00 MiB for SmolLM2 and 9.00 MiB for Gemma. Retaining the clean Top-3 expert identities and logits in addition raises these routing-audit traces to approximately 15.70 MiB and 9.42 MiB, respectively. The live 128-dimensional state alone is 0.50 MiB for a length-2048 sequence.
If this write trace is retained, later ledger contribution queries require neither another model forward pass nor a target-dependent backward pass. Likewise, if the ordinary-MoE residual components required by the Li decomposition are retained during the clean forward, Li-style fixed-denominator scores can be formed without model replay or backward propagation. The frozen cost evidence does not provide an equivalently defined serialized retained-byte footprint for that component trace, so we do not report an asymmetric byte comparison.
Gradientwrite and gradientcomponent additionally require target-dependent gradient information unless a differentiable clean graph is itself retained, which has a different memory trade-off.
Table 13 gives phase-isolated pilot measurements on a single NVIDIA GH200 480GB GPU. Each pilot uses one frozen sequence, batch size 8, and four target decisions. Model loading, immutable-input validation, ranking aggregation, and causal-oracle interventions are excluded from the displayed extraction phases. These timings are diagnostic measurements rather than the scientific 96-target ranking experiment.
| Family | Audit object | Capture (s) | Capture peak (GiB) | Post-capture score (s) | Gradient extraction (s) |
|---|---|---|---|---|---|
| SmolLM2 | Persistent ledger | 0.249 | 3.96 | included | 3.205 |
| Ordinary Li components | 0.752 | 2.85 | 0.152 | 1.146 | |
| Gemma | Persistent ledger | 0.428 | 10.88 | included | 1.260 |
| Ordinary Li components | 0.852 | 10.70 | 0.080 | 1.319 |
The pilot environment uses Python 3.12.14, PyTorch 2.10.0+cu130, and Transformers 5.1.0. The Persistent pilot contains 74 eligible source interventions over four SmolLM2 targets and 56 over four Gemma targets; the ordinary-MoE component audit contains 222 and 168 source components, respectively. Gradient extraction uses four autograd calls for the four targets.
These measurements establish an operational distinction but not a universal runtime ordering. A retained SA-MoE write trace supports later algebraic ledger queries directly; a sufficiently rich retained ordinary-MoE component trace can likewise avoid replay for Li-style queries. Retention policy, component granularity, target count, and whether gradients are required all change the relevant cost.
D.9 Replicated Downstream Consequences
The main downstream results use the ledger-selected source for Persistent and the corresponding native latest-state-selected source for latest-write-only. Table 14 reports the three-seed means underlying the main-text intervention figure. Complete seed values are additionally collected in Appendix E.
| Family | Architecture | Target | Downstream Top-2 divergence | Output KL |
|---|---|---|---|---|
| SmolLM2 | Persistent | |||
| Latest-write-only | ||||
| P–LWO | ||||
| Gemma | Persistent | |||
| Latest-write-only | ||||
| P–LWO |
For the ledger-control advantage of Equation 47, Persistent has a three-seed mean of margin units on SmolLM2 (95% CI ) and on Gemma (). The corresponding latest-write-only values are () and (). The paired Persistent-minus-latest-write-only contrasts are () and (), respectively.
Thus, ledger-selected Persistent writes have replicated effects on the audited target margin, later routing, and output distribution. Persistence does not, however, produce a uniformly resolved increase over latest-write-only for every intervention outcome, and the fixed-checkpoint ranking analysis shows that gradient-based sensitivity can outperform exact forward provenance when the objective is specifically to identify the source with the largest full-forward perturbation.
Appendix E Complete Per-Seed Results and Analysis Scope
This appendix collects the training-seed values underlying the replicated results in the main text and Appendices B–D. Unless stated otherwise, training seeds , , and are the replication units. Cross-seed intervals reported elsewhere in the paper use Student- 95% intervals with two degrees of freedom. Within-checkpoint bootstrap samples are never treated as additional training replicates.
E.1 Model Quality and Downstream Capability
Table 15 reports the seed-level quality contrasts underlying Section 5.1 and Appendix B.3. PPL is shown as the ratio to the matched Ordinary MoE; task columns are absolute percentage-point changes in the primary task metric. Thus a PPL ratio above one indicates higher perplexity, while a positive task value indicates higher accuracy.
| Family | Architecture | Seed | PPL ratio | Hella | ARC-E | PIQA | Wino | Macro |
|---|---|---|---|---|---|---|---|---|
| SmolLM2 | Latest-write-only | 42 | 1.00361 | |||||
| 43 | 1.00557 | |||||||
| 44 | 1.00625 | |||||||
| Persistent | 42 | 1.00213 | ||||||
| 43 | 1.00187 | |||||||
| 44 | 1.00161 | |||||||
| Gemma | Latest-write-only | 42 | 1.00077 | |||||
| 43 | 1.00432 | |||||||
| 44 | 1.00399 | |||||||
| Persistent | 42 | 1.00096 | ||||||
| 43 | 0.99928 | |||||||
| 44 | 0.99787 |
Positive-ratio quantities such as PPL are analyzed across seeds on the natural-log scale before transforming the mean and confidence interval back to the ratio scale. Accuracy differences and other additive quantities use the ordinary arithmetic scale.
E.2 State Use and Persistence Endpoints
Table 16 gives the seed-level Persistent dose-response values underlying Figure 2a. The no-op endpoint is exactly zero by construction.
| Family | Seed 42 | Seed 43 | Seed 44 | |
|---|---|---|---|---|
| SmolLM2 | 0.75 | 11.89 | 12.34 | 11.37 |
| 0.50 | 28.88 | 30.22 | 27.10 | |
| 0.25 | 54.25 | 56.48 | 51.61 | |
| 0 | 88.16 | 88.61 | 86.17 | |
| Gemma | 0.75 | 17.88 | 17.03 | 15.94 |
| 0.50 | 34.67 | 35.76 | 33.91 | |
| 0.25 | 52.08 | 53.47 | 51.44 | |
| 0 | 69.12 | 70.93 | 69.67 |
At full suppression, the corresponding latest-write-only rates are for SmolLM2 and for Gemma.
Table 17 gives the exact seed-level contrasts used in Figure 2b. These are the frozen persistence-endpoint reductions. They should not be substituted for the separately aggregated Figure 4 quantities reported later in this appendix.
| Family | Endpoint | Seed 42 | Seed 43 | Seed 44 | Mean [95% CI] |
|---|---|---|---|---|---|
| SmolLM2 | Long-horizon accessibility (pp) | 21.97 | 19.33 | 16.99 | |
| Ledger–control advantage | 0.143 | 0.060 | |||
| Ledger intervention effect | 0.174 | 0.124 | 0.016 | ||
| Gemma | Long-horizon accessibility (pp) | 9.68 | 15.09 | 11.75 | |
| Ledger–control advantage | 0.046 | 0.234 | 0.040 | ||
| Ledger intervention effect | 0.059 | 0.257 | 0.086 |
The long-horizon contrast is positive in all three training seeds for both model families. The two intervention-derived persistence contrasts have positive three-seed means but are less uniform and have intervals spanning zero.
E.3 Future-Routing Probe Curves
Table 18 gives the complete seed-level numerical record for the three representations used in the replicated main future-routing comparison: the Persistent native state, latest-write-only native state, and supervised 128-dimensional subspace of the Persistent model’s ordinary hidden stream. Values are transition-only accuracy, equal-pair averaged across all six frozen source–target pairs as defined in Appendix B.8.
| Family | Budget | Representation | Seed 42 | Seed 43 | Seed 44 |
|---|---|---|---|---|---|
| SmolLM2 | 10k | Persistent native | 79.28 | 78.37 | 80.18 |
| Latest-write-only native | 57.90 | 56.24 | 56.48 | ||
| P hidden-128 | 76.56 | 77.93 | 78.81 | ||
| 100k | Persistent native | 81.18 | 79.88 | 82.12 | |
| Latest-write-only native | 59.50 | 57.76 | 58.40 | ||
| P hidden-128 | 81.52 | 82.20 | 82.92 | ||
| 1M | Persistent native | 81.45 | 79.96 | 82.33 | |
| Latest-write-only native | 59.68 | 57.96 | 58.56 | ||
| P hidden-128 | 82.94 | 83.18 | 83.98 | ||
| Gemma | 10k | Persistent native | 73.91 | 75.50 | 75.45 |
| Latest-write-only native | 63.47 | 64.05 | 66.76 | ||
| P hidden-128 | 75.73 | 79.61 | 79.22 | ||
| 100k | Persistent native | 76.06 | 78.24 | 77.16 | |
| Latest-write-only native | 65.81 | 65.68 | 68.82 | ||
| P hidden-128 | 81.24 | 83.19 | 83.68 | ||
| 1M | Persistent native | 76.48 | 78.33 | 77.41 | |
| Latest-write-only native | 66.23 | 65.77 | 68.96 | ||
| P hidden-128 | 82.70 | 84.69 | 84.91 |
E.4 Distributed Persistent History
Table 19 collects the seed-level Persistent direct-provenance summaries underlying Appendix C.2. Latest-write-only has non-recent share zero, effective support one, and largest-source share one by construction under the structural direct-state audit object of Appendix C.3.
| Family | Metric | Seed 42 | Seed 43 | Seed 44 | Mean [95% CI] |
|---|---|---|---|---|---|
| SmolLM2 | Non-recent absolute share (%) | 91.24 | 92.20 | 91.76 | |
| Effective source support | 9.444 | 9.544 | 9.482 | ||
| Largest absolute source share (%) | 20.57 | 20.10 | 20.47 | ||
| Signed direct margin | 0.478 | 0.343 | 0.403 | ||
| Gemma | Non-recent absolute share (%) | 90.75 | 89.84 | 90.28 | |
| Effective source support | 5.776 | 5.923 | 5.774 | ||
| Largest absolute source share (%) | 32.12 | 30.58 | 31.70 | ||
| Signed direct margin | 0.269 | 0.276 | 0.218 |
E.5 Ordinary-MoE Post-Hoc Attribution Across Seeds
For completeness, Table 20 gives the training-seed values underlying the principal Ordinary-MoE attribution comparison in Section 5.5 and Appendix D.7. These quantities concern ordinary-MoE residual components and are not pooled with SA-MoE write interventions.
| Family | Method | Metric | Seed 42 | Seed 43 | Seed 44 |
|---|---|---|---|---|---|
| SmolLM2 | Li Top-2 boundary | Spearman | 0.577 | 0.557 | 0.576 |
| Top-source accuracy | 0.208 | 0.198 | 0.229 | ||
| Regret | 0.575 | 0.612 | 0.555 | ||
| Gradient component | Spearman | 0.731 | 0.697 | 0.734 | |
| Top-source accuracy | 0.281 | 0.292 | 0.406 | ||
| Regret | 0.413 | 0.450 | 0.350 | ||
| Gemma | Li Top-2 boundary | Spearman | 0.600 | 0.638 | 0.623 |
| Top-source accuracy | 0.438 | 0.312 | 0.417 | ||
| Regret | 0.322 | 0.321 | 0.257 | ||
| Gradient component | Spearman | 0.774 | 0.791 | 0.780 | |
| Top-source accuracy | 0.479 | 0.500 | 0.583 | ||
| Regret | 0.221 | 0.204 | 0.197 |
E.6 Full-Forward Downstream Consequences
Table 21 gives the exact training-seed values underlying Figure 4 and Table 14. “P–LWO” denotes the within-seed Persistent-minus-latest-write-only contrast.
| Family | Architecture / contrast | Metric | Seed 42 | Seed 43 | Seed 44 |
|---|---|---|---|---|---|
| SmolLM2 | Persistent | Target | 0.3289 | 0.2344 | 0.2641 |
| Downstream Top-2 | 0.0808 | 0.0958 | 0.0638 | ||
| Output KL | 0.001386 | 0.001709 | 0.001556 | ||
| Latest-write-only | Target | 0.1365 | 0.1138 | 0.2891 | |
| Downstream Top-2 | 0.0257 | 0.0257 | 0.0400 | ||
| Output KL | 0.001442 | 0.001504 | 0.001832 | ||
| P–LWO | Target | 0.1924 | 0.1206 | ||
| Downstream Top-2 | 0.0551 | 0.0701 | 0.0238 | ||
| Output KL | 0.000206 | ||||
| Gemma | Persistent | Target | 0.3850 | 0.4488 | 0.4446 |
| Downstream Top-2 | 0.2222 | 0.1968 | 0.1667 | ||
| Output KL | 0.013094 | 0.021198 | 0.014161 | ||
| Latest-write-only | Target | 0.2982 | 0.2206 | 0.3181 | |
| Downstream Top-2 | 0.0457 | 0.0324 | 0.0689 | ||
| Output KL | 0.008626 | 0.004632 | 0.005024 | ||
| P–LWO | Target | 0.0868 | 0.2282 | 0.1266 | |
| Downstream Top-2 | 0.1765 | 0.1644 | 0.0978 | ||
| Output KL | 0.004468 | 0.016566 | 0.009137 |
The seed-level record makes the qualification in Section 5.5 explicit. Persistent ledger-selected suppression has nonzero mean downstream effects in both families, but Persistent-minus-latest-write-only differences are not uniformly positive for every outcome and training seed.
E.7 Scope of Statistical Claims
Table 22 summarizes the replication and uncertainty unit for each class of analysis. This distinction is important because several supplementary forensic analyses intentionally use one frozen checkpoint and should not be interpreted as three-seed architecture-level evidence.
| Analysis | Checkpoint scope | Uncertainty / replication unit | Interpretation |
|---|---|---|---|
| Model quality and capability | Seeds 42/43/44 for each family and architecture | Training seed; Student- (). PPL analyzed on log-ratio scale. | Replicated architecture-level result. |
| Local state-suppression dose response | Persistent and latest-write-only, seeds 42/43/44 | Training seed; Student- () | Replicated local router-dependence result. |
| Long-horizon persistence endpoint | Persistent and latest-write-only, seeds 42/43/44 | Paired training seed; Student- () | Primary replicated persistence result. |
| Main future-routing probe curves | Seeds 42/43/44 | Training seed after within-checkpoint fit/pair reduction; Student- () | Replicated representation-accessibility result. |
| Additional hidden-state probe controls | Supplementary frozen representation analyses | Within-checkpoint probe evaluation unless explicitly reported across seeds | Representation diagnostic; not used as an independent training-replication claim. |
| Distributed-history provenance | Persistent and latest-write-only, seeds 42/43/44 | Training seed; Student- () | Replicated descriptive direct-provenance result. |
| Figure 4 source suppression | Persistent and latest-write-only, seeds 42/43/44 | Training seed for cross-checkpoint summaries; paired sequence bootstrap within checkpoint | Replicated downstream intervention result. |
| SA-MoE exhaustive ranking diagnostic | Persistent seed 42 per family | 24 source sequences; 2,000 paired sequence-bootstrap resamples | Fixed-checkpoint forensic comparison; not a three-seed architecture claim. |
| Ordinary-MoE Li attribution | Ordinary MoE seeds 42/43/44 | Training seed for cross-checkpoint summaries; sequence bootstrap within checkpoint | Replicated post-hoc attribution comparison on a distinct audit object. |
| Ledger numerical reconstruction | Frozen construction checkpoints | Maximum deterministic numerical residual | Implementation/numerical validation, not a statistical claim. |
| State-geometry diagnostic | Older fixed SmolLM2 checkpoint where stated | Descriptive | Mechanism/control diagnostic only. |
| Audit-cost pilot | One frozen workload per family on common hardware | Descriptive repeated timing phases | Operational diagnostic; not a training-seed or universal speed claim. |
Across all replicated analyses, the paper reports the three individual training-seed values rather than treating the large number of tokens, probe examples, routing decisions, or bootstrap draws as independent model replicates. Same-sign effects across all three seeds are useful replication evidence, but do not constitute a separate high-powered significance test.