# Derivative-Informed Fourier Neural Operator: Universal Approximation and Applications to PDE-Constrained Optimization

arXiv: https://arxiv.org/abs/2512.14086 · HTML: https://mlsanity.com/paper/2512.14086/

## Key facts

- Cited by 7 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 16 Dec 2025.

## Built on by (most cited first, 7 of 7)

- [Variationally correct operator learning: Reduced basis neural operator with a posteriori error estimation](https://mlsanity.com/paper/2512.21319/) (2025)
- [Performance of Neural and Polynomial Operator Surrogates](https://mlsanity.com/paper/2604.00689/) (2026)
- [Deep Hilbert--Galerkin Methods for Infinite-Dimensional PDEs and Optimal Control](https://mlsanity.com/paper/2603.19463/) (2026)
- [Universal Approximation of Nonlinear Operators and Their Derivatives](https://mlsanity.com/paper/2605.15285/) (2026)
- [LEIA: Learned Environment for Interactive Architected Materials](https://mlsanity.com/paper/2605.28368/) (2026)
- [Derivative-Informed Training of Neural Operators On-the-Fly via Sketched Tangent Consistency](https://mlsanity.com/paper/2609.32797/) (2026)
- [Tangent Schr"odinger Bridge Matching: Learning Stochastic Transport with Mechanistic Sensitivities](https://mlsanity.com/paper/2610.02906/) (2026)

## Abstract (arXiv)

We present approximation theories and efficient training methods for derivative-informed Fourier neural operators (DIFNOs) with applications to PDE-constrained optimization. A DIFNO is an FNO trained by minimizing its prediction error jointly on output and Fr'echet derivative samples of a high-fidelity operator (e.g., a parametric PDE solution operator). As a result, a DIFNO can closely emulate not only the high-fidelity operator's response but also its sensitivities. To motivate the use of DIFNOs instead of conventional FNOs as surrogate models, we show that accurate surrogate-driven PDE-constrained optimization requires accurate surrogate Fr'echet derivatives. Then, we establish (i) simultaneous universal approximation of continuously differentiable operators and their Fr'echet derivatives by FNOs on compact sets, and (ii) universal approximation of continuously differentiable operators by FNOs in weighted Sobolev spaces with input measures that have unbounded supports. Our theoretical results certify the capability of FNOs for accurate derivative-informed operator learning and for the solution of PDE-constrained optimization problems. Furthermore, we develop efficient training schemes that leverage dimensionality reduction and multi-resolution techniques to significantly reduce memory and computational costs in Fr'echet derivative learning. Numerical examples on nonlinear diffusion--reaction, Helmholtz, and Navier--Stokes equations demonstrate that DIFNOs are superior in sample complexity for operator learning and solving infinite-dimensional PDE-constrained inverse problems, achieving high accuracy at low training sample sizes.
