Physics-embedded Fourier Neural Network for Partial Differential Equations

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Xu, Qingsong, Thuerey, Nils, Shi, Yilei, Bamber, Jonathan, Ouyang, Chaojun, Zhu, Xiao Xiang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913432258740224
author Xu, Qingsong
Thuerey, Nils
Shi, Yilei
Bamber, Jonathan
Ouyang, Chaojun
Zhu, Xiao Xiang
author_facet Xu, Qingsong
Thuerey, Nils
Shi, Yilei
Bamber, Jonathan
Ouyang, Chaojun
Zhu, Xiao Xiang
contents We consider solving complex spatiotemporal dynamical systems governed by partial differential equations (PDEs) using frequency domain-based discrete learning approaches, such as Fourier neural operators. Despite their widespread use for approximating nonlinear PDEs, the majority of these methods neglect fundamental physical laws and lack interpretability. We address these shortcomings by introducing Physics-embedded Fourier Neural Networks (PeFNN) with flexible and explainable error control. PeFNN is designed to enforce momentum conservation and yields interpretable nonlinear expressions by utilizing unique multi-scale momentum-conserving Fourier (MC-Fourier) layers and an element-wise product operation. The MC-Fourier layer is by design translation- and rotation-invariant in the frequency domain, serving as a plug-and-play module that adheres to the laws of momentum conservation. PeFNN establishes a new state-of-the-art in solving widely employed spatiotemporal PDEs and generalizes well across input resolutions. Further, we demonstrate its outstanding performance for challenging real-world applications such as large-scale flood simulations.
format Preprint
id arxiv_https___arxiv_org_abs_2407_11158
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Physics-embedded Fourier Neural Network for Partial Differential Equations
Xu, Qingsong
Thuerey, Nils
Shi, Yilei
Bamber, Jonathan
Ouyang, Chaojun
Zhu, Xiao Xiang
Machine Learning
Numerical Analysis
We consider solving complex spatiotemporal dynamical systems governed by partial differential equations (PDEs) using frequency domain-based discrete learning approaches, such as Fourier neural operators. Despite their widespread use for approximating nonlinear PDEs, the majority of these methods neglect fundamental physical laws and lack interpretability. We address these shortcomings by introducing Physics-embedded Fourier Neural Networks (PeFNN) with flexible and explainable error control. PeFNN is designed to enforce momentum conservation and yields interpretable nonlinear expressions by utilizing unique multi-scale momentum-conserving Fourier (MC-Fourier) layers and an element-wise product operation. The MC-Fourier layer is by design translation- and rotation-invariant in the frequency domain, serving as a plug-and-play module that adheres to the laws of momentum conservation. PeFNN establishes a new state-of-the-art in solving widely employed spatiotemporal PDEs and generalizes well across input resolutions. Further, we demonstrate its outstanding performance for challenging real-world applications such as large-scale flood simulations.
title Physics-embedded Fourier Neural Network for Partial Differential Equations
topic Machine Learning
Numerical Analysis
url https://arxiv.org/abs/2407.11158