---
title: Neural Diffusion Processes
url: https://www.emergentmind.com/topics/neural-diffusion-processes-ndps
type: topic
---

# Neural Diffusion Processes

Searching arXiv for core and closely related papers on Neural Diffusion Processes.
arXiv search query: "Neural Diffusion Processes"
Neural Diffusion Processes (NDPs) are denoising diffusion generative models for distributions over functions. Rather than modeling images or audio on a fixed grid, an NDP models a stochastic process by learning its finite-dimensional marginals over function values at arbitrary input locations: functions are represented by samples \((\mathbf{x},\mathbf{y})\) at finitely many locations, diffusion is applied to the function values \(\mathbf{y}\), and a neural network learns to reverse the diffusion [2206.03992]. The framework was introduced as a novel approach that learns to sample from a rich distribution over functions through its finite marginals, with architecture-level treatment of stochastic-process symmetries such as exchangeability and input-dimension invariance [2206.03992].

## 1. Conceptual position within stochastic process modeling

In the NDP formulation, the central object is a stochastic process \(f:\mathbb{R}^D \to \mathbb{R}\), observed through finite sets of evaluations. This places NDPs in the same broad problem class as Gaussian Processes (GPs) and Neural Processes (NPs), but with a substantially different generative mechanism. GPs define a stochastic process such that every finite set of evaluations is jointly Gaussian; this gives exact posterior prediction but restricts all finite marginals to be multivariate Gaussian and makes prior design through kernels difficult in complex or high-dimensional settings. NPs use encoder-decoder architectures to model \(p(\mathbf{y}^* \mid \mathbf{x}^*, \mathcal{D})\), but the NDP literature emphasizes that Conditional NPs factorize targets independently, Latent NPs use approximate inference, Attentive NPs often produce jittery samples, and NP predictive distributions are not Bayes-consistent in the GP sense [2206.03992].

NDPs differ by generating samples through diffusion over the joint finite marginal rather than direct decoding. In the original formulation, this allows them to produce correlated samples, better approximate posterior structure, and maintain exchangeability within the sampled set. The reported empirical summary is that NDPs can capture functional distributions close to the true Bayesian posterior, can successfully emulate the behaviour of Gaussian processes, and surpass the performance of neural processes [2206.03992].

A common shorthand in subsequent work is to view NDPs as diffusion probabilistic models over functions. In the wind-power extension, this is stated explicitly: NDPs adapt the idea of diffusion probabilistic models from data generation to function regression, learning arbitrary function distributions via diffusion rather than assuming Gaussian finite-dimensional marginals [2510.03419]. A related later critique describes standard NDPs as essentially conditional DDPMs for function regression: the inputs are used in the reverse model, but the default diffusion itself is not anchored to the input [2508.07220].

## 2. Finite-marginal diffusion formulation

The original NDP construction follows standard denoising diffusion models but applies them to finite samples from a function. For a generic variable \(\mathbf{s}_0\), the forward process is
\[
q(\mathbf{s}_{0:T}) = q(\mathbf{s}_0)\prod_{t=1}^T q(\mathbf{s}_t \mid \mathbf{s}_{t-1}),
\]
with Gaussian corruption
\[
q(\mathbf{s}_t \mid \mathbf{s}_{t-1}) = \mathcal{N}\!\left(\mathbf{s}_t;\sqrt{1-\beta_t}\,\mathbf{s}_{t-1},\beta_t\mathbf{I}\right).
\]
Writing \(\alpha_t = 1-\beta_t\) and \(\bar{\alpha}_t=\prod_{j=1}^t \alpha_j\), the reparameterized form is
\[
\mathbf{s}_t = \sqrt{\bar{\alpha}_t}\,\mathbf{s}_0 + \sqrt{1-\bar{\alpha}_t}\,\boldsymbol{\varepsilon}, \qquad \boldsymbol{\varepsilon}\sim\mathcal{N}(\mathbf{0},\mathbf{I}).
\]
The denoising model learns \(\boldsymbol{\epsilon}_\theta(\mathbf{s}_t,t)\) by minimizing
\[
\mathcal{L}_\theta = \mathbb{E}\!\left[ \|\boldsymbol{\varepsilon}-\boldsymbol{\epsilon}_\theta(\mathbf{s}_t,t)\|^2 \right].
\]
For NDPs, the state is a function sample \(\mathbf{s}_t=(\mathbf{x}_t,\mathbf{y}_t)\), and the default construction keeps \(\mathbf{x}\) fixed while diffusing only \(\mathbf{y}\) [2206.03992].

