Generalizing PDE Emulation with Equation-Aware Neural Operators
Not on the map. Why?
Abstract (arXiv)
Solving partial differential equations (PDEs) can be prohibitively expensive using traditional numerical methods. Deep learning-based surrogate models typically specialize in a single PDE with fixed parameters. We present a framework for equation-aware emulation that generalizes to unseen PDEs, conditioning a neural model on a vector encoding representing the terms in a PDE and their coefficients. We present a baseline of four distinct modeling technqiues, trained on a family of 1D PDEs from the APEBench suite. Our approach achieves strong performance on parameter sets held out from the training distribution, with strong stability for rollout beyond the training window, and generalization to an entirely unseen PDE. This work was developed as part of a broader effort exploring AI systems that automate the creation of expert-level empirical software for scorable scientific tasks. The data and codebase are available at https://github.com/google-research/generalized-pde-emulator.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 15 of its 19 references that are in the map. First posted on arXiv 12 Nov 2025.
Builds on
15 of 19 references.
References · 15
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 527
- 2021Machine learning accelerated computational fluid dynamicscited by 471
- 2022Message Passing Neural PDE Solverscited by 386
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 197
- 2023Towards Foundation Models for Scientific Machine Learning: Characterizing Scaling and Transfer Behaviorcited by 131
- 2024Latent Neural Operator for Solving Forward and Inverse PDE Problemscited by 80
- 2024APEBench: A Benchmark for Autoregressive Neural Emulators of PDEscited by 60
- 2025An AI system to help scientists write expert-level empirical softwarecited by 60
3 more
Built on by
Citing papers per year: 2026: 2.
2 papers.