Graph networks as learnable physics engines for inference and control
Not on the map. Why?
Abstract (arXiv)
Understanding and interacting with everyday physical scenes requires rich knowledge about the structure of the world, represented either implicitly in a value or policy function, or explicitly in a transition model. Here we introduce a new class of learnable models--based on graph networks--which implement an inductive bias for object- and relation-centric representations of complex, dynamical systems. Our results show that as a forward model, our approach supports accurate predictions from real and simulated data, and surprisingly strong and efficient generalization, across eight distinct physical systems which we varied parametrically and structurally. We also found that our inference model can perform system identification. Our models are also differentiable, and support online planning via gradient-based trajectory optimization, as well as offline policy optimization. Our framework offers new opportunities for harnessing and exploiting rich knowledge about the world, and takes a key step toward building machines with more human-like representations of the world.
Key facts. Cited by 242 papers in the map (data through 7 Oct 2026). First posted on arXiv 4 Jun 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 58; 2022: 67; 2023: 48; 2024: 25; 2025: 22; 2026: 21.
242 papers.
Sort:
By area:
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2023VoxPoser: Composable 3D Value Maps for Robotic Manipulation with Language Modelscited by 733
- 2024ReKep: Spatio-Temporal Reasoning of Relational Keypoint Constraints for Robotic Manipulationcited by 305
- 2022GraphCast: Learning skillful medium-range global weather forecastingcited by 261
- 2024PhysGen: Rigid-Body Physics-Grounded Image-to-Video Generationcited by 169
- 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Modelscited by 146
- 2021Physion: Evaluating Physical Prediction from Vision in Humans and Machinescited by 145
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2022Pre-training via Denoising for Molecular Property Predictioncited by 124
- 2022Revisiting Over-smoothing and Over-squashing Using Ollivier-Ricci Curvaturecited by 96
- 2021Learned Coarse Models for Efficient Turbulence Simulationcited by 90
- 2021Size-Invariant Graph Representations for Graph Classification Extrapolationscited by 82
38 more
- 2022Predicting Physics in Mesh-reduced Space with Temporal Attentioncited by 82
- 2022MetaMorph: Learning Universal Controllers with Transformerscited by 78
- 2022AirfRANS: High Fidelity Computational Fluid Dynamics Dataset for Approximating Reynolds-Averaged Navier-Stokes Solutionscited by 77
- 2021A Survey on Interpretable Reinforcement Learningcited by 70
- 2021Port-Hamiltonian Neural Networks for Learning Explicit Time-Dependent Dynamical Systemscited by 66
- 2022When Physics Meets Machine Learning: A Survey of Physics-Informed Machine Learningcited by 64
- 2022Learning Multi-Object Dynamics with Compositional Neural Radiance Fieldscited by 63
- 2022A Generalist Neural Algorithmic Learnercited by 58
- 2022GraphFramEx: Towards Systematic Evaluation of Explainability Methods for Graph Neural Networkscited by 57
- 2022Multimodal learning with graphscited by 54
- 2021Learning to schedule job-shop problems: Representation and policy learning using graph neural network and reinforcement learningcited by 52
- 2022Graph Inverse Reinforcement Learning from Diverse Videoscited by 45
- 2021Simple GNN Regularisation for 3D Molecular Property Prediction & Beyondcited by 44
- 2022DiffSkill: Skill Abstraction from Differentiable Physics for Deformable Object Manipulations with Toolscited by 44
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 43
- 2022Efficient Learning of Mesh-Based Physical Simulation with BSMS-GNNcited by 43
- 2022Learning rigid dynamics with face interaction graph networkscited by 43
- 2021Simulating Continuum Mechanics with Multi-Scale Graph Neural Networkscited by 39
- 2021Efficient Differentiable Simulation of Articulated Bodiescited by 38
- 2022Learning Physical Dynamics with Subequivariant Graph Neural Networkscited by 38
- 2021Hamiltonian-based Neural ODE Networks on the SE(3) Manifold For Dynamics Learning and Controlcited by 37
- 2022Physical Design using Differentiable Learned Simulatorscited by 37
- 2023Learning Neural PDE Solvers with Parameter-Guided Channel Attentioncited by 37
- 2022SizeShiftReg: a Regularization Method for Improving Size-Generalization in Graph Neural Networkscited by 36
- 2022Distributed Graph Neural Network Training: A Surveycited by 32
- 2022Graph Lifelong Learning: A Surveycited by 30
- 2021Constructing Neural Network-Based Models for Simulating Dynamical Systemscited by 29
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2022MAgNet: Mesh Agnostic Neural PDE Solvercited by 28
- 2023Physics-Informed Machine Learning for Modeling and Control of Dynamical Systemscited by 28
- 2020Differentiable Physics Models for Real-world Offline Model-based Reinforcement Learningcited by 27
- 2021Constraint-based graph network simulatorcited by 27
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixelscited by 27
- 2023LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suitecited by 26
- 2022AnyMorph: Learning Transferable Polices By Inferring Agent Morphologycited by 25
- 2021An Extensible Benchmark Suite for Learning to Simulate Physical Systemscited by 24
- 2023A Generalist Dynamics Model for Controlcited by 24
- 2024HAMLET: Graph Transformer Neural Operator for Partial Differential Equationscited by 24
50 most cited of 242 All 242 (JSON)
- 2026ActionGround: Training-Free Runtime Refinement of Frozen VLA Policies27 Sep 2026
- 2026Comprehensive reconstruction of collider events with hypergraph representation learning and graph-conditioned diffusion16 Sep 2026
- 2026When does a spectral prior help graph learning? Connectivity-loss estimation under road-network disruptions10 Sep 2026
- 2026TRACE: A spatiotemporal contact memory graph network simulator for granular dynamics2 Sep 2026
- 2026MeshPriorDiT: Hierarchical Modeling for Action-Conditioned Cloth Dynamics27 Aug 2026
- 2026Distillation of Foundation Models for Time-dependent PDEs12 Aug 2026
- 2026A Structural Dynamics Graph World Model: Unified Modeling, Constrained Rollout, and Interpretable Calibration9 Aug 2026
- 2026Schreier-Coset Graph Rewiring29 Jul 2026
- 2026Planning as Emergent Behavior in Reinforcement Learning with Relational Hidden States20 Jul 2026
- 2026A multi-scale feature enhanced graph neural network for fluid dynamics prediction in complex geometries13 Jul 2026
- 2026Understanding Rollout Error in Graph World Models26 Jun 2026
- 2026Medical world models: representing medical states, modelling clinical dynamics and guiding intervention policies15 Jun 2026
38 more
- 2026PhysVLA: Towards Physically-Grounded VLA for Embodied Robotic Manipulation11 Jun 2026
- 2026OnlyDense: Reduced-Order Modeling for Lagrangian simulation8 Jun 2026
- 2026Cellular Sheaf Neural Operators for Structure-Preserving Surrogate Modeling of Constrained PDEs31 May 2026
- 2026MARVEL: Universal Murray's Law-informed Vessel Tree Segmentation and Topology Estimation25 May 2026
- 2026Learning Altruistic Collaboration in Heterogeneous Multi-Team Systems20 May 2026
- 2026Deep Wave Network for Modeling Multi-Scale Physical Dynamics5 May 2026
- 2026Momentum-Conserving Graph Neural Networks for Deformable Objects28 Apr 2026
- 2026Evolving Embodied Intelligence: Graph Neural Network--Driven Co-Design of Morphology and Control in Soft Robotics20 Mar 2026
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion2 Feb 2026
- 2025Neural Modular Physics for Elastic Simulation17 Dec 2025
- 2025OXE-AugE: A Large-Scale Robot Augmentation of OXE for Scaling Cross-Embodiment Policy Learning15 Dec 2025
- 2025Parameter-Efficient Conditioning for Material Generalization in Graph-Based Simulators7 Nov 2025
- 2025Context-aware Learned Mesh-based Simulation via Trajectory-Level Meta-Learning7 Nov 2025
- 2025Graph-Based Learning of Free Surface Dynamics in Generalized Newtonian Fluids using Smoothed Particle Hydrodynamics29 Sep 2025
- 2025Learning Contact Dynamics through Touching: Action-conditional Graph Neural Networks for Robotic Peg Insertion15 Sep 2025
- 2025Reservoir Predictive Path Integral Control for Unknown Nonlinear Dynamics4 Sep 2025
- 2025Automated discovery of finite volume schemes using Graph Neural Networks26 Aug 2025
- 2025Neural Robot Dynamics21 Aug 2025
- 2025Beyond Fixed Morphologies: Learning Graph Policies with Trust Region Compensation in Variable Action Spaces16 Aug 2025
- 2025Recurrent U-Net-Based Graph Neural Network (RUGNN) for Accurate Deformation Predictions in Sheet Material Forming10 Jul 2025
- 2025Mitigating Over-Squashing in Graph Neural Networks by Spectrum-Preserving Sparsification19 Jun 2025
- 2025GITO: Graph-Informed Transformer Operator for Learning Complex Partial Differential Equations16 Jun 2025
- 2025EqCollide: Equivariant and Collision-Aware Deformable Objects Neural Simulator6 Jun 2025
- 2025Hierarchical-embedding autoencoder with a predictor (HEAP) as efficient architecture for learning long-term evolution of complex multi-scale physical systems24 May 2025
- 2025Plain Transformers Can be Powerful Graph Learners17 Apr 2025
- 2025PowerGNN: A Topology-Aware Graph Neural Network for Electricity Grids26 Mar 2025
- 2025Learning High-Fidelity Robot Self-Model with Articulated 3D Gaussian Splatting7 Mar 2025
- 2025Learning Effective Dynamics across Spatio-Temporal Scales of Complex Flows11 Feb 2025
- 2025Sparse identification of nonlinear dynamics and Koopman operators with Shallow Recurrent Decoder Networks23 Jan 2025
- 2025ClusterViG: Efficient Globally Aware Vision GNNs via Image Partitioning18 Jan 2025
- 2025Data-Driven Radio Propagation Modeling using Graph Neural Networks8 Jan 2025
- 2024Adversarial Training for Graph Neural Networks via Graph Subspace Energy Optimization25 Dec 2024
- 2024Morphological-Symmetry-Equivariant Heterogeneous Graph Neural Network for Robotic Dynamics Learning2 Dec 2024
- 2024A Self-Explainable Heterogeneous GNN for Relational Deep Learning30 Nov 2024
- 2024Attribute-Enhanced Similarity Ranking for Sparse Link Prediction29 Nov 2024
- 2024Physics Encoded Blocks in Residual Neural Network Architectures for Digital Twin Models18 Nov 2024
- 2024Mean flow data assimilation using physics-constrained Graph Neural Networks14 Nov 2024
- 2024Learned Neural Physics Simulation for Articulated 3D Human Pose Reconstruction15 Oct 2024
50 newest of 242 All 242 (JSON)
- 2023A Generalist Dynamics Model for Controlcited by 24
- 2023Understanding Physical Dynamics with Counterfactual World Modelingcited by 11
- 2022Factored World Models for Zero-Shot Generalization in Robotic Manipulationcited by 6
- 2024Slot Structured World Modelscited by 5
- 2026Medical world models: representing medical states, modelling clinical dynamics and guiding intervention policiescited by 3
- 2026Understanding Rollout Error in Graph World Modelscited by 3
- 2026A Structural Dynamics Graph World Model: Unified Modeling, Constrained Rollout, and Interpretable Calibrationcited by 2
- 2026A Structural Dynamics Graph World Model: Unified Modeling, Constrained Rollout, and Interpretable Calibration9 Aug 2026
- 2026Understanding Rollout Error in Graph World Models26 Jun 2026
- 2026Medical world models: representing medical states, modelling clinical dynamics and guiding intervention policies15 Jun 2026
- 2024Slot Structured World Models8 Jan 2024
- 2023Understanding Physical Dynamics with Counterfactual World Modeling11 Dec 2023
- 2023A Generalist Dynamics Model for Control18 May 2023
- 2022Factored World Models for Zero-Shot Generalization in Robotic Manipulation10 Feb 2022
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2023VoxPoser: Composable 3D Value Maps for Robotic Manipulation with Language Modelscited by 733
- 2022GraphCast: Learning skillful medium-range global weather forecastingcited by 261
- 2024PhysGen: Rigid-Body Physics-Grounded Image-to-Video Generationcited by 169
- 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Modelscited by 146
- 2021Physion: Evaluating Physical Prediction from Vision in Humans and Machinescited by 145
- 2022Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphscited by 144
- 2022Pre-training via Denoising for Molecular Property Predictioncited by 124
- 2022Revisiting Over-smoothing and Over-squashing Using Ollivier-Ricci Curvaturecited by 96
- 2021Learned Coarse Models for Efficient Turbulence Simulationcited by 90
- 2021Size-Invariant Graph Representations for Graph Classification Extrapolationscited by 82
- 2022Predicting Physics in Mesh-reduced Space with Temporal Attentioncited by 82
12 most cited of 225 All 225 (JSON)
- 2026Comprehensive reconstruction of collider events with hypergraph representation learning and graph-conditioned diffusion16 Sep 2026
- 2026When does a spectral prior help graph learning? Connectivity-loss estimation under road-network disruptions10 Sep 2026
- 2026TRACE: A spatiotemporal contact memory graph network simulator for granular dynamics2 Sep 2026
- 2026Distillation of Foundation Models for Time-dependent PDEs12 Aug 2026
- 2026Schreier-Coset Graph Rewiring29 Jul 2026
- 2026Planning as Emergent Behavior in Reinforcement Learning with Relational Hidden States20 Jul 2026
- 2026A multi-scale feature enhanced graph neural network for fluid dynamics prediction in complex geometries13 Jul 2026
- 2026OnlyDense: Reduced-Order Modeling for Lagrangian simulation8 Jun 2026
- 2026Cellular Sheaf Neural Operators for Structure-Preserving Surrogate Modeling of Constrained PDEs31 May 2026
- 2026MARVEL: Universal Murray's Law-informed Vessel Tree Segmentation and Topology Estimation25 May 2026
- 2026Learning Altruistic Collaboration in Heterogeneous Multi-Team Systems20 May 2026
- 2026Deep Wave Network for Modeling Multi-Scale Physical Dynamics5 May 2026
12 newest of 225 All 225 (JSON)