Dense Mixture-of-Experts as a Reparameterized Wide FFN: A Granularity Sweep at Fixed Compute
Organizations: University of Information Technology · Vietnam National University, Ho Chi Minh City, Viet Nam
Abstract
Sparse Mixture-of-Experts (MoE) models combine learned routing with selection from a large expert pool. We isolate the contribution of dynamic expert combination using a dense analogue: SwiGLU experts, all active for every token and combined by a softmax gate, at fixed total FFN width. With no larger pool or discrete selection, the dense baseline is the case. Validation loss varies non-monotonically with : improves over the baseline by , whereas and worsen it by and , respectively. Routing generally remains soft and expert usage balanced, except in the first layer of the model, where routing is nearly one-hot. Forcing this gate to uniform increases loss by nats on a diagnostic subset, indicating that its concentrated routing is functionally important. We further show that the architecture is exactly a dense SwiGLU with token-dependent, simplex-constrained group scaling and contains the dense baseline in its function class. These single-run results suggest a granularity sweet spot for always-active, softly gated FFNs at fixed width, with performing best among the configurations tested.
Figures & tables
| Router params | Non-embedding params | Total params (tied embed./head) | ||
|---|---|---|---|---|
| 1 (dense) | 0 | 28,321,152 | 47,619,840 | |
| 2 | 768 | 28,330,368 | 47,629,056 | |
| 4 | 1,536 | 28,339,584 | 47,638,272 | |
| 6 | 2,304 | 28,348,800 | 47,647,488 |
| FFN block at initialization | Output std | Variance / baseline |
|---|---|---|
| Dense SwiGLU ( ) | 0.1056 | 1.00 |
| Experts, scaled init | 0.0111 | 0.011 |
| Experts, fan-in init | 0.0610 | 0.334 |
| Configuration | Val. loss (step 20,000) | Val vs. dense | |
|---|---|---|---|
| Dense SwiGLU ( ) | 3.8752 | — | |
| 3.8704 | |||
| 3.8805 | |||
| 3.8949 |
| Per-token entropy | Mean max weight (uniform) | Frac. | Usage entropy | |
|---|---|---|---|---|
| 2 | 0.719 | 0.762 (0.500) | 0.189 | 0.948 |
| 4 | 0.668 | 0.622 (0.250) | 0.097 | 0.901 |
| 6 | 0.713 | 0.517 (0.167) | 0.015 | 0.941 |
| Layer | Mean pairwise cosine sim. | Max pairwise cosine sim. |
|---|---|---|
| 0 | 0.414 | 0.650 |
| 1 | 0.161 | 0.249 |
| 2 | 0.085 | 0.187 |
| 3 | 0.049 | 0.156 |
| 4 | 0.044 | 0.143 |
| 5 | 0.040 | 0.125 |
| Configuration | Val. loss | Val. ppl. |
|---|---|---|
| Normal routing (trained gates) | 3.7493 | 42.49 |
| Layer 0 forced uniform | 6.1216 | 455.59 |
Mixture-of-Experts Models
Mixture-of-experts models use input-dependent routing to activate specialized subnetworks, increasing model capacity without running every parameter for each input.
See all 462 papers