Do Neural Operators Forget Geometry? The Forgetting Hypothesis in Deep Operator Learning
Yanming Xia, Angelica Aviles-Rivero
Abstract
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.
BibTeX
@inproceedings{
xia2026do,
title={Do Neural Operators Forget Geometry? The Forgetting Hypothesis in Deep Operator Learning},
author={Yanming Xia and Angelica I Aviles-Rivero},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=OnmYXKZ6oa}
}