Particle-Guided Diffusion Models for Partial Differential Equations
Andrew Millard, Fredrik Lindsten, Zheng Zhao
Abstract
We introduce a guided stochastic sampling method that augments sampling from diffusion models with physics-based guidance derived from partial differential equation (PDE) residuals and observational constraints, ensuring generated samples remain physically admissible. We embed this sampling procedure within a new Sequential Monte Carlo (SMC) framework, yielding a scalable generative PDE solver. Across multiple benchmark PDE systems as well as multiphysics and interacting PDE systems, our method produces solution fields with lower numerical error than existing state-of-the-art generative methods.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on20
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Diffusion Models Beat GANs on Image SynthesisPrafulla Dhariwal, Alexander Quinn NicholNeurIPS 2021 · 13,211 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Elucidating the Design Space of Diffusion-Based Generative ModelsTero Karras, Miika Aittala, Timo Aila, Samuli LaineNeurIPS 2022 · 3,959 citations
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
Related papers
- Physics-Informed Diffusion Models in Spectral SpaceDavide Gallon, Philippe von Wurstemberger, Patrick Cheridito, Arnulf JentzenICML 2026 · 3 citations
- Solving Linear-Gaussian Bayesian Inverse Problems with Decoupled Diffusion Sequential Monte CarloFilip Ekström Kelvinius, Zheng Zhao, Fredrik LindstenICML 2025
- Gradient-Free Generation for Hard-Constrained SystemsChaoran Cheng, Boran Han, Danielle C. Maddix, Abdul Fatir Ansari et al.ICLR 2025
- Physics-Constrained Flow Matching: Sampling Generative Models with Hard ConstraintsUtkarsh Utkarsh, Pengfei Cai, Alan Edelman, Rafael Gómez-Bombarelli et al.NeurIPS 2025 · 60 citations
- Monte Carlo guided Denoising Diffusion models for Bayesian linear inverse problemsGabriel Cardoso, Yazid Janati El Idrissi, Sylvain Le Corff, Eric MoulinesICLR 2024 · 74 citations
