mlsanityData to 7 Oct 2026

Loss Landscape Geometry of Partial Differential Equation Emulators: Or, Symmetry Learning via Gradient Alignment

arXiv 2601.20172 ↗First posted 28 Jan 2026cs.LG physics.comp-phcited by 0

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

  1. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  2. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
  3. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
  4. 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
  5. 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
  6. 2016Group Equivariant Convolutional Networkscited by 1,154
  7. 2021E(n) Equivariant Graph Neural Networkscited by 951
  8. 2017A Closer Look at Memorization in Deep Networkscited by 936
  9. 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 735
  10. 2014New insights and perspectives on the natural gradient methodcited by 435
  11. 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 396
  12. 2019Making Convolutional Networks Shift-Invariant Againcited by 390
25 more
  1. 20183D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Datacited by 348
  2. 2023ClimaX: A foundation model for weather and climatecited by 325
  3. 2016Steerable CNNscited by 298
  4. 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 205
  5. 2018Why do deep convolutional networks generalize so poorly to small image transformations?cited by 198
  6. 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 190
  7. 2024Poseidon: Efficient Foundation Models for PDEscited by 186
  8. 2020On Translation Invariance in CNNs: Convolutional Layers can Exploit Absolute Spatial Locationcited by 141
  9. 2023Towards Foundation Models for Scientific Machine Learning: Characterizing Scaling and Transfer Behaviorcited by 128
  10. 2022Approximately Equivariant Networks for Imperfectly Symmetric Dynamicscited by 95
  11. 2019On Second-Order Group Influence Functions for Black-Box Predictionscited by 78
  12. 2022Lie Point Symmetry Data Augmentation for Neural PDE Solverscited by 76
  13. 2021Residual Pathway Priors for Soft Equivariance Constraintscited by 74
  14. 2022On the Symmetries of Deep Learning Models and their Internal Representationscited by 63
  15. 2025Gradient Alignment in Physics-informed Neural Networks: A Second-Order Optimization Perspectivecited by 55
  16. 2022The Lie Derivative for Measuring Learned Equivariancecited by 51
  17. 2019Emergent properties of the local geometry of neural loss landscapescited by 46
  18. 2018Don't Unroll Adjoint: Differentiating SSA-Form Programscited by 44
  19. 2020Coherent Gradients: An Approach to Understanding Generalization in Gradient Descent-based Optimizationcited by 41
  20. 2019The Local Elasticity of Neural Networkscited by 37
  21. 2025RIGNO: A Graph-based framework for robust and accurate operator learning for PDEs on arbitrary domainscited by 22
  22. 2025MORPH: PDE Foundation Models with Arbitrary Data Modalitycited by 12
  23. 2025A Tale of Two Symmetries: Exploring the Loss Landscape of Equivariant Modelscited by 8
  24. 2023Equivariant geometric convolutions for emulation of dynamical systemscited by 3
  25. 2025Generalization vs. Memorization in Autoregressive Deep Learning: Or, Examining Temporal Decay of Gradient Coherencecited by 1

Built on by

Not cited yet.