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

Read full text
Authors: James Amarel, Robyn Miller, Nicolas Hengartner, Benjamin J Migliori, Emily Casleton, Alexei N. Skurikhin, Earl Lawrence, Gerd J. Kunde
Cited by: 0

Abstract

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.

Fields of Study & Subjects

Model Reduction and Neural Networks Neural Networks and Applications Gaussian Processes and Bayesian Inference

Publication details

Journal / source
arXiv (Cornell University)
Publication date
28 January 2026
Document type
Preprint