Salvato in:
Dettagli Bibliografici
Autori principali: George, Robert Joseph, Zhao, Jiawei, Kossaifi, Jean, Li, Zongyi, Anandkumar, Anima
Natura: Preprint
Pubblicazione: 2022
Soggetti:
Accesso online:https://arxiv.org/abs/2211.15188
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916145982865408
author George, Robert Joseph
Zhao, Jiawei
Kossaifi, Jean
Li, Zongyi
Anandkumar, Anima
author_facet George, Robert Joseph
Zhao, Jiawei
Kossaifi, Jean
Li, Zongyi
Anandkumar, Anima
contents Fourier Neural Operators (FNO) offer a principled approach to solving challenging partial differential equations (PDE) such as turbulent flows. At the core of FNO is a spectral layer that leverages a discretization-convergent representation in the Fourier domain, and learns weights over a fixed set of frequencies. However, training FNO presents two significant challenges, particularly in large-scale, high-resolution applications: (i) Computing Fourier transform on high-resolution inputs is computationally intensive but necessary since fine-scale details are needed for solving many PDEs, such as fluid flows, (ii) selecting the relevant set of frequencies in the spectral layers is challenging, and too many modes can lead to overfitting, while too few can lead to underfitting. To address these issues, we introduce the Incremental Fourier Neural Operator (iFNO), which progressively increases both the number of frequency modes used by the model as well as the resolution of the training data. We empirically show that iFNO reduces total training time while maintaining or improving generalization performance across various datasets. Our method demonstrates a 10% lower testing error, using 20% fewer frequency modes compared to the existing Fourier Neural Operator, while also achieving a 30% faster training.
format Preprint
id arxiv_https___arxiv_org_abs_2211_15188
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Incremental Spatial and Spectral Learning of Neural Operators for Solving Large-Scale PDEs
George, Robert Joseph
Zhao, Jiawei
Kossaifi, Jean
Li, Zongyi
Anandkumar, Anima
Machine Learning
Fourier Neural Operators (FNO) offer a principled approach to solving challenging partial differential equations (PDE) such as turbulent flows. At the core of FNO is a spectral layer that leverages a discretization-convergent representation in the Fourier domain, and learns weights over a fixed set of frequencies. However, training FNO presents two significant challenges, particularly in large-scale, high-resolution applications: (i) Computing Fourier transform on high-resolution inputs is computationally intensive but necessary since fine-scale details are needed for solving many PDEs, such as fluid flows, (ii) selecting the relevant set of frequencies in the spectral layers is challenging, and too many modes can lead to overfitting, while too few can lead to underfitting. To address these issues, we introduce the Incremental Fourier Neural Operator (iFNO), which progressively increases both the number of frequency modes used by the model as well as the resolution of the training data. We empirically show that iFNO reduces total training time while maintaining or improving generalization performance across various datasets. Our method demonstrates a 10% lower testing error, using 20% fewer frequency modes compared to the existing Fourier Neural Operator, while also achieving a 30% faster training.
title Incremental Spatial and Spectral Learning of Neural Operators for Solving Large-Scale PDEs
topic Machine Learning
url https://arxiv.org/abs/2211.15188