We propose DiffNO, a novel framework that synergizes diffusion models with kernel-integrated neural operators to solve nonlinear partial differential equations (PDEs). Compared to deterministic architectures, diffusion models have a better chance of learning the complex mappings in the evolution of nonlinear PDEs due to the introduction of stochastic variables. However, it is challenging for diffusion models to directly learn complex nonlinear mappings on their own. To address these limitations, DiffNO constructs a kernel-integrated diffusion operator, incorporating the structure of Green’s functions as prior knowledge into the diffusion drift term, thereby establishing a physically constrained stochastic evolution process. On a dataset containing various types of PDEs, especially including two highly nonlinear PDEs, the 2D Cahn-Hilliard and reaction-diffusion systems, our method significantly outperforms current state-of-the-art methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

DiffNO: Neural Operator Learning Using Physically Structured Constrained Diffusion Model

  • Zhichen Feng,
  • Xin Zhang

摘要

We propose DiffNO, a novel framework that synergizes diffusion models with kernel-integrated neural operators to solve nonlinear partial differential equations (PDEs). Compared to deterministic architectures, diffusion models have a better chance of learning the complex mappings in the evolution of nonlinear PDEs due to the introduction of stochastic variables. However, it is challenging for diffusion models to directly learn complex nonlinear mappings on their own. To address these limitations, DiffNO constructs a kernel-integrated diffusion operator, incorporating the structure of Green’s functions as prior knowledge into the diffusion drift term, thereby establishing a physically constrained stochastic evolution process. On a dataset containing various types of PDEs, especially including two highly nonlinear PDEs, the 2D Cahn-Hilliard and reaction-diffusion systems, our method significantly outperforms current state-of-the-art methods.