Time-Reversed Terminal-Gradient Optimizer
Implementation & benchmark of arXiv:2609.03281 — A Unified Theory of H-Duality in First-Order Methods
Source paper: A Unified Theory of H-Duality in First-Order Methods arXiv:2609.03281 ⓘ · analyzed Sep 4, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Construct a finite-horizon momentum schedule and its coefficient-reversed counterpart, then use the reversed schedule when the desired endpoint criterion is gradient norm rather than function-value decrease. Optimize the schedule on extremal-curvature quadratic proxies, where the dynamics reduce to an exactly simulable scalar recurrence; this creates a cheap, falsifiable optimizer-design procedure without changing the neural-network architecture.
Formulas
Mathematical statement
The paper's motivating function-gap guarantee is h(y_N)-h(y_star) <= tau_N^OGM times L/2 times the squared distance from y_0 to y_star, where h is an L-smooth convex function, y_star is its minimizer, y_0 is the initial point, y_N is the final iterate, and tau_N^OGM is the finite-horizon rate. The paper pairs such a criterion with a terminal residual criterion: tau/2 times the squared norm of (rho-mu)y_0+(1-rho)s_0 is at least 1/(2 tau) times the squared norm of (1+mu)y_N-2s_N, with rho=sqrt((1+mu^2)/2), states y_k and s_k, and problem parameter mu. In the operator setting, the right side becomes (1+mu)^2/(2 tau) times the squared norm of T(y_N), controlling a final oracle residual. The adaptation uses the scalar quadratic q_lambda(x)=lambda x^2/2, with curvature lambda in a proxy set Lambda contained in [mu,L], gradient g_k=lambda x_k, and recurrence x_{k+1}=x_k-alpha_k g_k+beta_k(x_k-x_{k-1}). The forward schedule is the coefficient sequence (alpha_k,beta_k); its time reversal uses alpha_k^rev=alpha_{N-1-k} and beta_k^rev=beta_{N-1-k}. The schedule objective is R(alpha,beta)=max over lambda in Lambda of absolute value of lambda x_N(lambda), which directly minimizes worst-case terminal gradient magnitude. The exact H-duality theorem applies to the paper's specified fixed-step convex or contractive settings; neural-network training is only an empirical transfer of its recurrence and schedule-reversal mechanism.
Implementation notes
Integrate this at the optimizer-update level, not inside the neural network. Maintain two optimizer variants: a conventional forward schedule and a time-reversed schedule with the same number of updates per training block. Parameterize a horizon-N momentum optimizer by positive learning rates alpha_k and bounded momentum coefficients beta_k, initialized from a cosine-decayed SGD schedule or standard momentum. Choose a proxy curvature set Lambda consisting of {mu, sqrt(mu L), L}, or use a logarithmic grid over [mu,L]. For every lambda, initialize x_{-1}=x_0=1 and run: g_k=lambda*x_k; x_next=x_k-alpha_k*g_k+beta_k*(x_k-x_{k-1}); then set x_{k-1}=x_k and x_k=x_next. Optimize R=max_lambda |lambda*x_N(lambda)| with an added penalty for intermediate amplification, such as gamma times max over k and lambda of |x_k(lambda)| squared. Reverse the learned coefficient arrays to create the second optimizer; do not reverse minibatch order or data augmentation. Apply the selected coefficients to every parameter tensor in each N-update block. Estimate mu and L either from a broad normalized proxy range or from minibatch gradient directional ratios; the proxy recurrence is exact, while the useful curvature range is empirical. Start with N=8 or 16 and compare SGD with momentum, cosine-decayed SGD, the learned forward schedule, and its reversed schedule on an MLP or ResNet-18 trained on MNIST or CIFAR-10. Use equal update counts and equal optimizer FLOPs. Record loss, global gradient norm at block endpoints, validation accuracy, peak gradient norm, and the fraction of unstable blocks. Success means lower terminal gradient norm and faster loss descent at equal compute, with no increase in gradient explosions; failure is no improvement over cosine SGD at matched update budget.
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.