Defect-and-Jacobian Residual Dynamics
Implementation & benchmark of arXiv:2607.24345 — Perturbative-NeuSA: A Structured Spectral Framework for Time-Dependent PDEs
Source paper: Perturbative-NeuSA: A Structured Spectral Framework for Time-Dependent PDEs arXiv:2607.24345 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace full-state prediction in a neural simulator or neural operator with prediction of a perturbation around a cheap structured background trajectory. Compute the background defect and known linearized or nonlinear corrections explicitly, and let the neural closure model only the remaining residual. Add a residual-magnitude gate so the learned closure is suppressed when the structured solver already explains the target dynamics.
Formulas
Mathematical statement
The paper decomposes the full state as h(t)=h_bg(t)+h_tilde(t), where h_bg is a low-fidelity background state and h_tilde is the unresolved perturbation. For the Klein-Gordon system, u_t=v and v_t=u_xx-m sin(u), with spatial coordinate x, time t, and scalar parameter m. Writing u=u_bg+u_tilde gives u_tilde_t=v_tilde and v_tilde_t=u_tilde_xx-m u_tilde+R_bg-m[cos(u_bg)-1]u_tilde-m r_2, where R_bg=u_bg,xx-m sin(u_bg)-v_bg,t is the background defect and r_2=sin(u_bg+u_tilde)-sin(u_bg)-cos(u_bg)u_tilde is the Taylor remainder. The term -m u_tilde is a fixed linear operator, -m[cos(u_bg)-1]u_tilde is a background-dependent correction, and r_2 is the unresolved nonlinear remainder. For a general known dynamical system partial_t h=F(h), use R_bg=F(h_bg)-partial_t h_bg, a chosen Jacobian or linear operator J_bg, and train a closure N_theta for the nonlinear remainder: partial_t h_tilde=J_bg h_tilde+R_bg+N_theta(h_bg,h_tilde).
Implementation notes
(1) Integration point: use this transition rule inside a neural PDE solver, latent world model, or continuous-time state-space model. Maintain a cheap background state h_bg at every step using a coarse finite-difference, spectral, linear, or low-resolution neural solver. Maintain the perturbation h_tilde=h-h_bg, and make the network predict only the unresolved derivative or next-step perturbation. A concrete first implementation is the one-dimensional periodic Klein-Gordon system, with FFT evaluation of u_xx and an Euler or RK2 background integrator. (2) Pseudocode: bg_next = background_step(bg, dt); defect = F(bg) - bg_derivative; linear_u = residual_v; linear_v = laplacian(residual_u) - m*residual_u; correction_v = -m*(cos(bg_u)-1)*residual_u; closure = net(concat(bg, residual, defect)); gate = sigmoid(a*(log(norm(defect)+eps)-b)); residual_derivative = linear + defect + correction + gate*closure; residual_next = integrator(residual, residual_derivative, dt); state_next = bg_next + residual_next. (3) Compute FFT derivatives, the background defect, and the Jacobian correction analytically. Estimate only normalization statistics and gate parameters a and b from training data. Train with state and derivative losses, such as ||state_pred-state_true||_2^2 + lambda*||residual_derivative-residual_derivative_true||_2^2, plus a small penalty on closure energy when the defect norm is small. (4) First experiment: train a small Fourier neural operator or four-layer CNN on one-dimensional Klein-Gordon trajectories. Compare full-state prediction against the perturbative model at equal parameter count and FLOPs, sweeping background resolution from very coarse to nearly exact. Measure one-step error, long-horizon rollout error, training loss versus optimizer steps, and closure norm versus background accuracy. Success means lower rollout error or faster loss descent at equal compute, with the closure norm decreasing as background resolution improves. Falsification is a neutral or negative result for all background resolutions, or instability caused by the residual integration.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significant win ✓
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 8/10
Built and ran a gated defect/Jacobian residual dynamics module for a reduced Klein–Gordon oscillator. The residual identity error was 0, the Taylor remainder log-log slope was 2.003 versus the predicted 2, its quadratic coefficient was 0.33345 versus 0.33343, and the gate midpoint/slope were 0.08035/3.000 versus predicted 0.08000/3. The residual model also improved one-step RMSE from 0.02396 to 0.01416 and 60-step rollout RMSE from 0.37699 to 0.24086, so the mechanism manifested in this toy test.
- Agent confidence
- 8/10
- Baseline
- Full-state MLP: one-step RMSE 0.0239556; 60-step rollout RMSE 0.376986
- Idea
- Defect/Jacobian residual MLP with sigmoid defect gate: one-step RMSE 0.0141616; 60-step rollout RMSE 0.240856; Taylor slope 2.0027 vs 2 predicted; gate midpoint 0.08035 vs 0.080 predicted; gate logit slope 3.000 vs 3 predicted
Limitations: This is a single Fourier-mode/reduced Klein–Gordon toy rather than a spatial PDE with FFT derivatives or a resolution sweep. It uses one fixed random seed, a small MLP, short training, and does not establish scaling, FLOP parity, or gains on a neural operator or high-dimensional simulator.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.24345)
Beats the tuned baseline by -0.0001 metric points (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- -0.0001 (n/a; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
Only the built-in controlled-pendulum dynamics track was tested, using 1000 training samples, 400 test samples, 15 epochs, and the shared rnn_small architecture. Spatial Klein-Gordon PDEs, FFT resolution scaling, long-horizon rollout, FLOP timing, and broader background-resolution sweeps were not tested.
How to run: python3 bench_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Worked ✓
Agent confidence: 8/10
The residual/Jacobian system was implemented end-to-end on the structurally matched controlled-pendulum dynamics track using the shared rnn_small GRU architecture. It significantly outperformed the tuned baseline across all 8 paired seeds: delta_mean=-0.000116099 and permutation p=0.0081. The trained-model signature confirmed the predicted mechanism because closure magnitude rose with defect, correlation 0.990.
- Baseline
- rnn_small, tuned lr=0.01; mean test MSE=0.0001201643, std=0.0000567395.
- Idea
- ResidualRNN with analytic pendulum background, residual coordinates, and sigmoid defect gate; best lr=0.003; mean test MSE=0.00000406562, std=0.00000015780.
Benchmark result
Beats the tuned baseline by -0.0001 metric points (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0
- Effect (Δ)
- -0.0001 (n/a; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
Only the built-in controlled-pendulum dynamics track was tested, using 1000 training samples, 400 test samples, 15 epochs, and the shared rnn_small architecture. Spatial Klein-Gordon PDEs, FFT resolution scaling, long-horizon rollout, FLOP timing, and broader background-resolution sweeps were not tested.
How to run: python3 bench_experiment.py
Artifacts
- bench_experiment.py 4.9 KB View Raw
- bench_report.json 4.8 KB View Raw
- experiment.py 5.5 KB View Raw
- report.md 1.4 KB View
- report_bench_2026-08-31T092200.md 3.1 KB View
- results.json 0.7 KB View Raw