Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Enhancing Fourier Neural Operators with Local Spatial Features

About

Partial Differential Equation (PDE) problems often exhibit strong local spatial structures, and effectively capturing these structures is critical for approximating their solutions. Recently, the Fourier Neural Operator (FNO) has emerged as an efficient approach for solving these PDE problems. By using parametrization in the frequency domain, FNOs can efficiently capture global patterns. However, this approach inherently overlooks the critical role of local spatial features, as frequency-domain parameterized convolutions primarily emphasize global interactions without encoding comprehensive localized spatial dependencies. Although several studies have attempted to address this limitation, their extracted Local Spatial Features (LSFs) remain insufficient, and computational efficiency is often compromised. To address this limitation, we introduce a convolutional neural network (CNN)-based feature pre-extractor to capture LSFs directly from input data, resulting in a hybrid architecture termed \textit{Conv-FNO}. Furthermore, we introduce two novel resizing schemes to make our Conv-FNO resolution invariant. In this work, we focus on demonstrating the effectiveness of incorporating LSFs into FNOs by conducting both a theoretical analysis and extensive numerical experiments. Our findings show that this simple yet impactful modification enhances the representational capacity of FNOs and significantly improves performance on challenging PDE benchmarks.

Chaoyu Liu, Davide Murari, Lihao Liu, Yangming Li, Chris Budd, Carola-Bibiane Sch\"onlieb• 2025

Related benchmarks

TaskDatasetResultRank
PDE solvingDarcy (test)
Relative Error0.98
11
Long-Horizon StabilityNS-2D (test)
MSE0.0065
8
PDE Rollout SimulationKS (test)
Relative L2 Error (Rollout)0.382
8
PDE Rollout PredictionBurgers trajectory set (test)
Rollout Relative H115.62
8
PDE Rollout PredictionKS trajectory set (test)
Relative H1 Error41.2
8
PDE Rollout PredictionNS-2D trajectory set (test)
Rollout Relative H1 Error0.89
8
PDE Rollout SimulationBurgers' (test)
Relative L2 Error8.9
8
PDE Rollout SimulationNS-2D (test)
Rollout Relative L2 Error0.6
8
PDE Rollout PredictionKdV trajectory set (test)
Rollout Relative H1 Error0.074
8
PDE Rollout PredictionAdv-2D trajectory set (test)
Rollout Relative H1 Error5.45
8
Showing 10 of 19 rows

Other info

Follow for update