arXiv · 2403.14404
Physics-Informed Diffusion Models
Abstract
Generative models such as denoising diffusion models are quickly advancing their ability to approximate highly complex data distributions. They are also increasingly leveraged in scientific machine learning, where samples from the implied data distribution are expected to adhere to specific governing equations. We present a framework that unifies generative modeling and partial differential equation fulfillment by introducing a first-principle-based loss term that enforces generated samples to fulfill the underlying physical constraints. Our approach reduces the residual error by up to two orders of magnitude compared to previous work in a fluid flow case study and outperforms task-specific frameworks in relevant metrics for structural topology optimization. We also present numerical evidence that our extended training objective acts as a natural regularization mechanism against overfitting. Our framework is simple to implement and versatile in its applicability for imposing equality and inequality constraints as well as auxiliary optimization objectives.
Explore related subjects
Keep this discovery
Jan-Hendrik Bastek, WaiChing Sun, Dennis M. Kochmann. 2024-03-21. Physics-Informed Diffusion Models. https://arxiv.org/abs/2403.14404
Cite the original work for its findings. Save a collection to share your selection of sources.