Skip to content

Geometry-Aware Fourier Neural Operator (Geo-FNO; m4e7-dev)

License

Notifications You must be signed in to change notification settings

m4e7/Geo-FNO

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Geometry-Aware Fourier Neural Operator (Geo-FNO)

Deep learning surrogate models have shown promise in solving partial differential equations (PDEs). Among them, the Fourier neural operator (FNO) achieves good accuracy, and is significantly faster compared to numerical solvers, on a variety of PDEs, such as fluid flows. However, the FNO uses the Fast Fourier transform (FFT), which is limited to rectangular domains with uniform grids. In this work, we propose a new framework, viz., geo-FNO, to solve PDEs on arbitrary geometries. Geo-FNO learns to deform the input (physical) domain, which may be irregular, into a latent space with a uniform grid. The FNO model with the FFT is applied in the latent space. The resulting geo-FNO model has both the computation efficiency of FFT and the flexibility of handling arbitrary geometries. Our geo-FNO is also flexible in terms of its input formats, viz., point clouds, meshes, and design parameters are all valid inputs. We consider a variety of PDEs such as the Elasticity, Plasticity, Euler's, and Navier-Stokes equations, and both forward modeling and inverse design problems. Geo-FNO is $10^5$ times faster than the standard numerical solvers and twice more accurate compared to direct interpolation on existing ML-based PDE solvers such as the standard FNO.

https://arxiv.org/abs/2207.05209

Requirements

  • The code only depends on pytorch (>=1.8.0) PyTorch.

Datasets

We provide the elasticity, plasticity, airfoil flows (naca) and pipe flows (ns) datasets we used in the paper. The data generation configuration can be found in the paper.

Examples

python3 elasticity/elas_geofno.py

About

Geometry-Aware Fourier Neural Operator (Geo-FNO; m4e7-dev)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 76.2%
  • Python 18.3%
  • Julia 5.5%