A Score-based Diffusion Model Approach for Adaptive Learning of Stochastic Partial Differential Equation Solutions
By: Toan Huynh , Ruth Lopez Fajardo , Guannan Zhang and more
Potential Business Impact:
Helps predict changing weather and science.
We propose a novel framework for adaptively learning the time-evolving solutions of stochastic partial differential equations (SPDEs) using score-based diffusion models within a recursive Bayesian inference setting. SPDEs play a central role in modeling complex physical systems under uncertainty, but their numerical solutions often suffer from model errors and reduced accuracy due to incomplete physical knowledge and environmental variability. To address these challenges, we encode the governing physics into the score function of a diffusion model using simulation data and incorporate observational information via a likelihood-based correction in a reverse-time stochastic differential equation. This enables adaptive learning through iterative refinement of the solution as new data becomes available. To improve computational efficiency in high-dimensional settings, we introduce the ensemble score filter, a training-free approximation of the score function designed for real-time inference. Numerical experiments on benchmark SPDEs demonstrate the accuracy and robustness of the proposed method under sparse and noisy observations.
Similar Papers
Training-free score-based diffusion for parameter-dependent stochastic dynamical systems
Machine Learning (Stat)
Lets computers predict how things change with different settings.
Score-based constrained generative modeling via Langevin diffusions with boundary conditions
Machine Learning (Stat)
Makes AI create images that follow rules.
Particle-Guided Diffusion Models for Partial Differential Equations
Machine Learning (CS)
Makes computer models follow science rules better.