Papers › Product of Gaussian Mixture Diffusion Models

Product of Gaussian Mixture Diffusion Models

19 Oct 2023arXiv:2310.12653archive 2025-07-28

Martin Zach, Erich Kobler, Antonin Chambolle, Thomas Pock

In this work we tackle the problem of estimating the density f_X of a random variable $ X $ by successive smoothing, such that the smoothed random variable $ Y $ fulfills the diffusion partial differential equation (∂ₜ - Δ₁)f_Y( · , t) = 0 with initial condition f_Y( · , 0) = f_X. We propose a product-of-experts-type model utilizing Gaussian mixture experts and study configurations that admit an analytic expression for f_Y ( · , t). In particular, with a focus on image processing, we derive conditions for models acting on filter-, wavelet-, and shearlet responses. Our construction naturally allows the model to be trained simultaneously over the entire diffusion horizon using empirical Bayes. We show numerical results for image denoising where our models are competitive while being tractable, interpretable, and having only a small number of learnable parameters. As a byproduct, our models can be used for reliable noise level estimation, allowing blind denoising of images corrupted by heteroscedastic noise.

PaperPDFCode

Code

vlogroup/pogmdm officialmentioned in papermentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

DenoisingImage DenoisingNoise Estimation

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

DiffusionFocus

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections