WaveDiff: Diverse Inverse & Generative Methods

• WaveDiff is a family of frameworks that recast ill-posed inverse problems into structured intermediate representations (spectral, wavefront, or wavelet) for diverse applications.
• The thermographic approach employs a generalized virtual wave transform to convert diffusive signals into causal, wave-like surrogates, highlighting intrinsic information loss and ill-posedness.
• In astronomy and image synthesis, WaveDiff methods leverage data-driven wavefront recovery and hybrid spectral diffusion to markedly improve reconstruction accuracy and efficiency.

WaveDiff is a polysemous research label rather than a single canonical method. In current arXiv usage, it names at least three technically distinct lines of work: a generalized virtual-wave transform for thermographic inversion, a wavefront-space framework for point spread function modeling and wavefront recovery in astronomy, and several wavelet-domain or hybrid spectral diffusion frameworks for image synthesis, restoration, and ultra-resolution. The common thread is not a shared implementation, but a recurring decision to move an ill-posed problem into a structured intermediate representation—spectral, wavefront, or wavelet—before inversion, denoising, or generation (Zhu et al., 7 Jun 2026).

1. Terminological scope and principal usages

The term is best treated as a family of names attached to unrelated frameworks rather than a single lineage. Confusion often arises because the same label is used for both inverse problems in physics and diffusion-based generative models.

Usage	Core representation	Representative paper(s)
Virtual-wave thermography	Spectral integral operator between diffusive and virtual wave fields	(Zhu et al., 7 Jun 2026)
Astronomical PSF modeling	Wavefront error field with differentiable optical forward model	(Liaudat et al., 2022, Centofanti et al., 1 Jul 2026)
Hybrid spectral image generation	Wavelet sub-bands plus partial Fourier representation	(Kiruluta et al., 4 Apr 2025)
Wavelet-domain image restoration	Wavelet low-frequency diffusion plus explicit high-frequency reconstruction	(Huang et al., 2023)
Remote-sensing ultra-resolution	Sequential conditional wavelet reconstruction with CSP constraint	(Shi et al., 2024)
Ray-based wave optics	Wigner Distribution Function and Wave BSDF	(Cuypers et al., 2011)

A persistent misconception is that “WaveDiff” denotes one architecture portable across domains. The literature does not support that reading. The thermographic, astronomical, optical-rendering, and wavelet-diffusion usages solve different inverse problems, use different state variables, and rely on different operators, despite partial overlap in vocabulary.

2. WaveDiff as a generalized virtual-wave transform

In thermography, WaveDiff is formulated as a generalized virtual wave transform (GVWT) that maps a diffusive thermal field to a mathematically constructed virtual wave field. The diffusive field $T(\mathbf r,t)$ satisfies

$$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$

while an ideal wave field $p(\mathbf r,t)$ would satisfy

$$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$

The virtual wave field $T_{\mathrm{virt}}(\mathbf r,t)$ is defined by imposing the same source $Q$ in the wave equation,

$$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right) T_{\mathrm{virt}}(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$

The construction is explicitly not a physical equivalence; it is a wave-like surrogate for the diffusive signal. The key technical step is an analytic continuation in the complex frequency plane. After Fourier transformation, the heat equation gives

$$\left(\nabla^2 - \sigma^2(\omega)\right)\tilde T(\mathbf r,\omega) = -\frac{1}{\kappa}\tilde Q(\mathbf r,\omega), \qquad \sigma^2(\omega)=\frac{i\omega}{\alpha},$$

with $\Re[\sigma(\omega)]>0$, while the virtual wave field satisfies

$$\left(\nabla^2 + k^2(\omega)\right)\tilde T_{\mathrm{virt}}(\mathbf r,\omega) = -\frac{1}{c^2}\tilde Q(\mathbf r,\omega), \qquad k(\omega)=\frac{\omega}{c}.$$

The resulting spectral correspondence is

$$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$0

In time domain, this becomes a Fredholm integral equation of the first kind,

$$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$1

with kernel

$$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$2

The operator $$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$3 defined by this kernel is described as causal, compact, and a nonstationary low-pass filter. The exponential factor damps rapid temporal variations and high spectral content, and the paper interprets this as the mathematical origin of the intrinsic information loss of diffusion and the fundamental ill-posedness of inverse reconstruction. Because the inverse attempts to undo a compact operator on $$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$4, unconstrained recovery of $$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$5 from $$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$6 is unstable (Zhu et al., 7 Jun 2026).

A major contribution of the framework is to reinterpret common excitation schemes as projections of a single composite operator,

$$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$7

where $$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$8 maps the excitation waveform $$\left(\nabla^2 - \frac{1}{\alpha}\frac{\partial}{\partial t}\right)T(\mathbf r,t) = -\frac{1}{\kappa}Q(\mathbf r,t),$$9 into a virtual wave field,

$p(\mathbf r,t)$0

Under this view, pulse excitation is the “full-space limit,” lock-in excitation is a rank-1 projection onto $p(\mathbf r,t)$1, chirped excitation traces a continuous trajectory in frequency space, coded excitation spans a finite-dimensional subspace $p(\mathbf r,t)$2, and PN excitation acts as a discrete spectral weighting mechanism. The paper’s broader claim is therefore unificatory: pulse, lock-in, chirped, coded, and PN formulations are different projections or samplings of one generalized operator rather than unrelated transforms.

3. WaveDiff as a wavefront-space PSF and wavefront recovery framework

In astronomy, WaveDiff denotes a data-driven point spread function model that shifts the modeling space from pixels to the wavefront error (WFE). The original formulation addresses wide-field, spatially varying, undersampled, noisy, and wavelength-integrated PSFs. Instead of learning the PSF image directly, the model learns a wavefront map $p(\mathbf r,t)$3 and propagates it through a differentiable optical forward model. The observation model is written as

$p(\mathbf r,t)$4

and the pupil-plane field is

$p(\mathbf r,t)$5

The monochromatic PSF is obtained by Fraunhofer diffraction,

$p(\mathbf r,t)$6

and the polychromatic prediction is

$p(\mathbf r,t)$7

The wavefront is parameterized as a hybrid of fixed Zernike modes and learned data-driven features,

$p(\mathbf r,t)$8

This formulation makes wavelength dependence a direct consequence of the $p(\mathbf r,t)$9 phase term and makes super-resolution a consequence of oversampled optical propagation rather than pixel-space hallucination. In the simplified space-telescope setting of the original paper, WaveDiff produced a performance breakthrough relative to the compared data-driven baselines: pixel reconstruction errors decreased 6-fold at observation resolution and 44-fold for a 3x super-resolution, ellipticity errors were reduced at least 20 times, and the size error was reduced more than 250 times, all from noisy broad-band in-focus observations (Liaudat et al., 2022).

A later paper reoriented the framework toward wide-field WFE recovery from in-focus stellar observations. There the WFE field at field position $$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$0 is modeled as

$$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$1

with a parametric Zernike component and learned non-parametric wavefront features. The crucial new element is wavefront feature projection, introduced because pixel-space optimization alone can reproduce observed PSFs while remaining far from the ground-truth wavefront. Since pupil obscuration breaks Zernike orthogonality on the actual aperture, the paper introduces an iterative projection algorithm rather than a naive inner-product projection. In a synthetic test of the projection procedure itself, this reduces WFE reconstruction error from about 10% to below $$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$2. In the full recovery pipeline, the optimization scenario changes in three ways: optimization of only the non-parametric part during each cycle, projection of non-parametric features onto the parametric basis at the end of each cycle, and reinitialization of the non-parametric part after projection. The number of optimization cycles increases from 2 to 12. Quantitatively, the WFE RMSE falls from 29.1 ± 11.9% in the original WaveDiff to 3.4 ± 0.9% in the parametric evaluation, while absolute WFE error falls from 22.9 ± 10.7 nm to 2.7 ± 0.8 nm; LR pixel RMSE remains 0.4 ± 0.2%, and SR pixel RMSE improves to 0.7 ± 0.2% (Centofanti et al., 1 Jul 2026).

The astronomical literature also clarifies a second misconception: excellent PSF reconstruction in pixel space does not imply accurate recovery of the exact ground-truth wavefront, and conversely, improving WFE recovery may require optimization strategies that are not visible from pixel residuals alone.

4. WaveDiff in hybrid spectral diffusion and image restoration

In generative modeling, “WaveDiff” has also been used as shorthand for Wavelet-Fourier-Diffusion, a conditional image-generation framework that replaces pixel-space Gaussian noising with a hybrid spectral corruption process. An image $$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$3 is decomposed by a wavelet transform,

$$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$4

and the low-frequency band is then transformed into the Fourier domain,

$$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$5

The diffusion state is the pair $$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$6, and the forward process uses a corruption operator $$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$7 that masks or attenuates Fourier coefficients in the low-frequency branch and adds noise or dropout to the wavelet high-frequency branches. The reverse process is learned by a conditional U-Net-like network $$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$8,

$$\left(\nabla^2 - \frac{1}{c^2}\frac{\partial^2}{\partial t^2}\right)p(\mathbf r,t) = -\frac{1}{c^2}Q(\mathbf r,t).$$9

with reconstruction by inverse Fourier transform and inverse wavelet transform. The architecture is a U-Net variant with two branches, one for the complex-valued Fourier representation and one for the wavelet high-frequency sub-bands; the Fourier input is treated as real and imaginary channels, and the network uses residual blocks, skip connections, and cross-attention for conditioning. On CIFAR-10, the paper reports FID 2.9 and IS 9.33, compared with FID 3.4 and IS 9.12 for pixel diffusion; on CelebA-HQ $T_{\mathrm{virt}}(\mathbf r,t)$0, WaveDiff reports FID 4.8; on a conditional ImageNet subset at $T_{\mathrm{virt}}(\mathbf r,t)$1, it reports FID 16.7 and IS 124.4 (Kiruluta et al., 4 Apr 2025).

A distinct restoration-oriented line is WaveDM, a wavelet-based diffusion model designed to address the long-time inference bottleneck of diffusion restoration. It applies a 2-level 2D Full Wavelet Packet Transform with the Haar wavelet, yielding tensors in $T_{\mathrm{virt}}(\mathbf r,t)$2, but diffuses only the first three low-frequency bands,

$T_{\mathrm{virt}}(\mathbf r,t)$3

while a High Frequency Refinement Module (HFRM) predicts the remaining 45 high-frequency bands in one pass,

$T_{\mathrm{virt}}(\mathbf r,t)$4

WaveDM also introduces Efficient Conditional Sampling (ECS), which stops DDIM sampling at an intermediate step $T_{\mathrm{virt}}(\mathbf r,t)$5 and directly predicts the clean low-frequency representation, reducing sampling to around 5 steps. The paper reports state-of-the-art restoration quality on twelve benchmark datasets and an efficiency over 100× faster than existing image restoration methods using vanilla diffusion models (Huang et al., 2023).

That restoration backbone is further customized in SemiDDM-Weather, where the paper states that WaveDiff is a wavelet-domain diffusion model and faster to sample than standard diffusion. In that framework, both teacher and student use the same customized backbone, training uses $T_{\mathrm{virt}}(\mathbf r,t)$6 patches, the DWT representation produces four subbands concatenated as $T_{\mathrm{virt}}(\mathbf r,t)$7 tensors, and the paper explicitly states that WaveDiff needs only four sampling steps instead of hundreds. Additional losses—auxiliary reconstruction, perceptual loss, and contrastive consistency for unlabeled learning—are introduced to support teacher-generated pseudo-labels in semi-supervised all-in-one adverse weather removal (Long et al., 2024).

5. Derivative and adjacent formulations

WaveDiff’s wavelet-domain logic has been extended into adjacent nomenclature. In remote sensing, WaveDiffUR defines ultra-magnification super-resolution as solving a conditional diffusion SDE in the wavelet domain. The image is decomposed by 2D-DWT into $T_{\mathrm{virt}}(\mathbf r,t)$8, $T_{\mathrm{virt}}(\mathbf r,t)$9, $Q$0, and $Q$1, and reconstruction is split into sequential low-frequency and high-frequency subproblems. Low-frequency reconstruction is conditioned by a pre-trained SR model through a projection $Q$2, while high-frequency reconstruction uses a Gaussian model centered at an upscaled version of the LR wavelet bands. The full image is reconstructed by

$Q$3

To prevent fixed boundary conditions from degrading very-large-scale reconstructions, the paper introduces the cross-scale pyramid (CSP) constraint, with separate conditions for low-frequency and for the vertical, horizontal, and diagonal bands. The paper reports robustness from $Q$4 to $Q$5 with only about 19.1% average degradation, approximately 1.5× improvement in PSNR and SRE for large-scale UR, and up to 3× improvement at $Q$6 (Shi et al., 2024).

An older, entirely different use of wave-based nomenclature appears in graphics and computational optics. The Wave BSDF (WBSDF) replaces the usual probability-of-scattering interpretation of a BSDF with a wave-optical scattering kernel derived from the Wigner Distribution Function (WDF). The outgoing WDF is written as

$Q$7

where $Q$8 is the surface’s wave-scattering kernel. Because Wigner functions are quasi-probability distributions, the WBSDF may have positive and negative coefficients, allowing destructive interference to be represented without explicit phase tracking. The framework is used for transmissive, reflective, and emissive diffraction phenomena, including multi-bounce diffraction materials, holograms, and high-frequency surfaces, and is designed to integrate with standard ray tracers such as PBRT (Cuypers et al., 2011).

Taken together, these adjacent formulations suggest that “WaveDiff” and related names are often attached to methods that replace raw-domain processing with a representation in which locality, spectral structure, or physical constraints become more tractable. That observation is interpretive rather than nominal: the papers do not claim a unified method family.

6. Cross-cutting themes, limitations, and recurrent misconceptions

Across the surveyed literature, the principal commonality is representational relocation. In thermography, inversion is moved from raw diffusion traces to a virtual wave field governed by a causal compact integral operator. In astronomy, PSF interpolation is moved from pixel space to wavefront space with explicit Fourier optics. In image synthesis and restoration, corruption and denoising are moved from pixels to wavelet or hybrid wavelet-Fourier coordinates. This suggests that WaveDiff-labeled methods are typically responses to ill-posedness rather than merely rebrandings of standard diffusion or inverse pipelines.

The limitations are domain-specific. The thermographic GVWT explicitly encodes intrinsic information loss through a nonstationary low-pass kernel, so excitation design can only partially control recoverability (Zhu et al., 7 Jun 2026). The astronomical WFE-recovery pipeline assumes that the parametric model is expressive enough to contain the true WFE field, uses known stellar SEDs, and still relies on a simplified optical and detector model (Centofanti et al., 1 Jul 2026). Wavelet-Fourier-Diffusion is more complex than pixel diffusion because it must handle complex Fourier data, multiple wavelet sub-bands, and parallel network branches (Kiruluta et al., 4 Apr 2025). WaveDM reduces inference cost but still requires millions of iterations over several days for training (Huang et al., 2023). WaveDiffUR depends on high-quality synchronous LR and reference image pairs and does not fully model degradation variability across sensors or acquisition conditions (Shi et al., 2024).

Three misconceptions recur. First, “WaveDiff” does not denote one universally recognized architecture. Second, wave-like surrogates are not necessarily physical waves: the virtual wave field in thermography is a mathematically constructed surrogate, not a physical equivalence. Third, moving to a physically structured representation does not automatically identify the physically correct latent cause. The astronomical literature makes this explicit: many different WFEs can yield similar in-focus PSFs, so strong pixel-space fidelity and strong wavefront fidelity are not identical objectives.

In that sense, WaveDiff is best understood not as a singular model class, but as a set of research programs that use wavefront, wavelet, or wave-propagation formalisms to reorganize inverse problems and generative modeling tasks. The name is shared; the mathematics, observables, and identifiability structure are not.