The forward kernel specialized to function values is
\[
q\!\left(
\begin{bmatrix}
\mathbf{x}_t\\
\mathbf{y}_t
\end{bmatrix}
\Bigg|
\begin{bmatrix}
\mathbf{x}_{t-1}\\
\mathbf{y}_{t-1}
\end{bmatrix}
\right)
=
\mathcal{N}\!\left(\mathbf{y}_t;\sqrt{1-\beta_t}\,\mathbf{y}_{t-1},\beta_t\mathbf{I}\right),
\]
with \(\mathbf{x}_t=\mathbf{x}_{t-1}=\cdots=\mathbf{x}_0\). The reverse kernel is parameterized as
\[
p_\theta\!\left(
\begin{bmatrix}
\mathbf{x}_{t-1}\\
\mathbf{y}_{t-1}
\end{bmatrix}
\Bigg|
\begin{bmatrix}
\mathbf{x}_t\\
\mathbf{y}_t
\end{bmatrix}
\right)
=
\mathcal{N}\!\left(\mathbf{y}_{t-1};\boldsymbol{\mu}_\theta(\mathbf{x}_t,\mathbf{y}_t,t),\tilde{\beta}_t\mathbf{I}\right),
\]
with
\[
\boldsymbol{\mu}_\theta(\mathbf{x}_t,\mathbf{y}_t,t)
=
\frac{1}{\sqrt{\alpha_t}}
\left(
\mathbf{y}_t -
\frac{\beta_t}{\sqrt{1-\bar{\alpha}_t}}\,
\boldsymbol{\epsilon}_\theta(\mathbf{x}_t,\mathbf{y}_t,t)
\right).
\]
The corresponding NDP objective is
\[
\mathcal{L}_\theta =
\mathbb{E}\!\left[
\left\|
\boldsymbol{\varepsilon} -
\boldsymbol{\epsilon}_\theta(\mathbf{x}_0,\mathbf{y}_t,t)
\right\|^2
\right],
\qquad
\mathbf{y}_t = \sqrt{\bar{\alpha}_t}\,\mathbf{y}_0 + \sqrt{1-\bar{\alpha}_t}\,\boldsymbol{\varepsilon}.
\]
This is the function-space analogue of DDPM training [2206.03992].

Sampling proceeds by fixing a set of input locations \(\mathbf{x}_0\), drawing \(\mathbf{y}_T \sim \mathcal{N}(\mathbf{0},\mathbf{I})\), and iterating the learned reverse process until \(\mathbf{y}_0\) is obtained. The result is a sample from a learned distribution over functions evaluated at the specified inputs [2206.03992].

## 3. Architecture, symmetry, and conditional inference

A central architectural contribution of the original NDP paper is the bi-dimensional attention block. Its motivation is explicitly stochastic-process-theoretic: a process model should be exchangeable over datapoint order and invariant to feature order. The model reshapes inputs \((\mathbf{x}_t,\mathbf{y}_t)\) into a tensor
\[
\mathbf{s}_t \in \mathbb{R}^{N\times D\times H},
\]
by replicating outputs across input dimensions and concatenating features. The block
\[
\mathcal{A}_t:\mathbb{R}^{N\times D\times H}\to\mathbb{R}^{N\times D\times H}
\]
applies two multi-head self-attention layers in parallel, one across the feature dimension \(D\) and one across the datapoint dimension \(N\), with residual connections across \(L\) blocks. The update is written as
\[
\mathcal{A}_t^\ell(\mathbf{s}^{\ell-1}_t)
=
\mathcal{A}^{\ell-1}_t +
\sigma\!\left(
\mathrm{MHSA}_d(\mathbf{s}^{\ell-1}_t) +
\mathrm{MHSA}_n(\mathbf{s}^{\ell-1}_t)
\right).
\]
For permutations \(\pi_n \in \Pi_N\) and \(\pi_d \in \Pi_D\), the block satisfies
\[
\pi_d \circ \pi_n \circ \mathcal{A}_t(\mathbf{s})
=
\mathcal{A}_t(\pi_d \circ \pi_n \circ \mathbf{s}),
\]
and the resulting noise predictor inherits the corresponding equivariance property [2206.03992].

