English

Prediction Is Not Physics: Learning and Evaluating Conserved Quantities in Neural Simulators

Machine Learning 2026-05-20 v1 Artificial Intelligence

Abstract

A diffusion model trained on Hamiltonian trajectories can achieve rollout MSE near 10310^{-3}, but the standard deviation of its energy over time is between 7500 and 36000 times larger than the ground-truth energy standard deviation, indicating a failure to preserve conservation laws. This gap motivates our central question of whether neural networks can learn or select globally conserved quantities from physical trajectories. We investigate this across three Hamiltonian systems: projectile motion, pendulum, and spring-mass. We use a structured T(v)+V(q)T(v)+V(q) energy model, a black-box Conservation Discovery Network (CDN), a polynomial CDN, and a conditional diffusion baseline. The structured network reaches R20.9999R^2 \geq 0.9999 against analytical energy on clean data, while the black-box CDN reaches R20.996R^2 \geq 0.996 when trained with temporal consistency plus a small alignment loss to analytical energy at t=0t=0 (λalign=0.2\lambda_{\mathrm{align}}=0.2). With λalign=0\lambda_{\mathrm{align}}=0, CDN Pearson R2R^2 collapses on pendulum and spring-mass (<103< 10^{-3}), showing that temporal consistency alone is not enough to reliably identify the true energy. Under 1%1\% additive Gaussian noise, the CDN outperforms the structured model on the projectile and spring-mass systems, suggesting that the CDN may be more robust to noisy inputs in this setting. However, the polynomial CDN is sensitive to training configuration: it achieves R2=0.78R^2=0.78 under a short training schedule on the pendulum system, but reaches R2=0.9998R^2=0.9998 with more training time and data, regardless of whether noise is added.

Keywords

Cite

@article{arxiv.2605.18883,
  title  = {Prediction Is Not Physics: Learning and Evaluating Conserved Quantities in Neural Simulators},
  author = {Andrew Bukowski and Aditya Kothari and Simba Shi and Ishir Rao},
  journal= {arXiv preprint arXiv:2605.18883},
  year   = {2026}
}

Comments

10 pages