Lie Point Symmetry Data Augmentation for Neural PDE Solvers
Not on the map. Why?
Abstract (arXiv)
Neural networks are increasingly being used to solve partial differential equations (PDEs), replacing slower numerical solvers. However, a critical issue is that neural PDE solvers require high-quality ground truth data, which usually must come from the very solvers they are designed to replace. Thus, we are presented with a proverbial chicken-and-egg problem. In this paper, we present a method, which can partially alleviate this problem, by improving neural PDE solver sample complexity -- Lie point symmetry data augmentation (LPSDA). In the context of PDEs, it turns out that we are able to quantitatively derive an exhaustive list of data transformations, based on the Lie point symmetry group of the PDEs in question, something not possible in other application areas. We present this framework and demonstrate how it can easily be deployed to improve neural PDE solver sample complexity by an order of magnitude.
Key facts. Cited by 76 papers in the map (data through 7 Oct 2026). Builds on 25 of its 63 references that are in the map. First posted on arXiv 15 Feb 2022.
Builds on
25 of 63 references.
References · 25
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2015Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)cited by 1,061
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
- 2019Hamiltonian Neural Networkscited by 702
- 2017DGM: A deep learning algorithm for solving partial differential equationscited by 674
- 2021Machine learning accelerated computational fluid dynamicscited by 461
- 2020Lagrangian Neural Networkscited by 388
- 2022Message Passing Neural PDE Solverscited by 368
13 more
- 2020Generalizing Convolutional Neural Networks for Equivariance to Lie Groups on Arbitrary Continuous Datacited by 248
- 2018Deep Hidden Physics Models: Deep Learning of Nonlinear Partial Differential Equationscited by 233
- 2020Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solverscited by 201
- 2018Bayesian Deep Convolutional Encoder-Decoder Networks for Surrogate Modeling and Uncertainty Quantificationcited by 199
- 2019Prediction of Aerodynamic Flow Fields Using Convolutional Neural Networkscited by 144
- 2020Incorporating Symmetry into Deep Dynamics Models for Improved Generalizationcited by 123
- 2019Hamiltonian Graph Networks with ODE Integratorscited by 107
- 2020Simplifying Hamiltonian and Lagrangian Neural Networks via Explicit Constraintscited by 102
- 2019Learning to Optimize Multigrid PDE Solverscited by 79
- 2019Learning Neural PDE Solvers with Convergence Guaranteescited by 77
- 2021Physics-based Deep Learningcited by 67
- 2020A physics-informed operator regression framework for extracting data-driven continuum modelscited by 60
- 2019Deep Eikonal Solverscited by 9
Built on by
Citing papers per year: 2022: 7; 2023: 14; 2024: 20; 2025: 20; 2026: 15.
76 papers.
Sort:
By area:
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 205
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
- 2023Scalable Transformer for PDE Surrogate Modelingcited by 123
- 2022Clifford Neural Layers for PDE Modelingcited by 91
- 2022AirfRANS: High Fidelity Computational Fluid Dynamics Dataset for Approximating Reynolds-Averaged Navier-Stokes Solutionscited by 77
- 2024On conditional diffusion models for PDE simulationscited by 56
- 2023Training neural operators to preserve invariant measures of chaotic attractorscited by 43
- 2023Group Equivariant Fourier Neural Operators for Partial Differential Equationscited by 42
- 2023Geometric Clifford Algebra Networkscited by 37
- 2024Data-Efficient Operator Learning via Unsupervised Pretraining and In-Context Learningcited by 32
- 2022Guaranteed Conservation of Momentum for Learning Particle-based Fluid Dynamicscited by 30
- 2023Self-Supervised Learning with Lie Symmetries for Partial Differential Equationscited by 29
38 more
- 2023Lie Point Symmetry and Physics Informed Networkscited by 24
- 2024HAMLET: Graph Transformer Neural Operator for Partial Differential Equationscited by 24
- 2024Lie Algebra Canonicalization: Equivariant Neural Operators under arbitrary Lie Groupscited by 23
- 2024Active Learning for Neural PDE Solverscited by 22
- 2023Machine Learning for Partial Differential Equationscited by 20
- 2024Learning Infinitesimal Generators of Continuous Symmetries from Datacited by 17
- 2024Mechanistic Neural Networks for Scientific Machine Learningcited by 16
- 2024Adversarial Learning for Neural PDE Solvers with Sparse Datacited by 16
- 2024Symmetry-Informed Governing Equation Discoverycited by 14
- 2024Strategies for Pretraining Neural Operatorscited by 14
- 2024Zebra: In-Context Generative Pretraining for Solving Parametric PDEscited by 14
- 2024Masked Autoencoders are PDE Learnerscited by 11
- 2024Linearization Turns Neural Operators into Function-Valued Gaussian Processescited by 11
- 2025Principled Approaches for Extending Neural Architectures to Function Spaces for Operator Learningcited by 11
- 2023Understanding the Expressivity and Trainability of Fourier Neural Operator: A Mean-Field Perspectivecited by 10
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogatescited by 8
- 2023Invariant preservation in machine learned PDE solvers via error correctioncited by 7
- 2024GEPS: Boosting Generalization in Parametric PDE Neural Solvers through Adaptive Conditioningcited by 7
- 2023MetaPhysiCa: OOD Robustness in Physics-informed Machine Learningcited by 6
- 2025Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solvingcited by 6
- 2025Discovering Symbolic Differential Equations with Symmetry Invariantscited by 6
- 2022Physics-guided Data Augmentation for Learning the Solution Operator of Linear Differential Equationscited by 5
- 2024Latent Neural PDE Solver: a reduced-order modelling framework for partial differential equationscited by 5
- 2024Scalable Mechanistic Neural Networks for Differential Equations and Machine Learningcited by 5
- 2025BARNN: A Bayesian Autoregressive and Recurrent Neural Networkcited by 5
- 2025Geometric and Physical Constraints Synergistically Enhance Neural PDE Surrogatescited by 5
- 2025Explicit Discovery of Nonlinear Symmetries from Dynamic Datacited by 5
- 2025Latent Mixture of Symmetries for Sample-Efficient Dynamic Learningcited by 5
- 2023G-Signatures: Global Graph Propagation With Randomized Signaturescited by 4
- 2024Learning Semilinear Neural Operators : A Unified Recursive Framework For Prediction And Data Assimilationcited by 4
- 2025Mechanistic PDE Networks for Discovery of Governing Equationscited by 4
- 2025Governing Equation Discovery from Data Based on Differential Invariantscited by 4
- 2025Active Learning with Selective Time-Step Acquisition for PDEscited by 4
- 2026Explainable AI: Learning from the Learnerscited by 4
- 2025MultiPDENet: PDE-embedded Learning with Multi-time-stepping for Accelerated Flow Simulationcited by 3
- 2025Overtone: Cyclic Patch Modulation for Clean, Efficient, and Flexible Physics Emulatorscited by 3
- 2025Adaptive Canonicalization with Application to Invariant Anisotropic Geometric Networkscited by 3
- 2022A Feedforward Unitary Equivariant Neural Networkcited by 2
50 most cited of 76 All 76 (JSON)
- 2026LieDiscover: Adaptive Symbolic Library Construction for Explicit Open-form Symmetry Discovery27 Sep 2026
- 2026Unsupervised Adaptation of PDE Foundation Models7 Aug 2026
- 2026LieStoNet: Learning Lie Symmetries from Spatiotemporal Data for Stochastic Dynamical Systems3 Aug 2026
- 2026On the post-hoc Evaluation of PDE Discovery: A Multifaceted Challenge of Scientific Advancement26 Jul 2026
- 2026How Much Memory Do We Need? Adaptive Memory Gate for Neural Operators11 Jun 2026
- 2026GENERIC-FNO: Embedding Energy Conservation and Entropy Production into Fourier Neural Operators6 Jun 2026
- 2026PDEInvBench: A Comprehensive Dataset and Design Space Exploration of Neural Networks for PDE Inverse Problems25 May 2026
- 2026Data-Efficient Neural Operator Training via Physics-Based Active Learning20 May 2026
- 2026PACE-FNO: Physics-Aligned Canonical Equivariance for Fourier Neural Operators18 May 2026
- 2026AdamFLIP: Adaptive Momentum Feedback Linearization Optimization for Hard Constrained PINN Training8 May 2026
- 2026Bias-Constrained Diffusion Schedules for PDE Emulations: Reconstruction Error Minimization and Efficient Unrolled Training9 Apr 2026
- 2026Learning Data-Efficient and Generalizable Neural Operators via Fundamental Physics Knowledge16 Feb 2026
38 more
- 2026Inverting Data Transformations via Diffusion Sampling9 Feb 2026
- 2026Loss Landscape Geometry of Partial Differential Equation Emulators: Or, Symmetry Learning via Gradient Alignment28 Jan 2026
- 2026Explainable AI: Learning from the Learners9 Jan 2026
- 2025Active Learning with Selective Time-Step Acquisition for PDEs22 Nov 2025
- 2025LieSolver: PDE-Constrained Learning for IBVPs via Lie Symmetries29 Oct 2025
- 2025Accelerating Data Generation for Nonlinear temporal PDEs via homologous perturbation in solution space24 Oct 2025
- 2025Control-Augmented Autoregressive Diffusion for Data Assimilation8 Oct 2025
- 2025Latent Mixture of Symmetries for Sample-Efficient Dynamic Learning4 Oct 2025
- 2025Explicit Discovery of Nonlinear Symmetries from Dynamic Data2 Oct 2025
- 2025Adaptive Canonicalization with Application to Invariant Anisotropic Geometric Networks29 Sep 2025
- 2025Merging Memory and Space: A State Space Neural Operator31 Jul 2025
- 2025Scale-Consistent Learning for Partial Differential Equations24 Jul 2025
- 2025Overtone: Cyclic Patch Modulation for Clean, Efficient, and Flexible Physics Emulators12 Jul 2025
- 2025Principled Approaches for Extending Neural Architectures to Function Spaces for Operator Learning12 Jun 2025
- 2025Geometric and Physical Constraints Synergistically Enhance Neural PDE Surrogates5 Jun 2025
- 2025Governing Equation Discovery from Data Based on Differential Invariants24 May 2025
- 2025Beyond Accuracy: EcoL2 Metric for Sustainable Neural PDE Solvers18 May 2025
- 2025Discovering Symbolic Differential Equations with Symmetry Invariants17 May 2025
- 2025Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solving1 Apr 2025
- 2025Mechanistic PDE Networks for Discovery of Governing Equations25 Feb 2025
- 2025Generalized Lie Symmetries in Physics-Informed Neural Operators1 Feb 2025
- 2025BARNN: A Bayesian Autoregressive and Recurrent Neural Network30 Jan 2025
- 2025MultiPDENet: PDE-embedded Learning with Multi-time-stepping for Accelerated Flow Simulation27 Jan 2025
- 2024Predicting Change, Not States: An Alternate Framework for Neural PDE Surrogates17 Dec 2024
- 2024GEPS: Boosting Generalization in Parametric PDE Neural Solvers through Adaptive Conditioning31 Oct 2024
- 2024Learning Infinitesimal Generators of Continuous Symmetries from Data29 Oct 2024
- 2024On conditional diffusion models for PDE simulations21 Oct 2024
- 2024On the VC dimension of deep group convolutional neural networks21 Oct 2024
- 2024Scalable Mechanistic Neural Networks for Differential Equations and Machine Learning8 Oct 2024
- 2024Zebra: In-Context Generative Pretraining for Solving Parametric PDEs4 Oct 2024
- 2024Lie Algebra Canonicalization: Equivariant Neural Operators under arbitrary Lie Groups3 Oct 2024
- 2024Adversarial Learning for Neural PDE Solvers with Sparse Data4 Sep 2024
- 2024Active Learning for Neural PDE Solvers2 Aug 2024
- 2024Deep deterministic policy gradient with symmetric data augmentation for lateral attitude tracking control of a fixed-wing aircraft13 Jul 2024
- 2024Strategies for Pretraining Neural Operators12 Jun 2024
- 2024Linearization Turns Neural Operators into Function-Valued Gaussian Processes7 Jun 2024
- 2024Symmetry-Informed Governing Equation Discovery27 May 2024
- 2024Masked Autoencoders are PDE Learners26 Mar 2024
50 newest of 76 All 76 (JSON)
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 205
- 2024On conditional diffusion models for PDE simulationscited by 56
- 2026Inverting Data Transformations via Diffusion Samplingcited by 2
- 2025Control-Augmented Autoregressive Diffusion for Data Assimilationcited by 0
- 2026Bias-Constrained Diffusion Schedules for PDE Emulations: Reconstruction Error Minimization and Efficient Unrolled Trainingcited by 0
- 2026Bias-Constrained Diffusion Schedules for PDE Emulations: Reconstruction Error Minimization and Efficient Unrolled Training9 Apr 2026
- 2026Inverting Data Transformations via Diffusion Sampling9 Feb 2026
- 2025Control-Augmented Autoregressive Diffusion for Data Assimilation8 Oct 2025
- 2024On conditional diffusion models for PDE simulations21 Oct 2024
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solvers10 Aug 2023
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
- 2023Scalable Transformer for PDE Surrogate Modelingcited by 123
- 2022Clifford Neural Layers for PDE Modelingcited by 91
- 2022AirfRANS: High Fidelity Computational Fluid Dynamics Dataset for Approximating Reynolds-Averaged Navier-Stokes Solutionscited by 77
- 2023Training neural operators to preserve invariant measures of chaotic attractorscited by 43
- 2023Group Equivariant Fourier Neural Operators for Partial Differential Equationscited by 42
- 2023Geometric Clifford Algebra Networkscited by 37
- 2024Data-Efficient Operator Learning via Unsupervised Pretraining and In-Context Learningcited by 32
- 2022Guaranteed Conservation of Momentum for Learning Particle-based Fluid Dynamicscited by 30
- 2023Self-Supervised Learning with Lie Symmetries for Partial Differential Equationscited by 29
- 2023Lie Point Symmetry and Physics Informed Networkscited by 24
- 2024HAMLET: Graph Transformer Neural Operator for Partial Differential Equationscited by 24
12 most cited of 71 All 71 (JSON)
- 2026LieDiscover: Adaptive Symbolic Library Construction for Explicit Open-form Symmetry Discovery27 Sep 2026
- 2026Unsupervised Adaptation of PDE Foundation Models7 Aug 2026
- 2026LieStoNet: Learning Lie Symmetries from Spatiotemporal Data for Stochastic Dynamical Systems3 Aug 2026
- 2026On the post-hoc Evaluation of PDE Discovery: A Multifaceted Challenge of Scientific Advancement26 Jul 2026
- 2026How Much Memory Do We Need? Adaptive Memory Gate for Neural Operators11 Jun 2026
- 2026GENERIC-FNO: Embedding Energy Conservation and Entropy Production into Fourier Neural Operators6 Jun 2026
- 2026PDEInvBench: A Comprehensive Dataset and Design Space Exploration of Neural Networks for PDE Inverse Problems25 May 2026
- 2026Data-Efficient Neural Operator Training via Physics-Based Active Learning20 May 2026
- 2026PACE-FNO: Physics-Aligned Canonical Equivariance for Fourier Neural Operators18 May 2026
- 2026AdamFLIP: Adaptive Momentum Feedback Linearization Optimization for Hard Constrained PINN Training8 May 2026
- 2026Learning Data-Efficient and Generalizable Neural Operators via Fundamental Physics Knowledge16 Feb 2026
- 2026Loss Landscape Geometry of Partial Differential Equation Emulators: Or, Symmetry Learning via Gradient Alignment28 Jan 2026
12 newest of 71 All 71 (JSON)