Unverified 2026

Information-budgeted Gibbs router

Implementation & benchmark of arXiv:2608.18061 — The concentration game: Bayesian updating, regret, and information

Usefulness6/10
Difficulty5/10
Novelty4/10

Source paper: The concentration game: Bayesian updating, regret, and information arXiv:2608.18061 ⓘ · analyzed Sep 1, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Replace a fixed-temperature softmax router over experts, adapters, or candidate optimizers with an exponential-weights distribution whose temperature is selected to satisfy an explicit cumulative information budget. The router reacts strongly when observed expert losses are predictable, but automatically cools down when outcomes create a large cumulant-information gap, avoiding variance-based heuristics that can be badly miscalibrated. A prior distribution over experts supplies a principled trust-region bias through relative entropy.

Formulas

$$L_{\Gamma}(S):=\sup\{\langle\nu,S\rangle:\nu\in\Delta([K]),\ \mathrm{KL}(\nu\|\pi)\le\Gamma\}$$
$$p_t(i)=\frac{\pi(i)e^{-\eta_t C_{t-1}(i)}}{\sum_{j=1}^{K}\pi(j)e^{-\eta_t C_{t-1}(j)}},\qquad C_t(i):=\sum_{s=1}^{t}c_s(i),\qquad C_0\equiv0$$
$$I_t(\eta):=\log\!\left(\sum_{i=1}^{K}p_t(i)e^{-\eta c_t(i)}\right)+\eta\sum_{i=1}^{K}p_t(i)c_t(i)\ge0$$
$$\Phi_t(\eta):=-\frac{1}{\eta}\log Z_t(\eta),\qquad Z_t(\eta):=\sum_{i=1}^{K}\pi(i)e^{-\eta C_t(i)},\qquad \sum_{s=1}^{t}I_s(\eta_t)\le B$$

Mathematical statement

The paper defines a strictly positive prior \(\pi\in\Delta([K])\), cumulative scores \(C_t(i)=\sum_{s=1}^t c_s(i)\), and the Gibbs/Bayes action \(p_t(i)=\pi(i)e^{-\eta_t C_{t-1}(i)}/Z_{t-1}(\eta_t)\), where \(c_t(i)\) is expert \(i\)'s loss, \(\eta_t>0\) is inverse temperature, and \(Z_t(\eta)=\sum_{j=1}^K\pi(j)e^{-\eta C_t(j)}\) is the log-partition function. The terminal robust comparator value is \(L_\Gamma(S)=\sup\{\langle\nu,S\rangle:\nu\in\Delta([K]),\mathrm{KL}(\nu\|\pi)\le\Gamma\}\), with comparator distribution \(\nu\) and KL radius \(\Gamma\). For an observed loss vector \(c_t\), define the exact one-step cumulant-information gap \(I_t(\eta)=\log\sum_i p_t(i)e^{-\eta c_t(i)}+\eta\sum_i p_t(i)c_t(i)\ge0\). This is the Jensen gap between the log moment-generating function and expected loss; it is zero when all active experts have equal loss and grows when losses vary across experts. Choose \(\eta_t\) as the largest candidate inverse temperature satisfying \(\sum_{s\le t}I_s(\eta_t)\le B\), where \(B\) is a tunable information budget. When \(\eta\) changes, use the exact retempered potential \(\Phi_t(\eta)=-(1/\eta)\log Z_t(\eta)\) rather than assuming the old potential is unchanged.

Implementation notes

Integrate this at the router or portfolio layer, not inside the expert networks. Let K be the number of MoE experts, LoRA adapters, augmentations, or candidate optimizer updates. At step t, obtain the per-example or minibatch loss vector c[0:K] by evaluating each candidate, or use detached proxy losses when full evaluation is too expensive. Maintain cumulative scores C[i] and a positive prior pi[i]. Use this loop: (1) choose eta from a logarithmic grid, or by bisection over eta if losses are clipped to [0,1], selecting the largest eta whose running information estimate is at most budget B; (2) compute logits[i]=log(pi[i])-eta*C[i] and p=softmax(logits); (3) execute the weighted output y=sum_i p[i] y_i; (4) after observing the candidate losses, compute q=sum_i p[i] exp(-eta*c[i]) and I=log(q)+eta*dot(p,c); (5) update C[i]+=c[i] and the running information accumulator. Use logsumexp for stable computation and clip or normalize losses before exponentiation. If eta changes, recompute p from current C rather than multiplying old probabilities; also record the retempering potential Phi=-(1/eta)*logsumexp(log(pi)-eta*C), which is the exact scale-dependent state variable. The mathematics determines the Gibbs update, nonnegative information gap, and KL-prior interpretation; B, the eta grid, and clipping range are empirical hyperparameters. First test on CIFAR-100 with a small four-expert ResNet or four LoRA adapters, comparing fixed-temperature softmax, standard fixed-eta Hedge, and the proposed method at equal expert FLOPs. Measure validation accuracy, router entropy, cumulative regret to the best expert, and loss spikes after synthetic distribution shifts. Success means lower best-expert regret, fewer routing collapses, or faster recovery after a corruption shift at equal compute.

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.