Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
We introduce Mixed Poisson Diffusion Models (MPDMs), a novel diffusion framework leveraging multivariate counting processes instead of Gaussian noise for generative modeling of continuous data. This approach is specifically motivated by the need to address Poisson inverse problems, which involve discrete, Poisson-distributed observations and are currently underexplored in diffusion model literature. MPDMs enable a natural framework for posterior sampling in these problems, with empirical evaluation demonstrating their suitability for realistic Poisson inverse scenarios.
Poisson noise, not just Gaussian, unlocks powerful generative models for a critical class of inverse problems.
Diffusion models are a powerful framework for learning complex probability distributions. While widely used diffusion models for continuous data rely on Gaussian noising processes, we instead develop diffusion models based on Poisson noise. More precisely, we introduce and study a new mathematical framework, called Mixed Poisson Diffusion Models (MPDM), with noising dynamics driven by multivariate counting processes. Our methodology is motivated by Poisson inverse problems. The success of Gaussian diffusion models has led to their use as prior models in a growing number of posterior sampling methods for inverse problems. However, most of these approaches focus on Gaussian inverse problems, in which observations are generated through a deterministic forward operator corrupted by additive Gaussian noise. By contrast, Poisson inverse problems, where observations are discrete and Poisson distributed, remain comparatively underexplored in this literature. Building on our Poisson generative dynamics, we develop a natural framework for posterior sampling in Poisson inverse problems and derive several posterior sampling algorithms based on the proposed model. We evaluate our approach on realistic Poisson inverse problems and show that the resulting framework is well suited to this class of problems.