Entropy-Aware Routing (EAR)
- Entropy-Aware Routing (EAR) is a dynamic uncertainty-aware approach that quantifies input ambiguity using entropy to guide routing decisions.
- It enables adaptive expert selection in systems like MoE by regulating the number and type of models activated based on input complexity.
- EAR principles extend beyond traditional MoE layers, influencing memory systems, recommendation engines, and speech recognition through efficient resource allocation.
Entropy-Aware Routing (EAR) denotes a class of routing mechanisms in which uncertainty, typically quantified by entropy or by an explicitly stated proxy for entropy, governs how computation, memory access, or expert activation is allocated. Across recent work, EAR appears in several distinct forms: token-wise expert-count selection in Mixture-of-Experts (MoE) large vision-LLMs, difficulty-biased routing in heterogeneous xLSTM MoE LLMs, entropy-regularized model selection among experts of different capability and cost, entropy-triggered specialization in multimodal recommendation, uncertainty-aware fallback in hierarchical memory systems, and entropy-weighted blending with shared experts in multi-domain speech recognition (Cai et al., 24 Jun 2026, Thiombiano et al., 1 May 2025, Zhang et al., 28 Jun 2026, Dai et al., 24 Feb 2026, Singh, 27 Feb 2026, Shi et al., 9 Jun 2026). The common premise is that routing should not remain static when the input-dependent uncertainty of the router varies substantially.
1. Conceptual scope and formal definitions
In the broadest sense, EAR treats routing as an uncertainty-sensitive decision problem. A router receives an input and produces either a distribution over experts, models, or memory candidates, or a decision rule derived from that distribution. The entropy of that distribution is then used to decide whether to broaden computation, sharpen selection, invoke fallback mechanisms, or mix in shared capacity.
The most explicit formulation appears in GeMoE, where the router for an MoE layer produces logits over experts, followed by a softmax routing distribution
If , the token-level gating entropy is
with log base 2 used in that paper’s experiments (Cai et al., 24 Jun 2026). Low entropy corresponds to a peaked routing distribution and is interpreted as an “easy” or unambiguous token; high entropy corresponds to a flatter distribution and is interpreted as a “hard” or ambiguous token.
A related but distinct construction appears in MoxE, where entropy is not used to choose the number of active experts but to train a difficulty predictor aligned with the normalized entropy of the unbiased router distribution:
That difficulty estimate then biases routing toward mLSTM experts for hard or rare tokens (Thiombiano et al., 1 May 2025).
EntroRouter generalizes EAR beyond MoE layers to model selection. Given a router policy over experts , it introduces an entropy-aware objective
where entropy and KL terms explicitly regulate exploration, exploitation, and trust-region behavior (Zhang et al., 28 Jun 2026).
Not all systems labeled EAR compute Shannon entropy directly. HTM-EAR operationalizes uncertainty through a two-signal gate: the top-1 similarity score from L1 memory and entity coverage of the top candidate. The paper also gives an optional entropy formalization over normalized 0-NN similarities, but states that the reported experiments use the simpler similarity-and-coverage gate (Singh, 27 Feb 2026). This distinction is important: EAR refers to uncertainty-aware routing more generally, but explicit entropy computation is only one implementation.
2. Theoretical rationales for entropy-aware routing
Several papers supply distinct formal rationales for why entropy should guide routing. In GeMoE, dynamic routing is framed as a Minimum Description Length (MDL) problem. For token 1 and selected subset of experts 2, the total description length is
3
with
4
and
5
Adding an expert 6 is justified when
7
Because evaluating candidate experts directly is expensive, the paper argues that gating entropy is a practical proxy for the information gain term: when entropy is high, additional experts tend to have non-negligible router weights and are more likely to produce gain above the per-expert complexity penalty (Cai et al., 24 Jun 2026).
MoxE provides a different mechanism. There, the router learns a difficulty score
8
and applies an additive routing bias
9
Under the approximation of balanced unbiased logits across groups, the probability ratio obeys
0
so routing to mLSTM grows exponentially with difficulty (Thiombiano et al., 1 May 2025). Here entropy serves as a supervisory target for difficulty estimation rather than a direct routing statistic at inference.
EntroRouter’s theoretical contribution is tied to a failure mode it terms Trust Region Collapse (TRC). In multi-round routing, if escalation to stronger experts has positive cost and sparse supervision causes out-of-distribution expert traces to be ignored, the advantage of escalation becomes strictly negative, and the effective trust region for increasing escalation probability becomes empty. Entropy regulation and KL anchoring are then used to preserve exploration initially and contract entropy only within a safe region later in training (Zhang et al., 28 Jun 2026). In this formulation, EAR is not merely an efficiency heuristic; it is a stabilizing principle for policy optimization under sparse rewards.
MAGNET uses entropy in yet another way: not as an input-level difficulty score but as a batch-level monitor of routing diversity. Let 1 denote the batch-mean dense routing distribution; then
2
This normalized entropy drives a two-stage controller that delays specialization until routing diversity has stabilized (Dai et al., 24 Feb 2026).
3. EAR in MoE architectures and model routing
In MoE systems, EAR primarily modulates either the number of active experts or the type of experts chosen. GeMoE introduces an Expert Assignment Predictor (EAP), a lightweight linear layer that predicts the number of experts to activate per token. For an allowed set 3, it computes
4
followed by
5
with a straight-through estimator during training. A monotonicity loss 6 enforces that 7 increases with gating entropy, and the full training objective is
8
Only the router and EAP are trained; experts remain native and are not fine-tuned (Cai et al., 24 Jun 2026).
Empirically, GeMoE is evaluated on MolmoE-1B-7B (64Top8) and DeepSeek-VL2-Tiny-1B-3B (64Top6), against DYNMoE, Top-p, AdaMoE, and MoE++. On MolmoE-1B-7B, static Top-8 gives Avg_P 9, Avg_k 0, and 1B, whereas GeMoE gives Avg_P 2, Avg_k 3, and 4B, corresponding to performance retention of about 5–6 and 7 fewer experts than Top-8. Across its reported settings, the paper states 8 average performance retention with 9 average expert activation sparsity improvement (Cai et al., 24 Jun 2026).
MoxE also belongs to the EAR family, but its mechanism is heterogeneous-expert selection rather than expert-count adaptation. It combines mLSTM and sLSTM experts in a sparse MoE with top-0 routing. High-entropy tokens are preferentially routed to mLSTM experts, which the paper associates with stronger recall via matrix memory, while low-entropy tokens are balanced across both expert types (Thiombiano et al., 1 May 2025). The total objective combines task loss with a difficulty loss 1, group-wise balancing loss 2, router Z-loss 3, and a load-balancing auxiliary loss:
4
A broader model-routing instantiation appears in EntroRouter. Its Stage I Soft Supervision fits the router to a high-entropy target
5
where 6 is the most economical qualified expert and 7 are the 8 cheapest qualified experts. Stage II then performs reinforcement learning with a soft anchor
9
stabilizing training and controlling entropy contraction (Zhang et al., 28 Jun 2026).
The following comparison summarizes these variants.
| System | Routing signal | Main routing action |
|---|---|---|
| GeMoE | Token-level gating entropy 0 | Adaptively choose expert count 1 |
| MoxE | Difficulty aligned to normalized entropy 2 | Bias hard tokens toward mLSTM experts |
| EntroRouter | Policy entropy with KL anchoring | Regulate exploration and model choice |
| MAGNET | Batch routing entropy 3 | Trigger coverage-to-confidence transition |
| HTM-EAR | Similarity and entity-coverage gate; optional entropy formulation | Decide L1-only vs. L1+L2 fallback |
| Speech-LLM EAR | Normalized entropy over fine-domain probabilities | Blend domain-specific and shared experts |
4. EAR beyond standard MoE layers
EAR has also been applied to memory systems, recommendation, and speech recognition, where the routed objects are not merely feed-forward experts.
HTM-EAR introduces a hierarchical tiered memory substrate with HNSW-based working memory (L1, capacity 4) and archival storage (L2, capacity 5). Queries first search L1 with 6. Let 7 be the top-1 similarity and 8 the top candidate. If either 9 with 0 or entity coverage fails, retrieval also searches L2 with 1, merges candidates, scores them by
2
with 3 and 4, and re-ranks the top 20 with an MS MARCO cross-encoder (Singh, 27 Feb 2026). The paper explicitly notes that although the title suggests entropy awareness, the implemented EAR uses similarity and entity-coverage proxies rather than explicit entropy. Under sustained saturation, the full model preserves active-query precision at MRR 5 and approaches oracle active performance 6; on BGL logs it achieves MRR 7 versus oracle 8, while LRU falls to 9 (Singh, 27 Feb 2026).
In MAGNET for multimodal recommendation, EAR appears as “progressive Entropy-Triggered Routing.” The router produces dense expert probabilities
0
over a default expert pool of 1 experts organized as 2 modality anchors times 3 expert families. Training begins in Stage 1 with a coverage regularizer
4
weighted by
5
and switches to Stage 2, where a confidence regularizer
6
is weighted by
7
The switch is triggered when normalized batch entropy exceeds a threshold 8 for 9 consecutive steps (Dai et al., 24 Feb 2026). This formulation makes EAR a training-time controller for sparse routing stability and specialization.
In the Speech-LLM case study on unified child-adult ASR, EAR is applied to a classifier-based domain router with a coarse-to-fine structure. The coarse classifier outputs 0, and the fine classifier over OGI-S age groups outputs 1. Uncertainty is measured by normalized entropy
2
computed over the OGI-S fine-grained age-group probabilities with 3. EAR then blends the routed output with a shared expert:
4
The same blending applies to both Mixture-of-Projectors and Mixture-of-LoRAs, while the encoder and LLM backbone remain frozen (Shi et al., 9 Jun 2026). Reported gains are concentrated on OGI-S: under soft routing, OGI-S average WER improves from 5 to 6, while MyST remains 7 and Libri-Clean remains 8 (Shi et al., 9 Jun 2026).
5. Empirical regularities and efficiency trade-offs
Across domains, EAR is consistently presented as a way to improve the trade-off between performance and resource usage rather than as a mechanism for maximizing raw accuracy irrespective of cost.
In GeMoE, reducing the average number of active experts from 9 to 0 on MolmoE reduces inference FLOPs/token from 1 to 2 GFLOPs, increases throughput from 3 to 4 tokens/s, lowers wall-clock time per sample from 5s to 6s, and leaves memory usage unchanged at 7 GB on A800-40G (Cai et al., 24 Jun 2026). On DeepSeek 64Top6, GeMoE achieves Avg_P 8 versus Top-6 9 while using Avg_k 00, a 01 reduction in experts (Cai et al., 24 Jun 2026). Notably, in generalization to OLMoE-1B-7B, GeMoE attains Avg_P 02 with Avg_k 03, exceeding Top-8 Avg_P 04, and improves HumanEval from 05 to 06? No—the reported direction is GeMoE 07 versus Top-8 08, indicating higher score under entropy-aware sparsity (Cai et al., 24 Jun 2026).
MoxE reports strong ablation sensitivity to the entropy-aware components. On Lambada perplexity, the full MoxE baseline gives 09, whereas removing entropy bias (10) yields 11, removing group-wise loss yields 12, using mLSTM-only experts gives 13, and using sLSTM-only experts gives 14 (Thiombiano et al., 1 May 2025). The paper therefore positions entropy-aware bias and balancing as central to stability and utilization.
EntroRouter reports a cost-accuracy result at the system level rather than within a single neural layer: it retains approximately 15 of the strongest expert’s accuracy while reducing computational cost by approximately 16 on math reasoning benchmarks (Zhang et al., 28 Jun 2026). Its ablations indicate that removing Soft Supervision causes a marked accuracy drop, removing the gate 17 increases cost, removing the Soft Anchor degrades the trade-off, and removing shaped rewards worsens stability and efficiency (Zhang et al., 28 Jun 2026).
MAGNET reports that removing routing regularizers (18) consistently reduces Recall@20 and NDCG@20 across Baby, Sports, Clothing, and Electronics. Coverage-only and confidence-only variants underperform the full two-stage regime, and a fixed-step switch lags the entropy-triggered switch (Dai et al., 24 Feb 2026). The reported post-switch behavior is also structurally consistent: per-sample routing entropy decreases, top-1–top-2 probability gaps increase, and MaxLoad share remains controlled.
HTM-EAR and the speech-domain EAR show the same trade-off pattern in non-MoE settings. In HTM-EAR, the full system is slower than LRU but prevents uncontrolled forgetting: LRU has latency 19 ms but permanently evicts 20 essential facts, whereas the full system has latency 21 ms and essential loss 22 (Singh, 27 Feb 2026). In Speech-LLM EAR, the added cost is small because entropy is computed over a 3-class vector and used only for a convex combination, while performance gains are concentrated on uncertain child-speech subdomains (Shi et al., 9 Jun 2026).
6. Limitations, misconceptions, and open directions
A recurrent limitation is calibration. If router probabilities are poorly calibrated, entropy may fail to represent true uncertainty. GeMoE identifies entropy miscalibration as a direct risk, noting that poor calibration may under- or over-allocate experts; its mitigation is joint router training with load balancing and a monotonicity loss using a dynamic margin 23 the entropy gap (Cai et al., 24 Jun 2026). The speech-domain EAR likewise depends on the quality of fine-grained classifier probabilities; if uncertainty is inflated, routing may overuse the shared expert (Shi et al., 9 Jun 2026).
A second misconception is that EAR always requires explicit Shannon entropy. HTM-EAR is the clearest counterexample: the reported implementation uses a similarity threshold and entity coverage gate, while an entropy gate over neighbors is presented only as a conceptual extension (Singh, 27 Feb 2026). A plausible implication is that EAR is best understood as a design principle centered on uncertainty-aware routing, not as a single formula.
A third misconception is that more experts should always improve quality. GeMoE explicitly argues against this, and its OLMoE experiments are presented as evidence that entropy-aware sparsity can outperform denser activation (Cai et al., 24 Jun 2026). EntroRouter makes the analogous point at the model-selection level: unrestricted exploration without staged entropy control can collapse to poor local optima, while controlled entropy contraction is needed for stable, decisive routing (Zhang et al., 28 Jun 2026).
Several open directions are stated explicitly across the papers. EntroRouter suggests adaptive anchor scheduling, periodic re-profiling of offline capability estimates, hierarchical and multi-level routers, and theory for trust-region safety and convergence under entropy schedules (Zhang et al., 28 Jun 2026). MAGNET proposes learnable or adaptive expert templates and more robust handling of missing or noisy modalities (Dai et al., 24 Feb 2026). HTM-EAR notes that fixed thresholds and heuristic parameters may misroute under domain shift, and that explicit entropy or calibration could better capture multimodal neighbor sets (Singh, 27 Feb 2026). The speech-domain work points to extensions beyond clear hierarchies, where multiple ambiguous axes may require additional shared experts or richer routing structures (Shi et al., 9 Jun 2026).
In aggregate, EAR has become a unifying pattern across contemporary routing problems: use entropy, or a stated proxy for uncertainty, to allocate computation, memory access, or specialization adaptively rather than uniformly. The specific object being routed varies by domain, but the technical motif is stable: uncertainty should alter the routing policy itself.