Measurement-Consistent Diffusion Methods
Beyond DPS β A Landscape of Consistency Methods
DPS enforces measurement consistency via soft guidance (a gradient step at each diffusion iteration). Several alternative methods enforce consistency through different mechanisms: SVD-based null-space preservation (DDRM), range-null space decomposition (DDNM), manifold projection (MCG), and denoising-diffusion restoration (DiffPIR). Each method offers a different tradeoff between reconstruction quality, computational cost, and assumptions on the forward model .
Definition: Denoising Diffusion Restoration Models (DDRM)
Denoising Diffusion Restoration Models (DDRM)
DDRM exploits the SVD of the forward model to enforce exact data consistency in the measurement subspace while allowing the diffusion model to fill in the null space.
At each reverse step, DDRM decomposes the update into:
- Range-space component: set to match the measurements via
- Null-space component: sampled from the conditional prior via the reverse diffusion process
The result is a reconstruction that satisfies exactly (in the noiseless case), with the prior filling in the missing information.
DDRM requires the SVD of , which is tractable for structured operators (convolution, subsampling, inpainting) but expensive for general sensing matrices. For RF imaging, the sensing matrix is often too large for explicit SVD computation.
Definition: Denoising Diffusion Null-Space Model (DDNM)
Denoising Diffusion Null-Space Model (DDNM)
DDNM is a zero-shot (no fine-tuning) method that enforces data consistency by replacing the range-space components of the Tweedie estimate at each step:
where is the Tweedie estimate from the unconditional diffusion model, is the pseudoinverse, and accounts for measurement noise.
The key insight: the correction only modifies the range space of , leaving the null-space content (generated by the prior) untouched.
Definition: Manifold Constrained Gradients (MCG)
Manifold Constrained Gradients (MCG)
MCG combines the DPS gradient with a hard projection step at each diffusion iteration:
- Compute the DPS guidance gradient (soft consistency)
- Apply a projection onto the measurement-consistent manifold:
The projection ensures exact measurement consistency after each step, while the gradient provides a smooth trajectory on the data manifold.
The projection requires , which may not exist or may be ill-conditioned. In practice, MCG uses a regularised pseudoinverse or applies the projection only in the range space.
Definition: DiffPIR (Diffusion-Based Plug-and-Play Image Restoration)
DiffPIR (Diffusion-Based Plug-and-Play Image Restoration)
DiffPIR integrates diffusion models into the half-quadratic splitting (HQS) framework from Chapter 21:
- Data-fidelity step: solve
- Prior step: run a few reverse diffusion steps starting from to produce
The diffusion model replaces the explicit denoiser in PnP, but operates over multiple noise levels within each HQS iteration. This provides a principled connection between PnP methods (Chapter 21) and diffusion-based reconstruction.
DiffPIR is computationally cheaper than full DPS because it runs only a few diffusion steps per HQS iteration rather than a full -step reverse process. Typical configurations use 15--20 HQS iterations with 5--10 diffusion steps each.
Theorem: Null-Space Preservation Theorem
Let be the SVD of the forward model with (underdetermined system). Any reconstruction consistent with the measurements () can be written as:
where contains the first right singular vectors, spans the null space of , and is arbitrary.
The role of the diffusion prior is to provide an informative distribution over , filling in the -dimensional null space with plausible content.
The measurements determine components of (the range space). The remaining components (the null space) are invisible to the measurements and must be filled by the prior. A stronger prior produces a more informative null-space estimate; a weaker prior produces a blurrier or noisier estimate in those components.
SVD decomposition
Write and partition . Any can be decomposed as .
Impose measurement consistency
, since for the null-space vectors. Therefore , which is uniquely determined, while is free.
Comparison of Measurement-Consistent Diffusion Methods
| Method | Consistency Type | Requires SVD | NFEs | Strengths |
|---|---|---|---|---|
| DPS | Soft (gradient) | No | General ; posterior sampling | |
| DDRM | Exact (null-space) | Yes | Exact consistency; no guidance scale | |
| DDNM | Exact (pseudoinverse) | Yes () | Zero-shot; no fine-tuning | |
| MCG | Soft + hard projection | Yes () | Manifold-aware; strong consistency | |
| DiffPIR | HQS + diffusion prior | No | -- | Faster; connects to PnP framework |
Example: DDRM for Image Inpainting
Consider image inpainting where is a diagonal binary mask ( if pixel is observed, otherwise). Describe the DDRM reconstruction.
SVD of the mask
The mask is already diagonal with singular values . The range space consists of observed pixels; the null space consists of masked pixels.
Range-space component
For observed pixels (): set (the observed value).
Null-space component
For masked pixels (): the diffusion model generates plausible content conditioned on the surrounding observed pixels. Each run of DDRM produces a different plausible fill-in.
Key advantage
DDRM guarantees that observed pixels are exactly preserved ( for ), while the diffusion prior provides a natural-looking completion of the masked region. No guidance scale is needed.
Reconstruction Quality vs. Consistency Strength
Compare the reconstruction quality (PSNR) and measurement consistency () for different methods as a function of the measurement noise level. DPS with strong guidance achieves perfect consistency but may sacrifice image quality; DDRM achieves exact consistency by construction; DiffPIR provides a computationally efficient middle ground.
Parameters
Common Mistake: Hidden Cost of SVD-Based Methods
Mistake:
Assuming DDRM and DDNM are always more efficient than DPS because they avoid backpropagation through the score network.
Correction:
DDRM and DDNM require the SVD of (or at least ), which costs . For large-scale RF imaging problems where with , this SVD is prohibitively expensive. In such cases, DPS (which requires only matrix-vector products and ) is the more practical choice.
Common Mistake: Hard Projection Can Disrupt the Diffusion Process
Mistake:
Applying hard measurement projection at every diffusion step to ensure throughout the reverse process.
Correction:
At intermediate diffusion times, is a noisy version of , and enforcing is not meaningful β the measurements correspond to , not to . Hard projection at intermediate steps can push off the noisy data manifold, causing artefacts or divergence. MCG addresses this by combining soft gradient guidance with a projection applied only in the estimated clean-image space.
Quick Check
In DDRM, the diffusion model's role is to fill in which component of the reconstruction?
The range space of
The null space of
Both the range and null spaces equally
The measurement noise component
The range-space component is determined by the measurements via . The null-space component () is invisible to the measurements and is filled by the diffusion prior.
DDRM (Denoising Diffusion Restoration Models)
A diffusion-based reconstruction method that uses the SVD of the forward model to separate the reconstruction into range-space (determined by measurements) and null-space (filled by the prior) components.
Related: Null Space, Truncated SVD (TSVD)
Key Takeaway
The landscape of measurement-consistent diffusion methods offers a spectrum from soft guidance (DPS) to exact consistency (DDRM/DDNM) to hybrid approaches (MCG, DiffPIR). The choice depends on the structure of : SVD-based methods excel when the SVD is cheap (convolution, masking), while gradient-based methods (DPS) are preferred when is a general large-scale operator β the typical situation in RF imaging.