FNO (2020)

Fourier Neural Operator for Parametric Partial Differential Equations
Zongyi Li; Nikola Kovachki; Kamyar Azizzadenesheli; Burigede Liu; Kaushik Bhattacharya; Andrew Stuart; Anima Anandkumar

Paper: link
Quick facts

Type: neural operator (Fourier)
Resolution-invariant design
Fast surrogate inference

← Research · Home

TL;DR

A resolution-invariant neural operator that learns solution operators of PDEs by parameterizing integral kernels in Fourier space, enabling efficient training and strong cross-resolution generalization.

Problem

Learn the mapping (operator) from input functions (e.g., initial conditions, coefficients, forcing) to PDE solution fields, with an architecture that generalizes across discretizations/resolutions.

Benefits vs others

Interesting detail

Core method (math)

Template for Operator learning. Paper-specific equations are added when manually curated.

\[v_{t+1}(x) = \sigma\left( W v_t(x) + (\mathcal{K} v_t)(x) \right),\quad (\mathcal{K}v)(x) = \int_D k(x,y)\,v(y)\,dy\] \[(\mathcal{K}v_t)(x) = \mathcal{F}^{-1}\big( R\,\mathcal{F}(v_t) \big)(x)\] \[(\mathcal{K}v_t)(x) = \sum_{|k|\le k_{\max}} e^{2\pi i\langle x,k\rangle}\,R_k\,\hat v_{t,k}\]

Main theoretical contribution

Not curated yet. Add bullet points under <code>theory</code> in JSON.

Main contribution

Main results (headline)

(Optional) Add main_results for a quick headline summary.

Experiments

PDE problems

  • Burgers equation
  • Darcy flow
  • Navier–Stokes
  • Fluid dynamics

Tasks

  • Operator learning / surrogate modeling
  • Super-resolution / cross-resolution generalization

Experiment setting (high level)

  • Supervised learning on simulated PDE datasets.
  • Often evaluated on resolution generalization and rollout stability.

Comparable baselines

Main results

Navier–Stokes (64×64): relative error at t=1 (avg over test set) + time/epoch

Lower is better. Time per epoch is seconds (as reported in the paper).

Method#ParamsTime/epoch (s)ν=1e−3 (N=1000)ν=1e−4 (N=1000)ν=1e−4 (N=10000)ν=1e−5 (N=1000)
U-Net4.3M1191.0e-23.3e-21.0e-28.4e-2
TF-Net2.3M1007.0e-34.8e-27.2e-31.1e-1
ResNet2.6M1505.6e-34.5e-24.0e-31.1e-1
FNO0.5M398.7e-46.9e-31.8e-31.5e-2

Burgers (1D): cross-resolution generalization (train s=2048, test varying s)

Metric: relative L2 error. Lower is better.

Methods=256s=512s=1024s=2048s=4096s=8192
NN1.6e-21.4e-21.2e-21.1e-21.0e-29.8e-3
GCN2.0e-21.8e-21.6e-21.6e-21.7e-21.8e-2
FCN1.5e-21.4e-21.3e-21.3e-21.3e-21.2e-2
PCA-NN2.6e-21.9e-21.7e-21.7e-21.6e-21.6e-2
GNO1.1e-15.8e-23.7e-22.5e-22.0e-21.8e-2
LNO1.2e-19.4e-27.8e-25.4e-24.2e-23.5e-2
MGNO8.8e-37.2e-35.9e-35.7e-35.3e-35.0e-3
FNO1.1e-37.3e-44.6e-43.4e-43.0e-42.7e-4

Darcy (2D): cross-resolution generalization (train s=421, test varying s)

Metric: relative L2 error. Lower is better.

Methods=85s=141s=211s=421
NN5.1e-24.9e-24.8e-24.7e-2
FCN8.9e-28.8e-28.5e-28.4e-2
PCA-NN3.2e-22.5e-22.4e-22.4e-2
RBM3.0e-22.0e-21.9e-21.5e-2
GNO1.1e-18.6e-27.8e-26.0e-2
LNO2.0e-11.8e-11.7e-1
MGNO3.2e-22.6e-22.4e-22.2e-2
FNO1.9e-21.3e-21.0e-28.8e-3

Citation (BibTeX)

@inproceedings{li2021fno,
  title={Fourier Neural Operator for Parametric Partial Differential Equations},
  author={Li, Zongyi and Kovachki, Nikola and Azizzadenesheli, Kamyar and Liu, Burigede and Bhattacharya, Kaushik and Stuart, Andrew and Anandkumar, Anima},
  booktitle={International Conference on Learning Representations (ICLR)},
  year={2021}
}