Nonreversible latent instanton sampler
Implementation & benchmark of arXiv:2607.05194 — Nucleation and time-reversal symmetry breaking in nonconserved scalar field theories
Source paper: Nucleation and time-reversal symmetry breaking in nonconserved scalar field theories arXiv:2607.05194 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Represent a rare transition in a neural latent space by a controlled path whose drift is optimized directly, instead of obtaining it by reversing the relaxation dynamics. Learn a state-dependent mobility or diffusion matrix so that the sampler allocates noise and control effort according to the local stochastic geometry. This should improve generation of low-probability transitions in nonequilibrium world models and reduce the number of failed trajectories.
Formulas
Mathematical statement
The paper's reduced stochastic dynamics is $\dot{R}=\mathcal{M}(R)[-\partial_R U(R)]+\sqrt{2T\mathcal{M}(R)}\,\eta(t)$, where $R$ is a scalar reaction coordinate, $U(R)$ is an effective free energy or quasipotential, $\mathcal{M}(R)>0$ is the state-dependent mobility, $T$ is the small-noise scale, and $\eta(t)$ is unit Gaussian white noise with $\langle\eta(t)\eta(t')\rangle=\delta(t-t')$. The transferable point is that the noise amplitude and deterministic drift are coupled through mobility, while nonequilibrium optimal paths need not be time reversals of relaxation. For a learned latent SDE $dz=b_\theta(z,t)dt+\sigma_\theta(z,t)dW_t$, use the Freidlin–Wentzell discretization $S[z]=\frac{1}{2}\sum_{k=0}^{K-1}\Delta t\,(\dot z_k-b_\theta(z_k,t_k))^\top a_\theta(z_k,t_k)^{-1}(\dot z_k-b_\theta(z_k,t_k))$, where $a_\theta=\sigma_\theta\sigma_\theta^\top$, $\dot z_k=(z_{k+1}-z_k)/\Delta t$, and $S$ measures the small-noise action of a path. Optimize a separate reverse transition drift $r_\phi(z,t)$ against this action and endpoint constraints; do not impose $r_\phi=-b_\theta$ or assume reverse paths equal relaxation paths.
Implementation notes
Integrate this at the latent transition module of a world model or at the reverse sampler of a diffusion-like generator. First train or freeze an encoder-decoder $(E,D)$ and a baseline latent drift $b_\theta(z,t)$ on ordinary trajectories. Add a positive diagonal mobility network $M_\psi(z,t)=\operatorname{softplus}(h_\psi(z,t))+\epsilon$ and set $a_\psi(z,t)=2T M_\psi(z,t)$; this is the vector analogue of the paper's scalar noise amplitude $2T\mathcal{M}(R)$. For each source-target pair $(z_A,z_B)$, initialize a $K$-step path by linear interpolation plus noise. Compute $\dot z_k=(z_{k+1}-z_k)/\Delta t$, evaluate the action $S$ above, and update either the path variables directly for 20--100 inner gradient steps or predict the path with a small path network $z_{\phi,k}=g_\phi(z_A,z_B,k)$. Use endpoint penalties with $\lambda_0,\lambda_1$ and optionally the reverse-drift consistency penalty in the third formula. Estimate $a^{-1}$ analytically because the mobility is diagonal; estimate the latent reaction coordinate only if desired, using the norm or a learned one-dimensional projection. The paper supplies the mobility-weighted stochastic form and the non-time-reversal principle; the discretized action and neural parameterization are the ML adaptation. Begin on a 2D double-well or nonequilibrium rotational Langevin system, comparing standard reverse-dynamics sampling against this action-optimized sampler at equal neural evaluations. Track successful endpoint reach rate, mean action, transition probability, and trajectory diversity. A useful signal is at least 2x higher successful transition rate or substantially lower action at equal rollout budget, without worse reconstruction likelihood on ordinary trajectories.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significantly worse ✗
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 6/10
Built a 2D nonequilibrium double-well MVP with state-dependent diagonal mobility, mobility-weighted Freidlin–Wentzell action, direct endpoint-constrained path optimization, and a reverse-relaxation baseline. The stochastic moment check matched the prescribed SDE, and action optimization reduced action from 14.27 for the straight path to 1.01 while perturbations increased it to 25.89. In 1,200 stochastic trials, the optimized-path controller reached the target in 98.75% of trials versus 0.17% for reverse relaxation, but the baseline often diverged, so this is a promising toy signal rather than a fair neural-model benchmark.
- Agent confidence
- 6/10
- Baseline
- reverse-relaxation success=0.0017; finite trajectories=173/1200; mean minimum target distance among finite trajectories=1.262
- Idea
- action-optimized path success=0.9875; finite trajectories=1200/1200; mean final target distance=0.148; mean minimum target distance=0.102; optimized action=1.007 versus straight-path action=14.274
Limitations: No encoder-decoder, learned neural drift, or learned mobility network was trained. The baseline reverse dynamics became numerically unstable and receives no analogous stabilizing path controller, so the success-rate gap may overstate the practical advantage. Only one 2D toy system, one seed, one discretization, and one target pair were tested; no FLOP-matched neural experiment or trajectory-diversity analysis was performed.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.05194)
Significantly WORSE than the tuned baseline (+1859.0%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0014
- Idea mean
- 0.0278
- Effect (Δ)
- 0.0264 (+1859.0%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Tested only the built-in 8-step actuated-pendulum regression track, one shared rnn_small architecture, 12 training epochs, and a three-point learning-rate sweep. The intervention used a hand-designed local physics proxy and diagonal mobility surrogate rather than a learned reverse drift or learned mobility network; no latent encoder-decoder, rare-transition success-rate, diversity, or generation-time trajectory benchmark was 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 Failed ✗
Agent confidence: 9/10
Built a local nonreversible latent-instanton-inspired training intervention for the structurally matched actuated-pendulum dynamics track, using the shared rnn_small architecture and a mobility-weighted Freidlin–Wentzell-style residual penalty. The baseline achieved mean test MSE 0.001420, while the idea achieved 0.027814; paired delta was +0.026394 (higher is worse) with permutation p=0.0081, so the idea was significantly worse. The trained-model mechanism signature did confirm the predicted weighting direction, but this does not override the negative task-metric result.
- Baseline
- best lr=0.01; sweep means: lr=0.001 -> 0.004299, lr=0.003 -> 0.003462, lr=0.01 -> 0.001487; full 8-seed mean MSE=0.001420, std=0.000398
- Idea
- best lr=0.01 with action penalty coefficient 0.02; full 8-seed mean MSE=0.027814, std=0.005122
- Mechanism signature
- map[confirmed:true observed_mobility_weighted_residual:6.362071514129639 observed_residual_mse:5.364831447601318 predicted_direction:weighted residual exceeds unweighted residual for mobility below one weighted_to_unweighted_ratio:1.1858846967082626]
- Paired delta
- idea - baseline = +0.026394 MSE
- Permutation p value
- 0.0081
Benchmark result
Significantly WORSE than the tuned baseline (+1859.0%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0014
- Idea mean
- 0.0278
- Effect (Δ)
- 0.0264 (+1859.0%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- n/a
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Tested only the built-in 8-step actuated-pendulum regression track, one shared rnn_small architecture, 12 training epochs, and a three-point learning-rate sweep. The intervention used a hand-designed local physics proxy and diagonal mobility surrogate rather than a learned reverse drift or learned mobility network; no latent encoder-decoder, rare-transition success-rate, diversity, or generation-time trajectory benchmark was tested.
How to run: python3 bench_experiment.py
Artifacts
- bench_experiment.py 6.2 KB View Raw
- bench_report.json 2.4 KB View Raw
- experiment.py 5.2 KB View Raw
- report.md 1.7 KB View
- report_bench_2026-09-04T175122.md 4.3 KB View
- results.json 1.1 KB View Raw