Physics is the Best Teacher: Consistency Learning for Time-Invariant Operators of Chaotic Dynamics
Organizations: National Taiwan University · California Institute of Technology · University of Washington
Abstract
Accelerating the prediction of long-term behavior in chaotic systems is crucial in scientific computing. However, existing methods rely on numerical solvers or autoregressive models that advance one small step at a time, which makes long horizons expensive. We instead view this problem as learning the system's time-invariant evolution operator, which jumps the state across a large time span in a single evaluation. To this end, we derive the consistency equations a time-invariant operator must satisfy, with differential and compositional objectives in physical time. These equations also connect the learned operator to the physics-prescribed instant dynamics, enabling physics embedding in consistency learning. Across five chaotic systems, we find that physics-distilled consistency makes both short-term trajectories and long-term statistics more accurate. The learned operator survives temporal extrapolation and requires one-tenth as many evaluations as autoregressive rollout, offering an efficient route to long-term simulation of chaotic dynamics.
Figures & tables
| in distribution | out of distribution | ||||||||||
| Method | NFE | ||||||||||
| Evolution operator | |||||||||||
| Data | 100 | 0.46 | 0.20 | 3.1 | 2.5 | 1.7 | 0.44 | 0.18 | 2.9 | 1.6 | 1.1 |
| Semigroup | 100 | 0.46 | 0.21 | 3.4 | 3.3 | 1.9 | 0.44 | 0.18 | 2.9 | 1.3 | 1.3 |
| Eulerian-self | 100 | 0.45 | 0.22 | 3.5 | 3.8 | 1.6 | 0.43 | 0.17 | 2.6 | 1.6 | 1.4 |
| Lagrangian-self | 100 | 0.45 | 0.21 | 3.4 | 3.2 | 1.5 | 0.48 | 0.18 | 2.9 | 2.0 | 0.94 |
| Lorenz | Chen | Rössler | Thomas | Halvorsen | |||||||
| Method | NFE | in-dist | OOD | in-dist | OOD | in-dist | OOD | in-dist | OOD | in-dist | OOD |
| Evolution operator | |||||||||||
| Data | 100 | 0.32 0.05 | 0.25 0.04 | 0.26 0.03 | 0.27 0.03 | 0.53 0.06 | 0.53 0.02 | 0.90 0.04 | 0.92 0.04 | 0.56 0.09 | 0.52 0.01 |
| Semigroup | 100 | 0.30 0.06 | 0.24 0.03 | 0.27 0.02 | 0.29 0.04 | 0.54 0.05 | 0.54 0.01 | 0.88 0.09 | 0.90 0.07 | 0.55 0.09 | 0.52 0.03 |
| Eulerian-self | 100 | 0.27 0.06 | 0.24 0.01 | 0.27 0.03 | 0.30 0.04 | 0.56 0.02 | 0.50 0.04 | 0.89 0.07 | 0.83 0.05 | 0.50 0.02 | 0.49 0.02 |
| Lagrangian-self | 100 | 0.22 0.03 | 0.31 0.14 | 0.31 0.05 | 0.32 0.06 | 0.56 0.06 | 0.54 0.03 | 0.85 0.15 | 0.87 0.04 | 0.55 0.07 | 0.56 0.02 |
| Method | Equation and distinguishing feature | NFE | Avg. ID / OOD |
| Evolution operator: elapsed-time step | |||
| Data | . Paired-trajectory MSE ( ). | 100 | 0.46 / 0.44 |
| Semigroup | . One jump agrees with two composed jumps. | 100 | 0.46 / 0.44 |
| Eulerian-self | . Model velocity at the starting state . | 100 | 0.45 / 0.43 |
| Lagrangian-self | . Model velocity at the predicted endpoint. | 100 | 0.45 / 0.48 |
| Eulerian-phys | . Governing-equation velocity at the starting state . | 100 | 0.28 / 0.27 |
| System | Parameters | Source | |
|---|---|---|---|
| Lorenz | Sprott [6] | ||
| Chen | Noorani et al. [30] , Eftekhari and Jafari [31] | ||
| Rössler | Sprott [6] | ||
| Thomas | Sprott and Chlouverakis [32] | ||
| Halvorsen | Sprott [33] |
| Hyperparameter | Value |
|---|---|
| Training | |
| Optimizer | Adam ( , ) [ 48 ] |
| LR scheduler | constant |
| Batch size | |
| Iterations | |
| Validation interval | iterations |
| Time (ms) | ||||||
|---|---|---|---|---|---|---|
| Method | NFE | Lorenz | Chen | Rössler | Thomas | Halvorsen |
| AR -pred | 1000 | 404.9 1.5 | 396.9 1.3 | 398.4 1.7 | 401.0 1.1 | 401.5 1.1 |
| CFO | 4000 | 1668.7 4.9 | 1636.4 4.3 | 1642.2 2.7 | 1650.3 4.0 | 1665.5 3.2 |
| Evolution operator | 100 | 45.1 0.1 | 44.3 0.2 | 44.0 0.1 | 45.0 0.8 | 44.8 0.2 |
| Lorenz | Chen | Rössler | Thomas | Halvorsen | |||||||
| Method | NFE | in-dist | OOD | in-dist | OOD | in-dist | OOD | in-dist | OOD | in-dist | OOD |
| Evolution operator | |||||||||||
| Data | 20 | 0.99 0.27 | 0.71 0.10 | 0.66 0.03 | 0.64 0.09 | 3.1 0.8 | 2.9 0.3 | 2.8 0.5 | 2.5 0.4 | 1.4 0.4 | 1.3 0.1 |
| Semigroup | 20 | 0.97 0.17 | 0.63 0.09 | 0.70 0.04 | 0.67 0.10 | 3.3 0.8 | 3.1 0.9 | 3.1 1.2 | 2.6 0.3 | 1.5 0.4 | 1.2 0.2 |
| Eulerian-self | 20 | 0.91 0.39 | 0.73 0.35 | 0.64 0.13 | 0.74 0.22 | 3.4 0.3 | 2.5 0.6 | 2.5 0.3 | 2.2 0.1 | 1.2 0.2 | 1.0 0.1 |
| Lagrangian-self | 20 | 0.71 0.19 | 0.83 0.19 | 1.1 0.4 | 0.86 0.24 | 3.5 1.1 | 2.8 0.5 | 2.4 0.7 | 2.4 0.3 | 1.5 0.7 | 1.5 0.2 |
| Thomas | Halvorsen | ||||||||||
| Method | NFE | ||||||||||
| Evolution operator | |||||||||||
| Data | 100 | 0.90 | 0.14 | 5.1 | 2.2 | 1.8 | 0.56 | 0.048 | 3.3 | 2.2 | 0.89 |
| Semigroup | 100 | 0.88 | 0.13 | 4.5 | 2.1 | 2.4 | 0.55 | 0.050 | 3.7 | 3.9 | 0.74 |
| Eulerian-self | 100 | 0.89 | 0.16 | 6.4 | 27 | 3.2 | 0.50 | 0.046 | 3.2 | 2.6 | 0.86 |
| Lagrangian-self | 100 | 0.85 | 0.15 | 5.6 | 1.7 | 2.6 | 0.55 | 0.049 | 3.8 | 6.0 | 1.1 |
| Thomas | Halvorsen | ||||||||||
| Method | NFE | ||||||||||
| Evolution operator | |||||||||||
| Data | 100 | 0.92 | 0.13 | 4.3 | 1.1 | 0.88 | 0.52 | 0.047 | 2.2 | 1.7 | 0.90 |
| Semigroup | 100 | 0.90 | 0.14 | 5.2 | 0.90 | 2.1 | 0.52 | 0.047 | 2.5 | 1.1 | 1.1 |
| Eulerian-self | 100 | 0.83 | 0.13 | 4.8 | 1.2 | 1.9 | 0.49 | 0.048 | 3.0 | 3.1 | 0.53 |
| Lagrangian-self | 100 | 0.87 | 0.14 | 4.1 | 0.77 | 1.5 | 0.56 | 0.049 | 4.0 | 5.8 | 0.72 |
| Lorenz | Chen | Rössler | Thomas | Halvorsen | ||||||
| Consistency term | in-dist | OOD | in-dist | OOD | in-dist | OOD | in-dist | OOD | in-dist | OOD |
| Self-distillation | ||||||||||
| Eulerian | 0.27 0.06 | 0.24 0.01 | 0.27 0.03 | 0.30 0.04 | 0.56 0.02 | 0.50 0.04 | 0.89 0.07 | 0.83 0.05 | 0.50 0.02 | 0.49 0.02 |
| Lagrangian | 0.22 0.03 | 0.31 0.14 | 0.31 0.05 | 0.32 0.06 | 0.56 0.06 | 0.54 0.03 | 0.85 0.15 | 0.87 0.04 | 0.55 0.07 | 0.56 0.02 |
| Eulerian Lagrangian | 0.21 0.04 | 0.31 0.13 | 0.27 0.04 | 0.27 0.03 | 0.55 0.03 | 0.50 0.02 | 0.81 0.05 | 0.84 0.07 | 0.49 0.03 | 0.46 0.02 |
| Physics-distillation | ||||||||||
Chaotic Time-Series Prediction
Chaotic time-series prediction studies how to forecast nonlinear systems whose sensitive dependence on initial conditions limits long-range accuracy.
See all 54 papers