Conditional inference is performed with a RePaint-style procedure. Given context \(\mathcal{D}=(\mathbf{x}_0^c,\mathbf{y}_0^c)\) and target inputs \(\mathbf{x}_0^*\), the model samples \(\mathbf{y}_T^* \sim \mathcal{N}(\mathbf{0},\mathbf{I})\), diffuses the context forward at each time step,
\[
\mathbf{y}_t^c \sim \mathcal{N}\!\left(\sqrt{\bar{\alpha}_t}\,\mathbf{y}_0^c,\,(1-\bar{\alpha}_t)\mathbf{I}\right),
\]
combines target and context into a joint finite set, and denoises. The original report notes that repeating the forward/backward context injection about 5 times improves sample quality [2206.03992].

The same paper also introduces a joint model that diffuses both \(\mathbf{x}\) and \(\mathbf{y}\), replacing the output-only objective with
\[
\mathcal{L}_\theta =
\mathbb{E}\!\Big[
\|\boldsymbol{\varepsilon}_x-\boldsymbol{\epsilon}_\theta^x(\mathbf{x}_t,\mathbf{y}_t,t)\|^2 +
\|\boldsymbol{\varepsilon}_y-\boldsymbol{\epsilon}_\theta^y(\mathbf{x}_t,\mathbf{y}_t,t)\|^2
\Big],
\]
with
\[
\mathbf{x}_t = \sqrt{\bar{\alpha}_t}\mathbf{x}_0+\sqrt{1-\bar{\alpha}_t}\boldsymbol{\varepsilon}_x,\qquad
\mathbf{y}_t = \sqrt{\bar{\alpha}_t}\mathbf{y}_0+\sqrt{1-\bar{\alpha}_t}\boldsymbol{\varepsilon}_y.
\]
This enables sampling from \(p(\mathbf{x},\mathbf{y}\mid \mathcal{D})\) and, in the reported experiments, a global-optimization strategy based on \(p(x^*\mid y^*,\mathcal{D})\) [2206.03992].

The consistency discussion in the original paper is notably cautious. NDPs do not guarantee full Kolmogorov consistency for arbitrary sets in the exact mathematical sense, because the neural approximation \(\boldsymbol{\epsilon}_\theta \approx \boldsymbol{\varepsilon}\) breaks exact consistency. However, if the noise predictor were perfect, the reverse process would define consistent finite marginals, and in practice the paper reports that variation across different context/target partitions is small [2206.03992].

## 4. Empirical profile and original use cases

The initial empirical program for NDPs covers synthetic GP emulation, implicit hyperparameter marginalization, non-Gaussian posterior prediction, image regression, and Bayesian optimization. On synthetic regression tasks, the paper trains on GP-sampled data using squared exponential and Matérn-\(\frac52\) kernels in dimensions \(D=1,2,3\), with \(2^{14}\) training paths, context size from \(1\) to \(10D\), target size fixed to 50, and evaluation over 128 test samples. The reported outcome is that NDPs are competitive with Gaussian Neural Processes in \(D=1\), are the most competitive neural baseline in \(D=2,3\), and generally outperform standard NP variants in approximating GP posteriors [2206.03992].

Implicit hyperparameter marginalization is demonstrated on GP samples with a Matérn-\(\frac32\) kernel and a log-normal prior over lengthscale,
\[
\log \mathcal{N}(\log 0.5, \sqrt{0.5}).
\]
With few context points, samples reflect the prior over lengthscales; as more data arrive, the implied lengthscale distribution narrows around the true generating value. Non-Gaussian posterior prediction is illustrated with a step-function family having random jump location \(u\sim \mathcal{U}[-1,1]\),
\[
f(x)=0 \text{ for } x\le u,\qquad f(x)=1 \text{ for } x>u.
\]
The reported result is that NDPs capture a bimodal posterior at \(x=0\), whereas GPs cannot represent this because every finite marginal is Gaussian [2206.03992].

