Grunbaum Entropy-Preserving Router
Implementation & benchmark of arXiv:2607.23269 — Entropic analogues of Grünbaum's inequality
Source paper: Entropic analogues of Grünbaum's inequality arXiv:2607.23269 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace arbitrary learned thresholds in a binary MoE or hierarchical latent router with a threshold at the batch mean of a learned scalar projection. Add a penalty when the entropy of either routed subgroup falls too far below the parent entropy, using the paper's sharp constant as the target. This discourages routing branches from becoming nearly deterministic or semantically impoverished while retaining a simple, cheap gating operation.
Formulas
Mathematical statement
For a log-concave random vector X with density f, a hyperplane H through its mean divides space into H^+ and H^- with probabilities satisfying 1/e <= P(X in H^+) and P(X in H^-) <= 1-1/e. In the one-dimensional formulation, if X is a log-concave scalar variable, h(X) is differential entropy, X|X<=E[X] is the conditional variable on the lower half-space, and H_2(p)=-p log p-(1-p)log(1-p) is binary entropy, the paper proves h(X)-C_G <= h(X|X<=E[X]) <= h(X), where C_G=(e/(e-1))H_2(1/e). The density is log-concave when f(x)=exp(-V(x))/Z for a convex function V and normalization Z. Apply this to a router projection z=w^T x with threshold mu=E[z]. For estimated parent entropy h_parent and child entropy h_child, define the entropy-deficit barrier L_def=max(0,h_parent-h_child-C_G)^2. The theorem predicts L_def should be zero for genuinely log-concave projected features, while violations identify routing collapse, non-log-concavity, or estimation error.
Implementation notes
Integrate this into a binary MoE router or a binary hierarchical latent partition. Given token features x in a batch, compute a scalar projection z=w^T x, use a detached batch mean mu=mean(z), and route tokens with z<=mu to expert 0 and the rest to expert 1. Keep the hard routing for the first MVP; a straight-through sigmoid around z-mu can be used if w must receive gradients. Every K steps, estimate the parent and child differential entropies from z using a fixed-bin histogram with a small additive pseudocount, or use the Gaussian proxy h_G=0.5 log(2 pi e sigma^2) as a low-variance first experiment. Compute C_G=e/(e-1)*[-(1/e)log(1/e)-(1-1/e)log(1-1/e)], then add lambda_def times the displayed entropy-deficit barrier. Also add a separate parent entropy floor, such as lambda_floor*max(0,h_min-h_parent)^2, because the deficit constraint alone permits both parent and child entropies to collapse together. Maintain an exponential moving average of mu for routing stability and skip the entropy term when either child has fewer than a minimum number of tokens. The mathematical quantities are the mean cut, the 1/e mass bound, and C_G; entropy estimators, smoothing, and regularization weights are empirical choices. First test on a 4-expert sparse MoE by recursively applying mean cuts, using a small Transformer on WikiText-2 or CIFAR-10 patches. Compare against standard top-1 routing at equal parameter count and FLOPs. Measure validation loss, dead-expert frequency, per-expert token counts, projected-feature entropy, and seed-to-seed training variance. Success means fewer dead experts and more stable training with equal or better validation loss and no meaningful throughput loss.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.