DiffNO: Neural Operator Learning Using Physically Structured Constrained Diffusion Model
摘要
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.