Rolling-Horizon Port-Hamiltonian Optimizer
Implementation & benchmark of arXiv:2608.17222 — Feedback approaches for set-point stabilization of interacting particle systems
Source paper: Feedback approaches for set-point stabilization of interacting particle systems arXiv:2608.17222 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace the usual first-order parameter update with controlled position-velocity dynamics. The loss is the potential energy, momentum is the velocity, and a one-step rolling-horizon control minimizes the predicted next-step energy plus a control penalty, producing an explicitly dissipative correction that can be applied only through a low-rank or blockwise control operator.
Formulas
Mathematical statement
The paper's Cucker-Smale template is the second-order controlled system \(\dot X_t=V_t\), \(\dot V_t=-\Psi(X_t)V_t-\nabla\mathbb V(X_t)+\mathbb B u(t)\), where \(X_t\) is position, \(V_t\) is velocity, \(\Psi(X_t)\) is a positive damping operator, \(\mathbb V\) is a potential, \(\mathbb B\) maps the control into state space, and \(u\) is the feedback. Transfer this to parameters \(\theta\) with momentum \(v\), potential \(\mathbb V(\theta)=L(\theta)\), damping \(\Psi=\Gamma\succeq0\), and control matrix \(B\). Use the mechanical Hamiltonian \(H(\theta,v)=L(\theta)+\tfrac12\|v\|^2\). For the Euler prediction with step size \(h\), \(\hat v(u)=v+h[-\Gamma v-g+B u]\), \(\hat\theta=\theta+h\hat v(u)\), where \(g=\nabla_\theta L(\theta)\). Linearizing the potential at \(\theta\), the instantaneous rolling-horizon objective is \(J(u)=\tfrac12\|\hat v(u)\|^2+h g^T\hat v(u)+\tfrac\rho2\|u\|^2\). Its exact minimizer is \(u^*=-(hB^TB+\rho h^{-1}I)^{-1}B^T[(1-h\Gamma)v-hg]\). The continuous-time energy identity motivating this choice is \(\dot H=-v^T\Gamma v+v^TBu\); with \(u=-kB^Tv\), the controlled channel contributes \(-k\|B^Tv\|^2\).
Implementation notes
1. Integration point: modify the optimizer update for neural-network parameters, not the forward pass. Flatten each parameter block into \(\theta\), maintain a same-shaped velocity \(v\), and choose \(B\) as either the identity, a per-layer scalar gate, or a low-rank random projector. Compute the ordinary minibatch gradient \(g\) before applying the update. 2. Pseudocode: g = grad(loss, theta); a = (1-h*Gamma)*v - h*g; solve (h*B.T@B + rho/h*I) u = -B.T@a; v = a + h*B@u; theta = theta + h*v. For diagonal or block-diagonal \(B\), solve elementwise; for a rank-r projector use Woodbury, so the control overhead is \(O(r)\) per block. Clip \(h\Gamma\) to below one and use \(\rho>0\) for numerical regularization. 3. Mathematics versus estimates: the rolling-horizon minimizer and energy identity are analytic; \(g\) is the minibatch gradient, while \(\Gamma\) can be fixed initially and later estimated from the ratio of successive velocity norms. The loss linearization is the only approximation, so optionally recompute the candidate loss for two controls and backtrack if actual loss rises. 4. First experiment: train a 3-layer MLP and a small ResNet on MNIST or CIFAR-10 using identical batch sizes and FLOP budgets, comparing SGD with momentum, AdamW, and this optimizer for matched parameter-update counts. Test identity \(B\), layerwise diagonal \(B\), and rank-8 \(B\). Measure training loss at equal FLOPs, gradient/velocity norm spikes, final validation accuracy, and loss increases after updates. Success means faster loss descent or fewer instability spikes than momentum at equal compute; failure is no improvement after tuning \(h,\rho,\Gamma\) on a held-out seed set.
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.