Loss Landscape Geometry of Partial Differential Equation Emulators: Or, Symmetry Learning via Gradient Alignment
Not on the map. Why?
Abstract (arXiv)
We study how neural emulators of partial differential equation solution operators internalize physical symmetries by introducing an influence-based diagnostic that measures the propagation of parameter updates between symmetry-related states, defined as the metric-weighted overlap of loss gradients evaluated along group orbits. This quantity probes the local geometry of the learned loss landscape and goes beyond forward-pass equivariance tests by directly assessing whether learning dynamics couple physically equivalent configurations. Applying our diagnostic to autoregressive fluid flow emulators, we show that orbit-wise gradient coherence provides the mechanism for learning to generalize over symmetry transformations and indicates when training selects a symmetry compatible basin. The result is a novel technique for evaluating if surrogate models have internalized symmetry properties of the known solution operator.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 37 of its 55 references that are in the map. First posted on arXiv 28 Jan 2026.
Builds on
37 of 55 references.
References · 37
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2016Group Equivariant Convolutional Networkscited by 1,154
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2017A Closer Look at Memorization in Deep Networkscited by 936
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 735
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 396
- 2019Making Convolutional Networks Shift-Invariant Againcited by 390
25 more
- 20183D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Datacited by 348
- 2023ClimaX: A foundation model for weather and climatecited by 325
- 2016Steerable CNNscited by 298
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 205
- 2018Why do deep convolutional networks generalize so poorly to small image transformations?cited by 198
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
- 2024Poseidon: Efficient Foundation Models for PDEscited by 186
- 2020On Translation Invariance in CNNs: Convolutional Layers can Exploit Absolute Spatial Locationcited by 141
- 2023Towards Foundation Models for Scientific Machine Learning: Characterizing Scaling and Transfer Behaviorcited by 128
- 2022Approximately Equivariant Networks for Imperfectly Symmetric Dynamicscited by 95
- 2019On Second-Order Group Influence Functions for Black-Box Predictionscited by 78
- 2022Lie Point Symmetry Data Augmentation for Neural PDE Solverscited by 76
- 2021Residual Pathway Priors for Soft Equivariance Constraintscited by 74
- 2022On the Symmetries of Deep Learning Models and their Internal Representationscited by 63
- 2025Gradient Alignment in Physics-informed Neural Networks: A Second-Order Optimization Perspectivecited by 55
- 2022The Lie Derivative for Measuring Learned Equivariancecited by 51
- 2019Emergent properties of the local geometry of neural loss landscapescited by 46
- 2018Don't Unroll Adjoint: Differentiating SSA-Form Programscited by 44
- 2020Coherent Gradients: An Approach to Understanding Generalization in Gradient Descent-based Optimizationcited by 41
- 2019The Local Elasticity of Neural Networkscited by 37
- 2025RIGNO: A Graph-based framework for robust and accurate operator learning for PDEs on arbitrary domainscited by 22
- 2025MORPH: PDE Foundation Models with Arbitrary Data Modalitycited by 12
- 2025A Tale of Two Symmetries: Exploring the Loss Landscape of Equivariant Modelscited by 8
- 2023Equivariant geometric convolutions for emulation of dynamical systemscited by 3
- 2025Generalization vs. Memorization in Autoregressive Deep Learning: Or, Examining Temporal Decay of Gradient Coherencecited by 1
Built on by
Not cited yet.