Do Neural Operators Forget Geometry? The Forgetting Hypothesis in Deep Operator Learning
Not on the map. Why?
Abstract (arXiv)
Neural operators perform well on structured domains, yet their behaviour on irregular geometries remains poorly understood. We show that this limitation is not merely an encoding issue, but a depth-wise failure mode inherent to deep operator architectures. We formalise the Geometric Forgetting Hypothesis: due to the Markovian structure of operator layers and their reliance on global mixing mechanisms, neural operators progressively lose access to domain geometry as depth increases. Using layer-wise geometric probing, we demonstrate that both spectral and attention-based operators systematically lose geometric fidelity. We show that this geometric forgetting degrades accuracy, stability, and generalisation. To counteract it, we introduce a lightweight geometry memory injection mechanism that restores geometric constraints at intermediate depths with minimal architectural overhead. This simple intervention consistently mitigates forgetting and exposes a geometric shortcut instability in transformer-based operators, revealing that geometric retention is a structural requirement rather than a design choice.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 39 references that are in the map. First posted on arXiv 7 May 2026.
Builds on
28 of 39 references.
References · 28
- 2017Attention Is All You Needcited by 60,361
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2020Implicit Neural Representations with Periodic Activation Functionscited by 2,238
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 2,018
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2015Deep Learning and the Information Bottleneck Principlecited by 802
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 734
- 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 527
- 2022Fourier Neural Operator with Learned Deformations for PDEs on General Geometriescited by 427
16 more
- 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 326
- 2023Geometry-Informed Neural Operator for Large-Scale 3D PDEscited by 303
- 2021U-FNO -- An enhanced Fourier neural operator-based deep-learning model for multiphase flowcited by 297
- 2022Transformers Learn Shortcuts to Automatacited by 268
- 2021Factorized Fourier Neural Operatorscited by 249
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 248
- 2025Transolver++: An Accurate Neural Solver for PDEs on Million-Scale Geometriescited by 97
- 2022AirfRANS: High Fidelity Computational Fluid Dynamics Dataset for Approximating Reynolds-Averaged Navier-Stokes Solutionscited by 82
- 2024CViT: Continuous Vision Transformer for Operator Learningcited by 51
- 2024A Library for Learning Neural Operatorscited by 38
- 2024On the Benefits of Memory for Modeling Time-Dependent PDEscited by 31
- 2024FlowBench: A Large Scale Benchmark for Flow Simulation over Complex Geometriescited by 19
- 2025Geometric Operator Learning with Optimal Transportcited by 16
- 2024Spatio-spectral graph neural operator for solving computational mechanics problems on irregular domain and unstructured gridcited by 6
- 2025Physics- and geometry-aware spatio-spectral graph neural operator for time-independent and time-dependent PDEscited by 3
- 2025Graph-Based Operator Learning from Limited Data on Irregular Domainscited by 1
Built on by
Not cited yet.