deepinv.sampling#

This package contains various posterior sampling algorithms, including diffusion-based methods and MCMC methods. Please refer to the user guide for more details.

Diffusion models with Stochastic Differential Equations for Image Generation and Posterior Sampling#

User Guide: refer to Diffusion models with Stochastic Differential Equations for Image Generation and Posterior Sampling for more information.

deepinv.sampling.BaseSDE

Base class for Stochastic Differential Equation (SDE):min_num_steps

deepinv.sampling.DiffusionSDE

Reverse-time Diffusion Stochastic Differential Equation defined by

deepinv.sampling.VarianceExplodingDiffusion

Variance-Exploding Stochastic Differential Equation (VE-SDE)

deepinv.sampling.PosteriorDiffusion

Posterior distribution sampling for inverse problems using diffusion models by Reverse-time Stochastic Differential Equation (SDE).

deepinv.sampling.NoisyDataFidelity

Preconditioned data fidelity term for noisy data f(xt,y)=d(A(xt),y).

deepinv.sampling.DPSDataFidelity

The DPS data-fidelity term.

deepinv.sampling.BaseSDESolver

Base class for solving Stochastic Differential Equations (SDEs) from deepinv.sampling.BaseSDE of the form:

deepinv.sampling.EulerSolver

Euler-Maruyama solver for SDEs.

deepinv.sampling.HeunSolver

Heun solver for SDEs.

deepinv.sampling.SDEOutput

A container for storing the output of an SDE solver, that behaves like a dict but allows access with the attribute syntax.

Custom diffusion posterior samplers#

User Guide: refer to Custom diffusion posterior samplers for more information.

deepinv.sampling.DDRM

Denoising Diffusion Restoration Models (DDRM).

deepinv.sampling.DiffPIR

Diffusion PnP Image Restoration (DiffPIR).

deepinv.sampling.DPS

Diffusion Posterior Sampling (DPS).

deepinv.sampling.DiffusionSampler

Turns a diffusion method into a Monte Carlo sampler

Markov Chain Monte Carlo Langevin#

User Guide: refer to Markov Chain Monte Carlo for more information.

deepinv.sampling.MonteCarlo

Base class for Monte Carlo sampling.

deepinv.sampling.ULA

Projected Plug-and-Play Unadjusted Langevin Algorithm.

deepinv.sampling.SKRock

Plug-and-Play SKROCK algorithm.