Generative Diffusion Surrogates with Analytical Variance Schedule
Abstract
Stochastic transport describes physical systems in which an initially structured distribution spreads under unresolved forcing, scattering, or heterogeneous media. Useful surrogates for such systems should be probabilistic, time-resolved, and able to represent non-Gaussian distributional structure. Generative diffusion models, which corrupt data with Gaussian noise and learn a reverse flow back to structured states, have these properties. Their noise schedules, however, are usually chosen heuristically: image and audio generation---the canonical use cases---provide no physical clock. In transport, by contrast, the variance, or mean-square displacement, is often known from macroscopic theory or empirical scaling even when the full distribution is not. Here we prescribe the forward noising rate as the time derivative of this variance, turning generative time into a calibrated transport clock. The variance path is enforced by construction, while the learned score field represents how non-Gaussian structure inherited from entrance data is smoothed along that path, requiring no intermediate-time physical transport data. For ballistic-to-diffusive transport in turbulent plasmas, the surrogate matches test-particle distributions, reproduces the laboratory-measured variance scale, and tracks the simulated kurtosis evolution without schedule tuning, enabling calibrated emulation and likelihood-based inference.
Versions
Related
- 3D dark matter map of the JWST COSMOS-Web field using diffusion-based generative priors
- Discovery of Interpretable Surrogates via Agentic AI: Application to Gravitational Waves
- Macrocanonical Generator Networks: Data-Efficient Neural Surrogates for Amortized Physics Simulation
- Learned Surrogates of Three-Dimensional Magnetohydrodynamic Turbulence
- Latin hypercube sampling to reduce the variance in Monte Carlo integration: a practical guide and an application to gravitational-wave astronomy
BibTeX
@misc{reichherzer2026generative,
title = {{Generative Diffusion Surrogates with Analytical Variance Schedule}},
author = {Patrick Reichherzer and Gianluca Gregori and David N. Hosking and Subir Sarkar},
howpublished = {arXiv},
year = {2026},
month = sep,
doi = {10.48550/arxiv.2609.01705},
eprint = {2609.01705},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2609.01705},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Generative Diffusion Surrogates with Analytical Variance Schedule
AU - Patrick Reichherzer
AU - Gianluca Gregori
AU - David N. Hosking
AU - Subir Sarkar
PY - 2026
DA - 2026/09/01
PB - arXiv
DO - 10.48550/arxiv.2609.01705
UR - https://arxiv.org/abs/2609.01705
AB - Stochastic transport describes physical systems in which an initially structured distribution spreads under unresolved forcing, scattering, or heterogeneous media. Useful surrogates for such systems should be probabilistic, time-resolved, and able to represent non-Gaussian distributional structure. Generative diffusion models, which corrupt data with Gaussian noise and learn a reverse flow back to structured states, have these properties. Their noise schedules, however, are usually chosen heuristically: image and audio generation---the canonical use cases---provide no physical clock. In transport, by contrast, the variance, or mean-square displacement, is often known from macroscopic theory or empirical scaling even when the full distribution is not. Here we prescribe the forward noising rate as the time derivative of this variance, turning generative time into a calibrated transport clock. The variance path is enforced by construction, while the learned score field represents how non-Gaussian structure inherited from entrance data is smoothed along that path, requiring no intermediate-time physical transport data. For ballistic-to-diffusive transport in turbulent plasmas, the surrogate matches test-particle distributions, reproduces the laboratory-measured variance scale, and tracks the simulated kurtosis evolution without schedule tuning, enabling calibrated emulation and likelihood-based inference.
ER -