Image regression treats MNIST and CelebA \(32\times 32\) as coordinate-to-pixel regression tasks. In that setting, the architecture is modified by removing the MHSA block over the feature dimension \(D\), since coordinate order matters in the image setting. The reported qualitative pattern is that NDPs learn coherent image structure from sparse context points and that MSE decreases sharply as context size increases [2206.03992].

For Bayesian optimization, NDPs are used as surrogate models on Hartmann 3D, Hartmann 6D, Rastrigin 4D, and Ackley 5D. The paper reports that NDP nearly matches Gaussian Process Regression, beats NP, ANP, and random search, and—unlike GPR—does not need retraining inside the BO loop. This use case is closely tied to the joint input-output diffusion model, which makes it possible to sample promising query points by conditioning on a low target value \(y^*\) [2206.03992].

The implementation profile reported in the appendix is also distinctive: 4 or 5 bi-dimensional attention blocks, hidden size \(H=64\), 8 heads, typically 500 or 1000 diffusion steps, a cosine schedule for \(\beta_t\), Adam, cosine decay with 20-epoch warm-up and max learning rate typically \(10^{-3}\), 250 epochs for most tasks, 500 epochs for hyperparameter marginalization, and 100 epochs for image regression. The same report notes that the model is input-size agnostic: weights do not depend on \(N\) or \(D\), so the same trained model can handle different dataset sizes and dimensions [2206.03992].

## 5. Major extensions and reformulations

Several later works retain the core NDP viewpoint—diffusion models over stochastic processes or finite marginals of functions—while altering the noising process, symmetry assumptions, or task structure.

| Framework | Modification relative to NDPs | Reported emphasis |
|---|---|---|
| NDP | Diffusion over finite marginals of function values | Exchangeability and input-dimension invariance |
| GeomNDP | Kernelized noising toward a geometric Gaussian process and equivariant score network | Scalar fields, vector fields, spherical codomains |
| MT-NDP | Task encoder \(v\in\mathbb{R}^{\kappa}\) with \(\kappa=8\) | Cross-turbine correlations and few-shot adaptation |
| NBP | Input-dependent bridge kernel | Stronger input coupling and endpoint coherence |
| DANP | Diffusion-style noising over target fidelities | Conceptual relative, not the same model class |

Geometric Neural Diffusion Processes extend NDPs to settings with symmetry and non-Euclidean codomains. Their forward finite-marginal noising process is a kernelized multivariate Ornstein–Uhlenbeck SDE,
\[
d Y_t(x) = \tfrac{1}{2}\big(m(x)-Y_t(x)\big)\beta_t\, dt + \beta_t^{1/2} K(x,x)^{1/2}\, dB_t,
\]
with limiting process \(Y_\infty \sim \mathrm{GP}(m,k)\). Equivariance is enforced both in the geometric Gaussian process prior,
\[
m(g\cdot x)=\rho(g)m(x),\qquad
k(g\cdot x,g\cdot x')=\rho(g)k(x,x')\rho(g)^\top,
\]
and in the score network,
\[
s_\theta(t,g\cdot x,\rho(g)y)=\rho(g)s_\theta(t,x,y).
\]
The reported applications include scalar fields, vector fields, and cyclone trajectories on \(S^2\), together with a Langevin-based conditional sampler [2307.05431].

A different line of work argues that standard NDPs suffer from weak input coupling because the forward process is unconditional and the conditioning signal enters only through the denoising network. Neural Bridge Processes replace the standard DDPM kernel
\[
q(y_t\mid y_{t-1})
=
\mathcal{N}\!\left(y_t;\sqrt{1-\beta_t}\,y_{t-1},\beta_t I\right)
\]
with the input-dependent bridge kernel
\[
q(y_t\mid y_{t-1},x)= \mathcal{N}\!\left(y_t;\sqrt{1-\beta_t}\,y_{t-1}+\gamma_t x,\beta_t I\right),
\]
and correspondingly
\[
y_t=\sqrt{\bar\alpha_t}y_0+\bar\gamma_t x+\sqrt{1-\bar\alpha_t}\epsilon.
\]
The paper states that this gives stronger gradient signals, guarantees endpoint coherence, and improves performance on synthetic regression, EEG signal regression, and image regression tasks. It also summarizes the comparison as follows: for NDP, endpoint match is implicit and path consistency is weak; for NBP, endpoint match and path consistency are both marked as present [2508.07220].

