Entropy-Geometry Anti-Acceleration Optimizer
Implementation & benchmark of arXiv:2607.27476 — Entropy-Smooth Convex Optimization Cannot Be Accelerated
Source paper: Entropy-Smooth Convex Optimization Cannot Be Accelerated arXiv:2607.27476 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace Euclidean momentum updates on simplex-valued neural variables with entropy mirror descent when the variable dimension is large and the loss is naturally smooth relative to negative entropy. The lower bound predicts that Nesterov-style acceleration cannot guarantee an asymptotic improvement in this geometry, while the geometry-matched update preserves positivity and can reduce boundary instability.
Formulas
Mathematical statement
The feasible set is the probability simplex Delta_d = {x in R^d : x_k >= 0 and sum_k x_k = 1}. The prox function is negative Shannon entropy h(x) = sum_k x_k log(x_k), with 0 log 0 = 0. Its Bregman divergence is D_h(y,x) = h(y) - h(x) - <grad h(x), y-x> = sum_k y_k log(y_k/x_k), which is KL divergence for strictly positive probability vectors. Relative L-smoothness means f(y) <= f(x) + <grad f(x), y-x> + L D_h(y,x). The paper proves a worst-case first-order lower bound of Omega(L/T) for this class when d = Omega(T^2), so generic acceleration cannot guarantee the usual O(1/T^2) rate. Minimizing the linearized loss plus (1/eta)D_h(y,x_t), with g_t = grad f(x_t), produces exponentiated gradient: x_{t+1,k} = x_{t,k} exp(-eta g_{t,k}) / sum_j x_{t,j} exp(-eta g_{t,j}). Here x_t is the current simplex-valued neural variable, g_t is its loss gradient, eta is the step size, and L is the relative-smoothness constant.
Implementation notes
(1) Integrate the method into the optimizer update for a tensor whose final axis is a probability simplex: an MoE router probability vector, attention weights, learned mixture weights, or a categorical policy head. Do not apply it directly to ordinary unconstrained weights. Keep probabilities in float32, compute the loss gradient g_t with respect to the probability tensor x_t, and update each simplex independently. (2) Pseudocode: initialize x = softmax(z); at each minibatch compute loss and g = d loss / d x; compute u_k = log(x_k + epsilon) - eta*g_k; set x_new = softmax(u) over the simplex axis; optionally store z = u. The common additive shift in u is irrelevant. Use eta = 1/L when a bound is available. Otherwise start with eta = 0.1 and perform backtracking: evaluate the proposed step and increase L, equivalently decrease eta, until loss(x_new) <= loss(x) + sum_k g_k*(x_new_k-x_k) + L*sum_k x_new_k*log((x_new_k+epsilon)/(x_k+epsilon)). (3) The mathematical contribution is the entropy Bregman geometry and the Omega(L/T) anti-acceleration lower bound. Estimate L empirically with minibatch backtracking; estimate D_h using the displayed KL expression and epsilon clamping. (4) First experiment: train a 64- or 256-expert Switch-style router on CIFAR-100 or a small language-model MoE. Compare AdamW on router logits, Nesterov-accelerated mirror descent, and entropy mirror descent with equal minibatches, gradient evaluations, and FLOPs. Measure validation loss, loss versus optimizer steps, routing entropy, fraction of dead experts, maximum gradient norm, and expert-load coefficient of variation. The falsifiable prediction is that acceleration will not improve loss scaling at large d and may be less stable, while entropy mirror descent will match or beat it in validation loss and produce fewer dead experts 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.