Stable and Counterfactually Robust Physical World Models from Imposed Structure and Learned Physics
Organizations: Stony Brook University · Georgia Institute of Technology · Westlake University
Abstract
A world model learns to forecast how a physical system evolves from recorded trajectories, yet the systems it imitates obey physical laws that are neither fully supplied nor reliably respected. The model may create energy, drift or diverge over long rollouts, and answer a changed law query using the law observed during training. We ask how much general physical structure must be hard coded into a world model, and how much system-specific physics can then be learned from data, for four properties to hold simultaneously: second law compatible dissipation, correct responses to interventions on physical parameters, stability out to one hundred times the training horizon, and robustness to disturbances. The imposed structure is general: dynamics are generated from the gradient of a learned energy through a fixed reversible operator, the energy is restricted to a confining class, a one way port can remove energy but never inject it, the drive channel is known, and the intervened parameter enters through a separable map. The model learns the energy functional, constitutive relations, dissipation rate, and couplings. Across an electromagnetic cavity, a particle in cell grid, and a shallow-water fluid, models with roughly nine thousand parameters recover constitutive functions with unit slope, separate conserving from dissipating worlds by four orders of magnitude using a single set of weights, and transfer changes in sign, magnitude, rate, and gravity to unseen values, where equal-capacity models without the same structure perform at chance or worse. A nonlinear constitutive law is recovered with its curvature preserved and predicts a held-out intervention - better than a converged linear model.
Figures & tables
| system | learned function | slope vs. reference | |
|---|---|---|---|
| grid | field constitutive derivative | ||
| grid | field constitutive derivative | ||
| fluid | mass density | ||
| fluid | potential force |
| training range | |||
|---|---|---|---|
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Behaviour (regime) | Structured | Unstructured baseline | Ablated control |
|---|---|---|---|
| Bounded seeds, horizon (cavity) | ; worst case – | non-factored coupling: | |
| Decay-rate error; stationary error (cavity) | ; | injects power; hot | sign-free port: negative rate on seeds |
| Charge-sign flip, seeds (cavity; grid) | ; | ; | non-factored: ; |
| Gravity , nMSE (fluid) | non-separable: | ||
| Refusal of an unsupplied map (cavity; fluid; grid) | ; ; | — | oracle with the map: ; ; |
| Fitted rate, conserving / dissipating (cavity, seeds) | / | informed: rollout | no port: rate error |
| class | gauge-aligned | best linear | rollout | |||
|---|---|---|---|---|---|---|
| nonlinear (learned) | ||||||
| linear comparator | ||||||
| nonlinear (learned) | ||||||
| linear comparator | ||||||
| nonlinear (learned) | ||||||
| linear comparator |
| Structural element | Buys (evidence) | Mechanism in one line |
|---|---|---|
| Locality + field mediation (shared per-object functions) | composition; object add/remove; in-distribution competence at params where global params fail | novel scenes decompose into pointwise-familiar inputs; interaction structure matches the generator’s |
| Factored (multiplicative) constants | support-moving counterfactuals: sign, magnitude, rate, gravity; energy-consistent coupling stability | an architectural channel carries values data cannot; one shared coefficient forces force and back-reaction to book energy consistently |
| Confining learned energy | boundedness at horizon | conservation constrains the state only if its level sets are compact; coercive sublevels restrict the state when an energy-sublevel bound holds |
| One-way (non-negative) port | quadratic-channel contraction; empirical decay and temperature checks | the fixed sign constrains the designated damping channel, not total entropy or arbitrary learned energy |
| Known explicit drives | injection tracked, never manufactured | gain and loss live in separate, separately-audited components |
| Fixed conservative flow + learned functionals | constitutive recovery on tested support; grid ; fitted rates | the operator exposes only its image of the learned derivatives; gauges and excitation limit recovery |
| Measure | Continuous | Discrete | MuJoCo | Adversarial |
|---|---|---|---|---|
| Four-way accuracy | 0.860 | 0.890 | 0.870 | 0.837 |
| One-candidate rate | 0.533 | 0.480 | 0.487 | 0.390 |
| Error among one-candidate decisions | 0.156 | 0.090 | 0.116 | 0.068 |
| True-candidate coverage | 0.900 | 0.873 | 0.860 | 0.727 |
| Ambiguity recall | 0.700 | 0.880 | 0.760 | 0.860 |
| Fitting-failure recall | 1.000 | 1.000 | 1.000 | 1.000 |
| system | verdict | competence ratio | refusal/guard |
|---|---|---|---|
| Pendulum | PASS | 4.23e+04 | refuse 1527337.4x ✓ |
| Double pendulum | PASS b (competence vs subLyapunov) | 13.4 | |
| Spring lattice | PASS | 1.78e+04 | refuse 244.0x ✓ |
| RLC circuit | PASS b (counterfactual-vs-oracle) | 9.53e+04 | class 27317.2x ✓ |
| Heat rod | PASS | 1.9e+07 | refuse 16246015.1x ✓ |
| Advection–diffusion | PASS | 7.06e+07 | refuse 2096997.6x ✓ |
| world | truth row | selected | alarm | correct |
|---|---|---|---|---|
| Pendulum | H | H | – | ✓ |
| Double pendulum | H/H(nonsep) | H(nonsep) | – | ✓ |
| Spring lattice | H | H | – | ✓ |
| RLC circuit | H+port+drive | H+port+drive | – | ✓ |
| Heat rod | port-only | port-only | – | ✓ |
| Advection–diffusion | bracket(skew+port+drive) | bracket(skew+port+drive) | – | ✓ |
| Method | Supplied | Learned | Changes at test | Guaranteed |
|---|---|---|---|---|
| HNN / SymODEN ( Greydanus et al., 2019 ; Zhong et al., 2020b ) | symplectic form | evaluated inputs | Hamiltonian-form continuous dynamics; solver-dependent discrete behaviour | |
| D-HNN ( Sosanya and Greydanus, 2022 ) | + Rayleigh split | , dissipation | friction coefficient (evaluated) | none on the sign |
| Port-HNN ( Desai et al., 2021 ) | port-Hamiltonian form | , forcing, dissipation | inputs | passivity of the port |
| Compositional pH ( Neary and Topcu, 2023 ) | pH interconnection | subsystem , ports | composition | pH structure |
| GFINNs ( Lee et al., 2021 ) | GENERIC with degeneracy | none | degeneracy laws | |
| Comparative evaluation ( Zhong et al., 2021a ) | various | various | none | benchmark only |
World Model Learning
World model learning builds predictive models of how environments change under actions to support planning, simulation, and decision-making.
See all 414 papers