mlsanityData to 14 Oct 2026

Double-Diffusion: Balancing Speed, Accuracy, and Uncertainty in Probabilistic Forecasting for Urban Sensor Networks

arXiv 2506.23053 ↗First posted 29 Jun 2025cs.LGcited by 0

On the map: Diffusion models → Flow matching, time series and science

Abstract (arXiv)

Urban sensor networks need forecasts that are accurate, carry useful uncertainty, and refresh fast enough to act on as new readings arrive. These goals conflict: deterministic models give no distribution, while diffusion forecasters model uncertainty but denoise from pure noise over many steps. We present Double-Diffusion, which integrates a closed-form graph-heat prior into a denoising diffusion model. The prior is a parameter-free low-pass forecast over the sensor graph, and it serves two roles: it is the residual target the model generates, and it conditions the denoiser. The reverse process therefore starts near the prior and denoises a short warm-started chain instead of synthesizing from pure noise; the name denotes this composition, a graph diffusion feeding a denoising diffusion. A compact denoiser, DD-Net, is trained as a Denoising Diffusion Probabilistic Model (DDPM) in the Resfusion warm-start formulation, so generation refines the prior over a short truncated chain rather than synthesizing from pure noise; a graph-spectral read-out of the prior residual sets its switchable spatial filter per domain from training data alone. On four real-world air quality and traffic networks, Double-Diffusion attains the best CRPS of all probabilistic methods on every dataset and stays competitive in point accuracy with the strongest baselines, at a fraction of the sampling cost of from-noise diffusion. The code is available at: https://github.com/teddyicare/Double-Diffusion

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 22 of its 47 references that are in the map. First posted on arXiv 29 Jun 2025.

Builds on

22 of 47 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025Elucidated Rolling Diffusion Models for Probabilistic Forecasting of Complex Dynamics24 Jun
  2. 2023Towards Unifying Diffusion Models for Probabilistic Spatio-Temporal Graph Learning26 Oct
  3. 2023PriSTI: A Conditional Diffusion Framework for Spatiotemporal Imputation20 Feb
  4. 2023DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Models31 Jan
  5. 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation7 Jul

Other references · 17

  1. 2020Denoising Diffusion Implicit Modelscited by 9,781
  2. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  3. 2018Neural Ordinary Differential Equationscited by 3,779
  4. 2016Convolutional Neural Networks on Graphs with Fast Localized Spectral Filteringcited by 2,089
  5. 2017Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecastingcited by 1,136
  6. 2017DeepAR: Probabilistic Forecasting with Autoregressive Recurrent Networkscited by 1,013
  7. 2020Adaptive Universal Generalized PageRank Graph Neural Networkcited by 595
  8. 2019GMAN: A Graph Multi-Attention Network for Traffic Predictioncited by 434
  9. 2021Pay Attention to MLPscited by 368
  10. 2023ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shiftingcited by 365
  11. 2019Revisiting Graph Neural Networks: All We Have is Low-Pass Filterscited by 261
  12. 2022AirFormer: Predicting Nationwide Air Quality in China with Transformerscited by 73
5 more
  1. 2020PM2.5-GNN: A Domain Knowledge Enhanced Graph Neural Network For PM2.5 Forecastingcited by 55
  2. 2021Quantifying Uncertainty in Deep Spatiotemporal Forecastingcited by 48
  3. 2024Air Quality Prediction with Physics-Guided Dual Neural ODEs in Open Systemscited by 26
  4. 2023Resfusion: Denoising Diffusion Probabilistic Models for Image Restoration Based on Prior Residual Noisecited by 19
  5. 2024SpecSTG: A Fast Spectral Diffusion Framework for Probabilistic Spatio-Temporal Traffic Forecastingcited by 5

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Collaborative Deterministic-Probabilistic Forecasting for Diverse Spatiotemporal Systems
  2. 2026DiffPTS: Rethinking Diffusion ELBO for Probabilistic Time Series Forecasting
  3. 2025STeP-Diff: Spatio-Temporal Physics-Informed Diffusion Models for Mobile Fine-Grained Pollution Forecasting
  4. 2026Graph Residual Conjugate Diffusion: SNR-Equalized Heat Flow for Graph Signals
  5. 2025Graph-Aware Diffusion for Signal Generation
  6. 2025Unveiling the Power of Noise Priors: Enhancing Diffusion Models for Mobile Traffic Prediction
  7. 2026Universal Diffusion-Based Probabilistic Downscaling
  8. 2026When Denoising Hurts: Rethinking the Terminal Step of Diffusion Time Series Forecasters -- Extended Version