Generative diffusion learning for parametric partial differential equations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Ting, Plechac, Petr, Knap, Jaroslaw
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910140489269248
author Wang, Ting
Plechac, Petr
Knap, Jaroslaw
author_facet Wang, Ting
Plechac, Petr
Knap, Jaroslaw
contents We develop a class of data-driven generative models that approximate the solution operator for parameter-dependent partial differential equations (PDE). We propose a novel probabilistic formulation of the operator learning problem based on recently developed generative denoising diffusion probabilistic models (DDPM) in order to learn the input-to-output mapping between problem parameters and solutions of the PDE. To achieve this goal we modify DDPM to supervised learning in which the solution operator for the PDE is represented by a class of conditional distributions. The probabilistic formulation combined with DDPM allows for an automatic quantification of confidence intervals for the learned solutions. Furthermore, the framework is directly applicable for learning from a noisy data set. We compare computational performance of the developed method with the Fourier Network Operators (FNO). Our results show that our method achieves comparable accuracy and recovers the noise magnitude when applied to data sets with outputs corrupted by additive noise.
format Preprint
id arxiv_https___arxiv_org_abs_2305_14703
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Generative diffusion learning for parametric partial differential equations
Wang, Ting
Plechac, Petr
Knap, Jaroslaw
Numerical Analysis
We develop a class of data-driven generative models that approximate the solution operator for parameter-dependent partial differential equations (PDE). We propose a novel probabilistic formulation of the operator learning problem based on recently developed generative denoising diffusion probabilistic models (DDPM) in order to learn the input-to-output mapping between problem parameters and solutions of the PDE. To achieve this goal we modify DDPM to supervised learning in which the solution operator for the PDE is represented by a class of conditional distributions. The probabilistic formulation combined with DDPM allows for an automatic quantification of confidence intervals for the learned solutions. Furthermore, the framework is directly applicable for learning from a noisy data set. We compare computational performance of the developed method with the Fourier Network Operators (FNO). Our results show that our method achieves comparable accuracy and recovers the noise magnitude when applied to data sets with outputs corrupted by additive noise.
title Generative diffusion learning for parametric partial differential equations
topic Numerical Analysis
url https://arxiv.org/abs/2305.14703