Energy-Adaptive Inertial Optimizer
Implementation & benchmark of arXiv:2608.02842 — Dynamics of Fractional Wave Equations with Nonlocal Damping
Source paper: Dynamics of Fractional Wave Equations with Nonlocal Damping arXiv:2608.02842 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace constant friction in a second-order neural-network optimizer by a scalar damping coefficient that grows as a power of the current parameter energy plus velocity energy. This should selectively damp large oscillations and unstable excursions while preserving lower friction during small, potentially useful movements.
Formulas
Mathematical statement
The paper studies the equation u_tt + A u + gamma ([u]_s^2 + ||u_t||^2)^q u_t + f(u) = h, where A = (-Delta)^s, gamma > 0, and q >= 1/2. For a neural-network objective L(theta), define theta_dot = v and v_dot = -grad V(theta) - gamma S(theta,v)^q v, where V(theta) is the training loss plus any optional anchoring potential. Let S(theta,v) = (theta - theta_0)^T K^s (theta - theta_0) + ||v||_2^2 + epsilon. Here theta is the parameter vector, v is its velocity, theta_0 is the initialization or fine-tuning reference, K is a positive semidefinite layer or parameter graph operator, K^s is its fractional spectral power, epsilon > 0 avoids zero damping, and q controls the nonlinear damping strength. For E(theta,v) = V(theta) + ||v||_2^2/2, the continuous dynamics satisfy dE/dt = -gamma S(theta,v)^q ||v||_2^2 <= 0. Thus the damping coefficient increases with the energy scale and the energy decay rate is explicitly measurable.
Implementation notes
Integrate this as a PyTorch optimizer for an MLP, CNN, transformer, or parameter-efficient fine-tuning problem. Maintain a velocity tensor for every trainable parameter. Begin with K = 0, giving S_k = ||v_k||^2 + epsilon; only add layer-graph coupling after the scalar mechanism is validated. Let V equal the minibatch loss, optionally plus lambda times the squared distance from the initial parameters. At each step, compute the minibatch gradient g_k, flatten or aggregate all velocity tensors, calculate S_k = sum_i v_k,i^2 + epsilon, set c_k = gamma times S_k^q, and update v_k using v = v - eta*g - eta*c_k*v. Then update theta using theta = theta + eta*v. Also test the semi-implicit damping update v = (v - eta*g)/(1 + eta*c_k), which is safer when damping becomes large. Quantities computed directly are the gradient, velocity norm, damping coefficient, and empirical energy. The paper gives the continuous Lyapunov law, not an unconditional discrete-time guarantee. First train a two-layer MLP on MNIST with SGD-momentum, fixed-friction heavy-ball, and the adaptive optimizer, matching learning rates and parameter counts. Log E_k and D_k = gamma*S_k^q*||v_k||^2. For small eta, predict E_(k+1) - E_k approximately equal to -eta*D_k plus an order-eta-squared error; regression of energy change against -eta*D_k should have slope near one. Sweep eta and record the first point where fewer than 95 percent of steps decrease energy. The adaptive method should shift the oscillation or divergence boundary upward, while excessively large gamma should visibly reduce progress near minima.
Verification
Stage 1 · Toy mechanism gate: Failed ✗
Stage 2 · Mechanism transferred to benchmark: Not tested
Stage 2 · Practical benchmark result: Not run
Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Built an energy-adaptive inertial PyTorch optimizer with semi-implicit damping, plus toy verification and an MLP comparison. The continuous Lyapunov-law regression had slope 1.00006 with mean relative error 8.5e-5, and the predicted damping scaling 2q=1.5 was measured as 1.50000. However, the discrete stability sweep gave adaptive eta=0.28 versus fixed-friction eta=0.30, so the claimed upward stability-boundary shift was not demonstrated; the MLP result was promising but only one tiny fixed-seed task.
- Agent confidence
- 9/10
- Baseline
- Fixed-friction heavy-ball: test loss 0.2249, accuracy 90.67%, final train loss 0.3177; stability boundary eta=0.30.
- Idea
- Energy-adaptive: test loss 0.2000, accuracy 91.33%, final train loss 0.2991; continuous-law slope 1.00006, damping log-log slope 1.50000, but stability boundary eta=0.28.
Limitations: Only a scalar quadratic toy system and a small two-layer moons MLP were tested; no MNIST, CNN, transformer, graph operator K, fractional powers, anchoring potential, broad hyperparameter sweep, multiple seeds, wall-clock benchmark, or rigorous discrete-time stability analysis was performed.
How to run: python3 experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- experiment.py 6.4 KB View Raw
- report.md 1.4 KB View
- results.json 15.1 KB View Raw