The multi-task extension for wind-power forecasting keeps the NDP backbone but introduces a task encoder that maps context information to a task embedding \(v\in\mathbb{R}^{\kappa}\), with \(\kappa=8\) in the reported experiments. The context embeddings are averaged into \(\bar v\), replicated across target inputs, and concatenated to the original covariates. On SCADA data from six Senvion MM92 turbines at Kelmarsh wind farm, spanning January 2016 to July 2021 at 10-minute resolution with over 1.7 million records and 110 variables, the paper reports that NDP-based models provide better calibration than a GP baseline and that MT-NDP improves both point accuracy and calibration for a more distinctive target turbine. On turbine 1, GP coverage error is reported as \(18.075\%\) versus \(4.584\%\) for a 1D single-task NDP; on turbine 6 with 50 context points, 5D MT-NDP gives MAE \(47.268\), RMSE \(65.636\), and CE \(5.691\%\), compared with 5D ST-NDP at MAE \(49.676\), RMSE \(68.465\), and CE \(7.376\%\) [2510.03419].

Diffusion-Augmented Neural Processes occupy a nearby but explicitly non-identical position. That work describes DANPs as not the same as NDPs: NDPs are generative diffusion models whose denoising process is parameterized by a Neural Process, whereas DANPs are Neural Process models for direct conditional prediction using a diffusion-style auxiliary noising scheme over target fidelities. The relationship is therefore conceptual rather than architectural [2311.09848].

## 6. Terminological boundaries, misconceptions, and adjacent literatures

The phrase “neural diffusion” is used across several research strands that are not Neural Diffusion Processes in the function-distribution sense. This terminological overlap is a recurrent source of confusion.

“Neural Network Diffusion” generates high-performing neural network parameters by training an autoencoder on subsets of trained weights and then learning a latent diffusion model over the resulting codes. The reported pipeline is explicitly parameter generation rather than stochastic-process regression, and the method usually synthesizes only selected normalization parameters rather than all weights [2402.13144]. “HyperDiffusion” likewise operates in parameter space, training a diffusion model directly on flattened MLP weights that encode per-instance implicit neural fields; its target is unconditional generative modeling of 3D shapes and 4D mesh animations through synthesized neural-field parameters [2303.17015]. These methods are diffusion models over neural representations, but not NDPs as introduced in [2206.03992].

Other papers use “diffusion” in still different senses. “Dynamic Tensor Decomposition via Neural Diffusion-Reaction Processes” proposes a continuous-time latent embedding model whose core dynamics are
\[
\frac{\partial U(t)}{\partial t} = (\mathcal{W}-\mathcal{A})U(t) + F(U(t),t),
\]
combining graph diffusion with neural reaction terms for dynamic tensor decomposition [2310.19666]. “Infinitely deep neural networks as diffusion processes” studies a depth limit of residual networks and derives convergence in law to an SDE under \(O(\Delta t)\) drift and \(O(\sqrt{\Delta t})\) random increments, establishing a link between infinitely deep residual networks and diffusion processes rather than introducing a diffusion model over function samples [1905.11065]. “Neural McKean-Vlasov Processes” are described as a closely related alternative framework and partial precursor, with explicit law dependence \(p_t\) in the drift of an MV-SDE; they are adjacent to NDPs but specialized to interacting-particle and mean-field settings [2404.09402]. An earlier physics-informed line represents the evolving pdf or characteristic function of diffusion processes with neural networks constrained by Fokker–Planck or characteristic-function equations, and is explicitly described as not an NDP in the modern generative-model sense [2001.05437].

The original NDP literature also records substantive limitations. More diffusion steps improve sample quality but slow inference; performance degrades when test inputs are outside the training range; full Kolmogorov consistency is not guaranteed in the general mathematical sense; and, in later critiques, the default unconditional forward process is said to produce weak input coupling and semantic mismatch at the diffusion endpoint [2206.03992]. The subsequent appearance of bridge-conditioned, geometric, and multi-task variants suggests that the principal active directions in the area concern stronger conditioning, stricter inductive structure, and better calibrated posterior sampling rather than abandonment of the finite-marginal diffusion paradigm itself [2508.07220].

Source: https://www.emergentmind.com/topics/neural-diffusion-processes-ndps