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 A\mathbf{A}.

Definition:

Denoising Diffusion Restoration Models (DDRM)

DDRM exploits the SVD of the forward model A=UΞ£VH\mathbf{A} = \mathbf{U}\boldsymbol{\Sigma}\mathbf{V}^H 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 Σ†UHy\boldsymbol{\Sigma}^\dagger \mathbf{U}^H \mathbf{y}
  • Null-space component: sampled from the conditional prior via the reverse diffusion process

The result is a reconstruction that satisfies Ax^0=y\mathbf{A}\hat{\mathbf{x}}_0 = \mathbf{y} exactly (in the noiseless case), with the prior filling in the missing information.

DDRM requires the SVD of A\mathbf{A}, 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)

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:

x^0DDNM=x^0βˆ’A†(Ax^0βˆ’y)+A†napprox,\hat{\mathbf{x}}_0^{\text{DDNM}} = \hat{\mathbf{x}}_0 - \mathbf{A}^\dagger(\mathbf{A}\hat{\mathbf{x}}_0 - \mathbf{y}) + \mathbf{A}^\dagger\mathbf{n}_{\text{approx}},

where x^0\hat{\mathbf{x}}_0 is the Tweedie estimate from the unconditional diffusion model, A†\mathbf{A}^\dagger is the pseudoinverse, and napprox\mathbf{n}_{\text{approx}} accounts for measurement noise.

The key insight: the correction A†(Ax^0βˆ’y)\mathbf{A}^\dagger(\mathbf{A}\hat{\mathbf{x}}_0 - \mathbf{y}) only modifies the range space of A\mathbf{A}, leaving the null-space content (generated by the prior) untouched.

Definition:

Manifold Constrained Gradients (MCG)

MCG combines the DPS gradient with a hard projection step at each diffusion iteration:

  1. Compute the DPS guidance gradient (soft consistency)
  2. Apply a projection onto the measurement-consistent manifold: xtβˆ’1←xtβˆ’1βˆ’AH(AAH)βˆ’1(Axtβˆ’1βˆ’y)\mathbf{x}_{t-1} \leftarrow \mathbf{x}_{t-1} - \mathbf{A}^{H}(\mathbf{A}\mathbf{A}^{H})^{-1}(\mathbf{A}\mathbf{x}_{t-1} - \mathbf{y})

The projection ensures exact measurement consistency after each step, while the gradient provides a smooth trajectory on the data manifold.

The projection requires (AAH)βˆ’1(\mathbf{A}\mathbf{A}^{H})^{-1}, 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 integrates diffusion models into the half-quadratic splitting (HQS) framework from Chapter 21:

  1. Data-fidelity step: solve zk=arg⁑min⁑zβ€…β€Š12Οƒn2βˆ₯yβˆ’Azβˆ₯2+ρ2βˆ₯zβˆ’x^kβˆ₯2\mathbf{z}_k = \arg\min_\mathbf{z}\;\frac{1}{2\sigma^2_{n}}\|\mathbf{y} - \mathbf{A}\mathbf{z}\|^2 + \frac{\rho}{2}\|\mathbf{z} - \hat{\mathbf{x}}_k\|^2
  2. Prior step: run a few reverse diffusion steps starting from zk\mathbf{z}_k to produce x^k+1\hat{\mathbf{x}}_{k+1}

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 TT-step reverse process. Typical configurations use 15--20 HQS iterations with 5--10 diffusion steps each.

Theorem: Null-Space Preservation Theorem

Let A=UΞ£VH\mathbf{A} = \mathbf{U}\boldsymbol{\Sigma}\mathbf{V}^H be the SVD of the forward model with rank(A)=r<n\text{rank}(\mathbf{A}) = r < n (underdetermined system). Any reconstruction x^\hat{\mathbf{x}} consistent with the measurements (Ax^=y\mathbf{A}\hat{\mathbf{x}} = \mathbf{y}) can be written as:

x^=VrΞ£rβˆ’1UrHy⏟range-spaceΒ (determinedΒ byΒ data)+VβŠ₯η⏟null-spaceΒ (freeΒ parameter),\hat{\mathbf{x}} = \underbrace{\mathbf{V}_r\boldsymbol{\Sigma}_r^{-1}\mathbf{U}_r^H\mathbf{y}}_{\text{range-space (determined by data)}} + \underbrace{\mathbf{V}_\perp\boldsymbol{\eta}}_{\text{null-space (free parameter)}},

where Vr\mathbf{V}_r contains the first rr right singular vectors, VβŠ₯\mathbf{V}_\perp spans the null space of A\mathbf{A}, and η∈Rnβˆ’r\boldsymbol{\eta} \in \mathbb{R}^{n-r} is arbitrary.

The role of the diffusion prior is to provide an informative distribution over Ξ·\boldsymbol{\eta}, filling in the (nβˆ’r)(n - r)-dimensional null space with plausible content.

The measurements determine rr components of x^\hat{\mathbf{x}} (the range space). The remaining nβˆ’rn - r 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.

Comparison of Measurement-Consistent Diffusion Methods

MethodConsistency TypeRequires SVDNFEsStrengths
DPSSoft (gradient)No∼2T\sim 2TGeneral A\mathbf{A}; posterior sampling
DDRMExact (null-space)Yes∼T\sim TExact consistency; no guidance scale
DDNMExact (pseudoinverse)Yes (A†\mathbf{A}^\dagger)∼T\sim TZero-shot; no fine-tuning
MCGSoft + hard projectionYes (AH\mathbf{A}^{H})∼2T\sim 2TManifold-aware; strong consistency
DiffPIRHQS + diffusion priorNo∼100\sim 100--200200Faster; connects to PnP framework

Example: DDRM for Image Inpainting

Consider image inpainting where A=M\mathbf{A} = \mathbf{M} is a diagonal binary mask (Mii=1M_{ii} = 1 if pixel ii is observed, 00 otherwise). Describe the DDRM reconstruction.

Reconstruction Quality vs. Consistency Strength

Compare the reconstruction quality (PSNR) and measurement consistency (βˆ₯Ax^βˆ’yβˆ₯\|\mathbf{A}\hat{\mathbf{x}} - \mathbf{y}\|) 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
0.05
0.5

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 A\mathbf{A} (or at least A†\mathbf{A}^\dagger), which costs O(min⁑(m,n)2max⁑(m,n))O(\min(m,n)^2\max(m,n)). For large-scale RF imaging problems where A∈CmΓ—n\mathbf{A} \in \mathbb{C}^{m \times n} with m,n>104m, n > 10^4, this SVD is prohibitively expensive. In such cases, DPS (which requires only matrix-vector products Av\mathbf{A}\mathbf{v} and AHu\mathbf{A}^{H}\mathbf{u}) 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 Axt=y\mathbf{A}\mathbf{x}_t = \mathbf{y} throughout the reverse process.

Correction:

At intermediate diffusion times, xt\mathbf{x}_t is a noisy version of x0\mathbf{x}_0, and enforcing Axt=y\mathbf{A}\mathbf{x}_t = \mathbf{y} is not meaningful β€” the measurements correspond to x0\mathbf{x}_0, not to xt\mathbf{x}_t. Hard projection at intermediate steps can push xt\mathbf{x}_t 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 A\mathbf{A}

The null space of A\mathbf{A}

Both the range and null spaces equally

The measurement noise component

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 A\mathbf{A}: SVD-based methods excel when the SVD is cheap (convolution, masking), while gradient-based methods (DPS) are preferred when A\mathbf{A} is a general large-scale operator β€” the typical situation in RF imaging.