Spatial Prediction with Guided Diffusion
Abstract
Spatial prediction is a critically important task in data analyses that involve spatially or spatio-temporally-referenced data. It is commonly carried out using hierarchical models, which comprise a process model describing the latent process of interest, and a data model describing how the process is observed. The target quantity is the predictive distribution, which is the conditional distribution of the latent spatial process given the observed data. Unfortunately, the predictive distribution is rarely available in closed form and can be difficult to sample from. Here, we show how recent developments in guided diffusion modelling from machine learning can address a wide class of problems in spatial prediction when the likelihood function derived from the data model is known and differentiable, possibly after relaxation. To show the versatility of the approach, we apply it to both Gaussian and non-Gaussian process models and a variety of data models that describe direct partial observations, linear and areal aggregates, counts, right censoring, and area-exceedance functionals. Where possible, we validate the framework against exact and Markov-chain-Monte-Carlo (MCMC)-derived predictive distributions. We demonstrate our approach on several simulation studies and in a controlled experiment of methane-flux inversion over the UK and Ireland. We show that guided diffusion can sample from the correct predictive distribution in a variety of settings, that it can be applied to models where MCMC is intractable, and that it scales well with the dimensionality of the latent process.
Related
- Stochastic fractional diffusion equations with spatial Gaussian noise in the Stratonovich regime
- Theoretical Guarantees for SMC-Guided Diffusion Sampling
- MGD: Moment Guided Diffusion for Maximum Entropy Generation
- LiftDiff: Guided Discrete Diffusion for Quantum Code Construction
- Denoising as Projection: Constrained Optimization with Gradient-Guided Diffusion
BibTeX
@misc{zammitmangion2026spatial,
title = {{Spatial Prediction with Guided Diffusion}},
author = {Andrew Zammit-Mangion and Lachlan Astfalck and Raphaël Huser and Christopher Nemeth and Henry B. Moss},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.12967},
eprint = {2610.12967},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.12967},
}
RIS
TY - UNPB
TI - Spatial Prediction with Guided Diffusion
AU - Andrew Zammit-Mangion
AU - Lachlan Astfalck
AU - Raphaël Huser
AU - Christopher Nemeth
AU - Henry B. Moss
PY - 2026
DA - 2026/10/09
PB - arXiv
DO - 10.48550/arxiv.2610.12967
UR - https://arxiv.org/abs/2610.12967
AB - Spatial prediction is a critically important task in data analyses that involve spatially or spatio-temporally-referenced data. It is commonly carried out using hierarchical models, which comprise a process model describing the latent process of interest, and a data model describing how the process is observed. The target quantity is the predictive distribution, which is the conditional distribution of the latent spatial process given the observed data. Unfortunately, the predictive distribution is rarely available in closed form and can be difficult to sample from. Here, we show how recent developments in guided diffusion modelling from machine learning can address a wide class of problems in spatial prediction when the likelihood function derived from the data model is known and differentiable, possibly after relaxation. To show the versatility of the approach, we apply it to both Gaussian and non-Gaussian process models and a variety of data models that describe direct partial observations, linear and areal aggregates, counts, right censoring, and area-exceedance functionals. Where possible, we validate the framework against exact and Markov-chain-Monte-Carlo (MCMC)-derived predictive distributions. We demonstrate our approach on several simulation studies and in a controlled experiment of methane-flux inversion over the UK and Ireland. We show that guided diffusion can sample from the correct predictive distribution in a variety of settings, that it can be applied to models where MCMC is intractable, and that it scales well with the dimensionality of the latent process.
ER -