Incorporating Symmetry into Deep Dynamics Models for Improved Generalization
Not on the map. Why?
Abstract (arXiv)
Recent work has shown deep learning can accelerate the prediction of physical dynamics relative to numerical solvers. However, limited physical accuracy and an inability to generalize under distributional shift limit its applicability to the real world. We propose to improve accuracy and generalization by incorporating symmetries into convolutional neural networks. Specifically, we employ a variety of methods each tailored to enforce a different symmetry. Our models are both theoretically and experimentally robust to distributional shift by symmetry group transformations and enjoy favorable sample complexity. We demonstrate the advantage of our approach on a variety of physical dynamics including Rayleigh B'enard convection and real-world ocean currents and temperatures. Compared with image or text applications, our work is a significant step towards applying equivariant neural networks to high-dimensional systems with complex dynamics. We open-source our simulation, data, and code at urlhttps://github.com/Rose-STL-Lab/Equivariant-Net.
Key facts. Cited by 126 papers in the map (data through 14 Oct 2026). First posted on arXiv 8 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 16; 2022: 23; 2023: 28; 2024: 29; 2025: 23; 2026: 7.
126 papers.
Sort:
By area:
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 197
- 2021Enhancing Computational Fluid Dynamics with Machine Learningcited by 138
- 2021Scalars are universal: Equivariant machine learning, structured like classical physicscited by 127
- 2022Approximately Equivariant Networks for Imperfectly Symmetric Dynamicscited by 97
- 2022Clifford Neural Layers for PDE Modelingcited by 92
- 2021Automatic Symmetry Discovery with Lie Algebra Convolutional Networkcited by 83
- 2022Lie Point Symmetry Data Augmentation for Neural PDE Solverscited by 77
- 2023Machine learning with data assimilation and uncertainty quantification for dynamical systems: a reviewcited by 75
- 2024Equivariant Diffusion Policycited by 66
- 2022DEVIANT: Depth EquiVarIAnt NeTwork for Monocular 3D Object Detectioncited by 48
- 2023Group Equivariant Fourier Neural Operators for Partial Differential Equationscited by 45
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 44
38 more
- 2023Generative Adversarial Symmetry Discoverycited by 42
- 2021Meta-Learning Dynamics Forecasting Using Task Inferencecited by 38
- 2021Hamiltonian-based Neural ODE Networks on the SE(3) Manifold For Dynamics Learning and Controlcited by 38
- 2022Physics-Embedded Neural Networks: Graph Neural PDE Solvers with Mixed Boundary Conditionscited by 33
- 2022The Surprising Effectiveness of Equivariant Models in Domains with Latent Symmetrycited by 33
- 2022Guaranteed Conservation of Momentum for Learning Particle-based Fluid Dynamicscited by 30
- 2023Self-Supervised Learning with Lie Symmetries for Partial Differential Equationscited by 29
- 2023Latent Space Symmetry Discoverycited by 29
- 2023The Exact Sample Complexity Gain from Invariances for Kernel Regressioncited by 28
- 2024Symmetry Considerations for Learning Task Symmetric Robot Policiescited by 28
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Surveycited by 28
- 2024Zero-shot forecasting of chaotic systemscited by 28
- 2024Clifford-Steerable Convolutional Neural Networkscited by 25
- 2024Lie Algebra Canonicalization: Equivariant Neural Operators under arbitrary Lie Groupscited by 24
- 2023Learning Layer-wise Equivariances Automatically using Gradientscited by 23
- 2023Machine Learning for Partial Differential Equationscited by 21
- 2021Combining Physics and Deep Learning to learn Continuous-Time Dynamics Modelscited by 20
- 2023Discovering Symmetry Breaking in Physical Systems with Relaxed Group Convolutioncited by 20
- 2023A General Theory of Correct, Incorrect, and Extrinsic Equivariancecited by 19
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 19
- 2024Equivariant Reinforcement Learning under Partial Observabilitycited by 18
- 2022Dimensionless machine learning: Imposing exact units equivariancecited by 17
- 2022Data Augmentation vs. Equivariant Networks: A Theory of Generalization on Dynamics Forecastingcited by 17
- 2022Set Norm and Equivariant Skip Connections: Putting the Deep in Deep Setscited by 17
- 2024Graph Neural PDE Solvers with Conservation and Similarity-Equivariancecited by 17
- 2024Learning Infinitesimal Generators of Continuous Symmetries from Datacited by 17
- 2024Adversarial Learning for Neural PDE Solvers with Sparse Datacited by 16
- 2024Symmetry-Informed Governing Equation Discoverycited by 15
- 2024Knowledge-guided Machine Learning: Current Trends and Future Prospectscited by 14
- 2021Applying Machine Learning to Study Fluid Mechanicscited by 13
- 2022Machine learning and invariant theorycited by 13
- 2024P2C2Net: PDE-Preserved Coarse Correction Network for efficient prediction of spatiotemporal dynamicscited by 13
- 2021Neural Pharmacodynamic State Space Modelingcited by 12
- 2021Physics-constrained Deep Learning for Robust Inverse ECG Modelingcited by 12
- 2022Deep learning-enhanced ensemble-based data assimilation for high-dimensional nonlinear dynamical systemscited by 12
- 2023Learning Lagrangian Fluid Mechanics with E(3)-Equivariant Graph Neural Networkscited by 12
- 2023On Robot Grasp Learning Using Equivariant Modelscited by 12
- 2023Physics-aware Machine Learning Revolutionizes Scientific Paradigm for Machine Learning and Process-based Hydrologycited by 11
50 most cited of 126 All 126 (JSON)
- 2026Dimension-Free Scaling Laws for Size Generalization: An Exponential Separation13 Oct 2026
- 2026Learning Operators for Symmetry Discovery: From Lie Groups to Local Symmetry Geometry12 Oct 2026
- 2026Equivariant Visual-Tactile Diffusion Policy for Contact-Rich Manipulation2 Oct 2026
- 2026LieStoNet: Learning Lie Symmetries from Spatiotemporal Data for Stochastic Dynamical Systems3 Aug 2026
- 2026GENERIC-FNO: Embedding Energy Conservation and Entropy Production into Fourier Neural Operators6 Jun 2026
- 2026PACE-FNO: Physics-Aligned Canonical Equivariance for Fourier Neural Operators18 May 2026
- 2026Smoothness Errors in Dynamics Models and How to Avoid Them5 Feb 2026
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetry5 Dec 2025
- 2025Symmetry and Generalisation in Neural Approximations of Renormalisation Transformations18 Oct 2025
- 2025Conditional Clifford-Steerable CNNs for PDE Modeling15 Oct 2025
- 2025Can Data-Driven Dynamics Reveal Hidden Physics? There Is A Need for Interpretable Neural Operators3 Oct 2025
- 2025Explicit Discovery of Nonlinear Symmetries from Dynamic Data2 Oct 2025
38 more
- 2025Towards Generalizable PDE Dynamics Forecasting via Physics-Guided Invariant Learning29 Sep 2025
- 2025Scale-Consistent Learning for Partial Differential Equations24 Jul 2025
- 2025Flow Equivariant Recurrent Neural Networks20 Jul 2025
- 2025A Scalable and Quantum-Accurate Foundation Model for Biomolecular Force Field via Linearly Tensorized Quadrangle Attention1 Jul 2025
- 2025Learnable-Differentiable Finite Volume Solver for Accelerated Simulation of Flows23 Jun 2025
- 2025SlotPi: Physics-informed Object-centric Reasoning Models12 Jun 2025
- 2025Geometric and Physical Constraints Synergistically Enhance Neural PDE Surrogates5 Jun 2025
- 2025Learning dissection trajectories from expert surgical videos via imitation learning with equivariant diffusion5 Jun 2025
- 2025FD-Bench: A Modular and Fair Benchmark for Data-driven Fluid Simulation25 May 2025
- 2025Governing Equation Discovery from Data Based on Differential Invariants24 May 2025
- 2025Surrogate Modeling of 3D Rayleigh-Benard Convection with Equivariant Autoencoders19 May 2025
- 2025Hamiltonian Neural PDE Solvers through Functional Approximation19 May 2025
- 2025Discovering Symbolic Differential Equations with Symmetry Invariants17 May 2025
- 2025Morphologically Symmetric Reinforcement Learning for Ambidextrous Bimanual Manipulation8 May 2025
- 2025Harnessing Equivariance: Modeling Turbulence with Graph Neural Networks10 Apr 2025
- 2025Generalization Bounds for Equivariant Networks on Markov Data1 Mar 2025
- 2025Generalized Lie Symmetries in Physics-Informed Neural Operators1 Feb 2025
- 2025MultiPDENet: PDE-embedded Learning with Multi-time-stepping for Accelerated Flow Simulation27 Jan 2025
- 2024Morphological-Symmetry-Equivariant Heterogeneous Graph Neural Network for Robotic Dynamics Learning2 Dec 2024
- 2024Approximate Equivariance in Reinforcement Learning6 Nov 2024
- 2024P2C2Net: PDE-Preserved Coarse Correction Network for efficient prediction of spatiotemporal dynamics29 Oct 2024
- 2024Learning Infinitesimal Generators of Continuous Symmetries from Data29 Oct 2024
- 2024On the VC dimension of deep group convolutional neural networks21 Oct 2024
- 2024Unitary convolutions for learning on graphs and groups7 Oct 2024
- 2024Lie Algebra Canonicalization: Equivariant Neural Operators under arbitrary Lie Groups3 Oct 2024
- 2024Segment as You Wish -- Free-Form Language-Based Segmentation for Medical Images2 Oct 2024
- 2024Incorporating Arbitrary Matrix Group Equivariance into KANs1 Oct 2024
- 2024Zero-shot forecasting of chaotic systems24 Sep 2024
- 2024Symmetry constrained neural networks for detection and localization of damage in metal plates9 Sep 2024
- 2024Adversarial Learning for Neural PDE Solvers with Sparse Data4 Sep 2024
- 2024Equivariant Reinforcement Learning under Partial Observability26 Aug 2024
- 2024Recent Advances on Machine Learning for Computational Fluid Dynamics: A Survey22 Aug 2024
- 2024Learning production functions for supply chains with graph neural networks26 Jul 2024
- 2024Physics-embedded Fourier Neural Network for Partial Differential Equations15 Jul 2024
- 2024Deep deterministic policy gradient with symmetric data augmentation for lateral attitude tracking control of a fixed-wing aircraft13 Jul 2024
- 2024Equivariant Diffusion Policy1 Jul 2024
- 2024Science-Informed Design of Deep Learning With Applications to Wireless Systems: A Tutorial29 Jun 2024
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency11 Jun 2024
50 newest of 126 All 126 (JSON)
- 2024Equivariant Reinforcement Learning under Partial Observabilitycited by 18
- 2024Approximate Equivariance in Reinforcement Learningcited by 9
- 2025Morphologically Symmetric Reinforcement Learning for Ambidextrous Bimanual Manipulationcited by 5
- 2024Deep deterministic policy gradient with symmetric data augmentation for lateral attitude tracking control of a fixed-wing aircraftcited by 1
- 2025Morphologically Symmetric Reinforcement Learning for Ambidextrous Bimanual Manipulation8 May 2025
- 2024Approximate Equivariance in Reinforcement Learning6 Nov 2024
- 2024Equivariant Reinforcement Learning under Partial Observability26 Aug 2024
- 2024Deep deterministic policy gradient with symmetric data augmentation for lateral attitude tracking control of a fixed-wing aircraft13 Jul 2024
- 2025SlotPi: Physics-informed Object-centric Reasoning Modelscited by 5
- 2025SlotPi: Physics-informed Object-centric Reasoning Models12 Jun 2025
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 197
- 2021Enhancing Computational Fluid Dynamics with Machine Learningcited by 138
- 2021Scalars are universal: Equivariant machine learning, structured like classical physicscited by 127
- 2022Approximately Equivariant Networks for Imperfectly Symmetric Dynamicscited by 97
- 2022Clifford Neural Layers for PDE Modelingcited by 92
- 2021Automatic Symmetry Discovery with Lie Algebra Convolutional Networkcited by 83
- 2022Lie Point Symmetry Data Augmentation for Neural PDE Solverscited by 77
- 2023Machine learning with data assimilation and uncertainty quantification for dynamical systems: a reviewcited by 75
- 2022DEVIANT: Depth EquiVarIAnt NeTwork for Monocular 3D Object Detectioncited by 48
- 2023Group Equivariant Fourier Neural Operators for Partial Differential Equationscited by 45
- 2021Physics-Guided Deep Learning for Dynamical Systems: A Surveycited by 44
- 2023Generative Adversarial Symmetry Discoverycited by 42
12 most cited of 116 All 116 (JSON)
- 2026Dimension-Free Scaling Laws for Size Generalization: An Exponential Separation13 Oct 2026
- 2026Learning Operators for Symmetry Discovery: From Lie Groups to Local Symmetry Geometry12 Oct 2026
- 2026LieStoNet: Learning Lie Symmetries from Spatiotemporal Data for Stochastic Dynamical Systems3 Aug 2026
- 2026GENERIC-FNO: Embedding Energy Conservation and Entropy Production into Fourier Neural Operators6 Jun 2026
- 2026PACE-FNO: Physics-Aligned Canonical Equivariance for Fourier Neural Operators18 May 2026
- 2026Smoothness Errors in Dynamics Models and How to Avoid Them5 Feb 2026
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetry5 Dec 2025
- 2025Symmetry and Generalisation in Neural Approximations of Renormalisation Transformations18 Oct 2025
- 2025Conditional Clifford-Steerable CNNs for PDE Modeling15 Oct 2025
- 2025Can Data-Driven Dynamics Reveal Hidden Physics? There Is A Need for Interpretable Neural Operators3 Oct 2025
- 2025Explicit Discovery of Nonlinear Symmetries from Dynamic Data2 Oct 2025
- 2025Towards Generalizable PDE Dynamics Forecasting via Physics-Guided Invariant Learning29 Sep 2025
12 newest of 116 All 116 (JSON)