arXiv · 2410.19149
Structured Diffusion Models with Mixture of Gaussians as Prior Distribution
Abstract
We propose a class of structured diffusion models, in which the prior distribution is chosen as a mixture of Gaussians, rather than a standard Gaussian distribution. The specific mixed Gaussian distribution, as prior, can be chosen to incorporate certain structured information of the data. We develop a simple-to-implement training procedure that smoothly accommodates the use of mixed Gaussian as prior. Theory is provided to quantify the benefits of our proposed models, compared to the classical diffusion models. Numerical experiments with synthetic, image and operational data are conducted to show comparative advantages of our model. Our method is shown to be robust to mis-specifications and in particular suits situations where training resources are limited or faster training in real time is desired.
Explore related subjects
Keep this discovery
Nanshan Jia, Tingyu Zhu, Haoyu Liu, Zeyu Zheng. 2024-10-24. Structured Diffusion Models with Mixture of Gaussians as Prior Distribution. https://arxiv.org/abs/2410.19149
Cite the original work for its findings. Save a collection to share your selection of sources.