Learning to Control PDEs with Differentiable Physics
Not on the map. Why?
Abstract (arXiv)
Predicting outcomes and planning interactions with the physical world are long-standing goals for machine learning. A variety of such tasks involves continuous physical systems, which can be described by partial differential equations (PDEs) with many degrees of freedom. Existing methods that aim to control the dynamics of such systems are typically limited to relatively short time frames or a small number of interaction parameters. We present a novel hierarchical predictor-corrector scheme which enables neural networks to learn to understand and control complex nonlinear physical systems over long time frames. We propose to split the problem into two distinct tasks: planning and control. To this end, we introduce a predictor network that plans optimal trajectories and a control network that infers the corresponding control parameters. Both stages are trained end-to-end using a differentiable PDE solver. We demonstrate that our method successfully develops an understanding of complex physical systems and learns to control them for tasks involving PDEs such as the incompressible Navier-Stokes equations.
Key facts. Cited by 131 papers in the map (data through 7 Oct 2026). First posted on arXiv 21 Jan 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 25; 2022: 30; 2023: 22; 2024: 27; 2025: 11; 2026: 16.
131 papers.
Sort:
By area:
- 2022Liquid Structural State-Space Modelscited by 134
- 2021PlasticineLab: A Soft-Body Manipulation Benchmark with Differentiable Physicscited by 100
- 2023DeepZero: Scaling up Zeroth-Order Optimization for Deep Model Trainingcited by 76
- 2022RoboCraft: Learning to See, Simulate, and Shape Elasto-Plastic Objects with Graph Networkscited by 69
- 2023Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulationcited by 65
- 2021DiffCloth: Differentiable Cloth Simulation with Dry Frictional Contactcited by 64
- 2022When Physics Meets Machine Learning: A Survey of Physics-Informed Machine Learningcited by 64
- 2021DiffPD: Differentiable Projective Dynamicscited by 62
- 2021Fast and Feature-Complete Differentiable Physics for Articulated Rigid Bodies with Contactcited by 50
- 2022NeuPhysics: Editable Neural Geometry and Physics from Monocular Videoscited by 50
- 2024MotionCraft: Physics-based Zero-Shot Video Generationcited by 46
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 43
38 more
- 2024ConFIG: Towards Conflict-free Training of Physics Informed Neural Networkscited by 43
- 2021NTopo: Mesh-free Topology Optimization using Implicit Neural Representationscited by 41
- 2021Solving PDE-constrained Control Problems Using Operator Learningcited by 41
- 2022Learned Turbulence Modelling with Differentiable Fluid Solvers: Physics-based Loss-functions and Optimisation Horizonscited by 39
- 2022Differentiable Simulation of Soft Multi-body Systemscited by 39
- 2021Efficient Differentiable Simulation of Articulated Bodiescited by 38
- 2022Learning to Accelerate Partial Differential Equations via Latent Global Evolutioncited by 38
- 2022Learning to Solve PDE-constrained Inverse Problems with Graph Networkscited by 37
- 2023Geometric Clifford Algebra Networkscited by 37
- 2023FluidLab: A Differentiable Environment for Benchmarking Complex Fluid Manipulationcited by 37
- 2022NeuroFluid: Fluid Dynamics Grounding with Particle-Driven Neural Radiance Fieldscited by 36
- 2022A posteriori learning for quasi-geostrophic turbulence parametrizationcited by 36
- 2023Physics-Informed Machine Learning for Modeling and Control of Dynamical Systemscited by 28
- 2024Wavelet Diffusion Neural Operatorcited by 28
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Surveycited by 27
- 2021Learning neural network potentials from experimental data via Differentiable Trajectory Reweightingcited by 26
- 2024Clifford-Steerable Convolutional Neural Networkscited by 24
- 2021Causal Navigation by Continuous-time Neural Networkscited by 23
- 2023DexDeform: Dexterous Deformable Object Manipulation with Human Demonstrations and Differentiable Physicscited by 23
- 2025PDE-Controller: LLMs for Autoformalization and Reasoning of PDEscited by 23
- 2022Differentiable Programming for Earth System Modelingcited by 21
- 2023Deep Learning in Deterministic Computational Mechanicscited by 21
- 2021Combining Physics and Deep Learning to learn Continuous-Time Dynamics Modelscited by 20
- 2025Sparse identification of nonlinear dynamics and Koopman operators with Shallow Recurrent Decoder Networkscited by 19
- 2021Global Transport for Fluid Reconstruction with Learned Self-Supervisioncited by 18
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packingcited by 18
- 2021Learning optimal multigrid smoothers via neural networkscited by 17
- 2024DiffPhyCon: A Generative Approach to Control Complex Physical Systemscited by 17
- 2021SAGCI-System: Towards Sample-Efficient, Generalizable, Compositional, and Incremental Robot Learningcited by 16
- 2022Training Efficient Controllers via Analytic Policy Gradientcited by 16
- 2023Solving Inverse Physics Problems with Score Matchingcited by 15
- 2024Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modelingcited by 13
- 2024SineNet: Learning Temporal Dynamics in Time-Dependent Partial Differential Equationscited by 13
- 2022FINDE: Neural Differential Equations for Finding and Preserving Invariant Quantitiescited by 12
- 2023Disentangled Multi-Fidelity Deep Bayesian Active Learningcited by 11
- 2023Almost Equivariance via Lie Algebra Convolutionscited by 10
- 2024Approximating Numerical Fluxes Using Fourier Neural Operators for Hyperbolic Conservation Lawscited by 10
- 2024NeuralFluid: Neural Fluidic System Design and Control with Differentiable Simulationcited by 10
50 most cited of 131 All 131 (JSON)
- 2026Physics-Informed Method of Group Data Handling: Adaptive Construction of Functional Representations with an Application to the Navier-Stokes Equations30 Sep 2026
- 2026Gradient Surgery for Physics-Informed Neural Networks25 Sep 2026
- 2026LaGSplat: Inferring Physics-Governed Interactive Simulation from Monocular Video Using Latent Lagrangian Gaussian Splatting17 Aug 2026
- 2026Differentiate the Solver, Not the Equation: Reverse-Sweep Adjoints for Block Implicit Simulation9 Aug 2026
- 2026Large language models for partial differential equation workflows4 Aug 2026
- 2026Toward Goal-Agnostic Joint-Embedding Predictive Control of Partial Differential Equations21 Jul 2026
- 2026Physics-enhanced reinforcement learning for real-time optimal control of dynamical systems17 Jul 2026
- 2026Learning Physics-Guided Residual Dynamics for Deformable Object Simulation15 Jul 2026
- 2026Integrating Mechanistic and Data-Driven Models for Neurological Disorders through Differentiable Programming4 Jun 2026
- 2026Deep Learning-based Algebraic Reynolds Stress Closures for RANS Simulations of Turbulent Flows25 May 2026
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospects16 May 2026
- 2026Provable imitation learning for control of instability in partially-observed Vlasov--Poisson equations6 May 2026
38 more
- 2026A solver-in-the-loop framework for end-to-end differentiable coastal hydrodynamics8 Apr 2026
- 2026Neural Field Thermal Tomography: A Differentiable Physics Framework for Non-Destructive Evaluation11 Mar 2026
- 2026Learning Physical Operators using Neural Operators26 Feb 2026
- 2026Accelerated Sequential Flow Matching: A Bayesian Filtering Perspective5 Feb 2026
- 2025INC: An Indirect Neural Corrector for Auto-Regressive Hybrid PDE Solvers16 Nov 2025
- 2025Autonomous Vehicle Path Planning by Searching With Differentiable Simulation14 Nov 2025
- 2025Reversible GNS for Dissipative Fluids with Consistent Bidirectional Dynamics26 Sep 2025
- 2025Self-Supervised Coarsening of Unstructured Grid with Automatic Differentiation24 Jul 2025
- 2025From 2D to 3D Cognition: A Brief Survey of General World Models25 Jun 2025
- 2025PICT -- A Differentiable, GPU-Accelerated Multi-Block PISO Solver for Simulation-Coupled Learning Tasks in Fluid Dynamics22 May 2025
- 2025Emergence of Computational Structure in a Neural Network Physics Simulator16 Apr 2025
- 2025Unlocking Efficient Vehicle Dynamics Modeling via Analytic World Models14 Feb 2025
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Control4 Feb 2025
- 2025PDE-Controller: LLMs for Autoformalization and Reasoning of PDEs3 Feb 2025
- 2025Sparse identification of nonlinear dynamics and Koopman operators with Shallow Recurrent Decoder Networks23 Jan 2025
- 2024Wavelet Diffusion Neural Operator6 Dec 2024
- 2024Physics Encoded Blocks in Residual Neural Network Architectures for Digital Twin Models18 Nov 2024
- 2024Multi-modal Policies with Physics-informed Representations in Complex Fluid Environments20 Oct 2024
- 2024Recent Trends in Modelling the Continuous Time Series using Deep Learning: A Survey13 Sep 2024
- 2024Autonomous Vehicle Controllers From End-to-End Differentiable Simulation12 Sep 2024
- 2024Bridging Autoencoders and Dynamic Mode Decomposition for Reduced-order Modeling and Control of PDEs9 Sep 2024
- 2024Differentiable programming across the PDE and Machine Learning barrier9 Sep 2024
- 2024A domain decomposition-based autoregressive deep learning model for unsteady and nonlinear partial differential equations26 Aug 2024
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Survey22 Aug 2024
- 2024ConFIG: Towards Conflict-free Training of Physics Informed Neural Networks20 Aug 2024
- 2024CL-DiffPhyCon: Closed-loop Diffusion Control of Complex Physical Systems31 Jul 2024
- 2024DiffPhyCon: A Generative Approach to Control Complex Physical Systems9 Jul 2024
- 2024RoboPack: Learning Tactile-Informed Dynamics Models for Dense Packing1 Jul 2024
- 2024Approximately Equivariant Neural Processes19 Jun 2024
- 2024ConDiff: A Challenging Dataset for Neural Solvers of Partial Differential Equations7 Jun 2024
- 2024NeuralFluid: Neural Fluidic System Design and Control with Differentiable Simulation22 May 2024
- 2024MotionCraft: Physics-based Zero-Shot Video Generation22 May 2024
- 2024PDE Control Gym: A Benchmark for Data-Driven Boundary Control of Partial Differential Equations18 May 2024
- 2024Stabilizing Backpropagation Through Time to Learn Complex Physics3 May 2024
- 2024Neural Operators Learn the Local Physics of Magnetohydrodynamics24 Apr 2024
- 2024SineNet: Learning Temporal Dynamics in Time-Dependent Partial Differential Equations28 Mar 2024
- 2024JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework7 Mar 2024
- 2024Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modeling29 Feb 2024
50 newest of 131 All 131 (JSON)
- 2023Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulationcited by 65
- 2024MotionCraft: Physics-based Zero-Shot Video Generationcited by 46
- 2024Wavelet Diffusion Neural Operatorcited by 28
- 2023Solving Inverse Physics Problems with Score Matchingcited by 15
- 2024CL-DiffPhyCon: Closed-loop Diffusion Control of Complex Physical Systemscited by 8
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Controlcited by 3
- 2026Accelerated Sequential Flow Matching: A Bayesian Filtering Perspectivecited by 1
- 2026Accelerated Sequential Flow Matching: A Bayesian Filtering Perspective5 Feb 2026
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Control4 Feb 2025
- 2024Wavelet Diffusion Neural Operator6 Dec 2024
- 2024CL-DiffPhyCon: Closed-loop Diffusion Control of Complex Physical Systems31 Jul 2024
- 2024MotionCraft: Physics-based Zero-Shot Video Generation22 May 2024
- 2023Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation4 Sep 2023
- 2023Solving Inverse Physics Problems with Score Matching24 Jan 2023
- 2022Liquid Structural State-Space Modelscited by 134
- 2021PlasticineLab: A Soft-Body Manipulation Benchmark with Differentiable Physicscited by 100
- 2023DeepZero: Scaling up Zeroth-Order Optimization for Deep Model Trainingcited by 76
- 2021DiffCloth: Differentiable Cloth Simulation with Dry Frictional Contactcited by 64
- 2022When Physics Meets Machine Learning: A Survey of Physics-Informed Machine Learningcited by 64
- 2021DiffPD: Differentiable Projective Dynamicscited by 62
- 2021Fast and Feature-Complete Differentiable Physics for Articulated Rigid Bodies with Contactcited by 50
- 2022NeuPhysics: Editable Neural Geometry and Physics from Monocular Videoscited by 50
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 43
- 2024ConFIG: Towards Conflict-free Training of Physics Informed Neural Networkscited by 43
- 2021NTopo: Mesh-free Topology Optimization using Implicit Neural Representationscited by 41
- 2021Solving PDE-constrained Control Problems Using Operator Learningcited by 41
12 most cited of 116 All 116 (JSON)
- 2026Physics-Informed Method of Group Data Handling: Adaptive Construction of Functional Representations with an Application to the Navier-Stokes Equations30 Sep 2026
- 2026Gradient Surgery for Physics-Informed Neural Networks25 Sep 2026
- 2026LaGSplat: Inferring Physics-Governed Interactive Simulation from Monocular Video Using Latent Lagrangian Gaussian Splatting17 Aug 2026
- 2026Differentiate the Solver, Not the Equation: Reverse-Sweep Adjoints for Block Implicit Simulation9 Aug 2026
- 2026Large language models for partial differential equation workflows4 Aug 2026
- 2026Toward Goal-Agnostic Joint-Embedding Predictive Control of Partial Differential Equations21 Jul 2026
- 2026Physics-enhanced reinforcement learning for real-time optimal control of dynamical systems17 Jul 2026
- 2026Learning Physics-Guided Residual Dynamics for Deformable Object Simulation15 Jul 2026
- 2026Integrating Mechanistic and Data-Driven Models for Neurological Disorders through Differentiable Programming4 Jun 2026
- 2026Deep Learning-based Algebraic Reynolds Stress Closures for RANS Simulations of Turbulent Flows25 May 2026
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospects16 May 2026
- 2026A solver-in-the-loop framework for end-to-end differentiable coastal hydrodynamics8 Apr 2026
12 newest of 116 All 116 (JSON)