Samuels Chance-Budget Regularizer
Implementation & benchmark of arXiv:2608.18392 — On Samuels' Conjecture
Source paper: On Samuels' Conjecture arXiv:2608.18392 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Use Samuels' exact lower bound as a differentiable certificate for the probability that a random neural-network cost remains below a hard budget, under independent nonnegative component costs and known means. This can regularize stochastic MoE loads, activation memory, dynamic depth, or per-example loss decompositions without assuming variances or bounded support.
Formulas
Mathematical statement
The paper considers independent nonnegative random variables X_1,...,X_n with ordered means 0 <= mu_1 <= ... <= mu_n, and a threshold lambda > sum_i mu_i. It defines delta = lambda - sum_i mu_i > 0 and D_i = lambda - sum_{k=1}^{i-1} mu_k = delta + sum_{k=i}^n mu_k. Samuels' theorem states that the exact worst-case probability over all independent laws satisfying E[X_i] = mu_i is c(mu,lambda) = inf P(sum_i X_i < lambda) = min_{1 <= i <= n} q_i(mu,lambda), where q_i(mu,lambda) = product_{j=i}^n (1 - mu_j / D_i). In the neural adaptation, X_i is a stochastic nonnegative cost or error contribution, mu_i is its estimated mean, and lambda is the desired budget. The ordering assumption is enforced by sorting the current mean vector before evaluating the expression. To make the nonsmooth minimum trainable, replace it during optimization by softmin_tau(q) = -tau log(sum_i exp(-q_i/tau)); the exact hard minimum is used for evaluation.
Implementation notes
Integrate this into stochastic compute control, preferably a small MoE or dynamic-depth Transformer. Define X_i for one example as the nonnegative number of active tokens routed to expert i, activation bytes used by layer i, or FLOPs incurred by optional block i. For a simple MVP, use stochastic depth: X_i = c_i B_i, where c_i is the measured cost of block i and B_i is its Bernoulli execution indicator. On every minibatch, measure x_i for each component and update a running mean with mu_i <- beta mu_i + (1-beta) mean_batch(x_i). Sort the means into ascending order and retain the sorting permutation. Compute D_i = lambda - cumulative_sum(mu_sorted before i), then q_i = product over j from i to n of (1 - mu_sorted[j] / D_i). If lambda <= sum_i mu_i, increase lambda or add a separate overload penalty; never apply the theorem outside its stated condition. Add alpha times L_budget to the task loss, using the softmin formula during training. Pseudocode: collect costs x; mu=EMA(mu,mean(x)); m,perm=sort(mu); D[i]=lambda-sum(m[:i]); q[i]=prod(1-m[i:]/D[i]); cert=softmin(q,tau); loss=task_loss-alpha*log(cert); backpropagate and update parameters. The theorem supplies the exact mean-only certificate; independence and the quality of the EMA estimates must be checked empirically. First experiment: train a 4-layer stochastic-depth MLP or tiny Transformer on CIFAR-10 or WikiText-2 at a fixed expected compute budget. Compare no regularizer, a mean-cost penalty, and this regularizer. Record empirical P(total_cost < lambda), worst-batch overload probability, accuracy or perplexity, and training variance. Success means higher budget satisfaction at matched accuracy and a useful certificate that remains conservative under distribution shift.
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.