atlas

Research map / Probabilistic and scientific modeling / Dynamical systems

Harnessing the Power of Neural Operators with Automatically Encoded Conservation Laws

ICML 2024 · Probabilistic and scientific modeling › Dynamical systems · approach: Physics-informed dynamics learning · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces clawNOs, neural operators whose outputs are divergence-free by construction (curl of a skew-symmetric matrix field via a fixed numerical differentiation layer), enforcing mass conservation automatically and improving small-data accuracy across Navier-Stokes, shallow water, atmospheric and material deformation tasks.

Problem

Neural operators learn physical laws from data and only approximately satisfy conservation laws (e.g., mass/continuity) due to finite and noisy data, which hurts accuracy and physical consistency especially in small-data regimes.

Approach

Parameterize the output as the divergence of a skew-symmetric matrix-valued function (from differential forms, d-star-d = 0), so the continuity equation holds automatically; the divergence is computed by a pre-calculated numerical differentiation layer (Fourier spectral differentiation on periodic uniform grids, or meshfree weighted finite differences on non-uniform grids) appended to existing FNO

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add the divergence-free conservation-law architecture to GFNO (ablation)GFNO · Navier–Stokes, T_in=10, T=20; 10, 100, and 1000 training samplesrelative L2 error: 29.56% ± 3.32% -> 12.10% ± 1.08%; 7.59% ± 0.34% -> 4.76% ± 0.24%; 3.11% ± 0.93% -> 1.21% ± 0.09%Helped
Add the divergence-free conservation-law architecture to FNO (ablation)FNO · Navier–Stokes, T_in=10, T=20; 10, 100, and 1000 training samplesrelative L2 error: 22.57% ± 1.87% -> 12.68% ± 0.85%; 5.68% ± 0.41% -> 5.17% ± 0.27%; 1.47% ± 0.18% -> 1.46% ± 0.11%Helped
Use equivariant GFNO rather than FNO as the clawNO baseclawNO · Navier–Stokes, T_in=10, T=20relative L2 error: 12.68% ± 0.85% -> 12.10% ± 1.08%; 5.17% ± 0.27% -> 4.76% ± 0.24%; 1.46% ± 0.11% -> 1.21% ± 0.09%Helped
Use clawGFNO instead of the strongest non-claw baselineNavier–Stokes models · Navier–Stokes, both rollout settingsrelative L2 error: 40.3%, 16.2% and 17.6% in case 1; 39.9%, 26.8% and 9.1% in case 2Helped
Increase GFNO latent dimension from 11 to 20 (ablation)GFNO · Navier–Stokes, T_in=10, T=20, 1000 training samplesrelative L2 error: 1.19% ± 0.16%Helped
Add the divergence-free conservation-law architecture to FNO (ablation)FNO · Radial dam break, 2 and 10 training samplesrelative L2 error: 29.88% ± 0.17% -> 21.24% ± 1.17%; 4.04% ± 0.10% -> 2.69% ± 0.04%Helped
Add the divergence-free conservation-law architecture to FNO (ablation)FNO · Radial dam break, 100 training samplesrelative L2 error: 1.88% ± 0.05% -> 1.84% ± 0.01%Helped
Use clawNO rather than the best non-claw baselineRadial dam break models · Radial dam break, 2 and 10 training samplesrelative L2 error: 22.2% in small data regime; 29.2% in medium data regimeHelped
Add the divergence-free conservation-law architecture to FNO (ablation)FNO · Atmospheric modelingrelative L2 error: 12.54% ± 0.26% -> 6.64% ± 0.23%Helped
Add the divergence-free conservation-law architecture to INO (ablation)INO · Material deformation, 2, 20, and 100 training samplesrelative L2 error: 16.75% ± 1.24% -> 14.12% ± 0.23%; 7.69% ± 0.51% -> 5.41% ± 0.67%; 2.40% ± 0.53% -> 1.00% ± 0.02%Helped
Use clawINO rather than the best non-claw baselineMaterial deformation models · Material deformation, 2, 20, and 100 training samplesrelative L2 error: 15.7%, 26.4%, and 23.1%Helped
Use clawNOs rather than unconstrained neural operatorsNeural operators · Navier–Stokes test set, 1000 training samplesL2-norm divergence: 1.2e-3; 3.8e-4 vs 6.6e-2; 4.7e-2Helped

Surprising findings

Limitations

Experimental setup

Datasets: incompressible Navier-Stokes simulations (pseudo-spectral Crank-Nicolson solver) radial dam break simulations SpeedyWeather.jl atmospheric simulations Mooney-Rivlin material deformation simulations

Benchmarks: incompressible Navier-Stokes radial dam break atmospheric shallow water (SpeedyWeather.jl) Mooney-Rivlin material deformation

Metrics: relative L2 error L2 norm of divergence trainable parameter count runtime GPU memory

Baselines: FNO G-FNO UNet LSM KNO UNO BOON LNN GNO INO MP-PDE MeshGraphNets

Builds on: Fourier neural operator graph neural operator invariant neural operator group-equivariant FNO physics-informed neural networks

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy