arXiv Machine Learning

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

Double‑Diffusion is a probabilistic forecasting framework for urban sensor networks that blends a closed‑form graph‑heat prior with a denoising diffusion model. The prior provides a low‑pass, parameter‑free forecast that both serves as the residual target and conditions the denoiser, allowing the reverse diffusion process to start near the prior and refine it over a short chain. Experiments on four real‑world air‑quality and traffic datasets show that Double‑Diffusion achieves the lowest CRPS among probabilistic methods while remaining competitive in point accuracy and requiring far fewer sampling steps than traditional from‑noise diffusion models.

arXiv AI
Oct 28

GAS-PFN: Graph-Aware Spatiotemporal Prior-Data Fitted Networks for Zero-Shot Traffic Forecasting

The paper introduces GAS-PFN, a Graph‑Aware Spatiotemporal Prior‑Data Fitted Network that incorporates road topology into synthetic task priors via random‑walk diffusion and uses a two‑mask Transformer architecture to separately model sensor‑specific temporal dynamics and spatiotemporal interactions. This design allows the model to adapt to new traffic networks without parameter updates, achieving a 34.0% average MSE reduction over TimePFN and a 27.7% reduction over recent industry zero‑shot forecasters across six benchmarks at a 90‑minute horizon.

By Joseph Natter, Long Vu, Yifan Zhang, Bobin Deng, Kun Suo, Rui Wu
arXiv Machine Learning
Sep 3

DynG-Diff: A State-Aware Dynamic Guidance Diffusion Framework for Probabilistic Time Series Forecasting

DynG-Diff is a new diffusion-based framework for probabilistic multivariate time‑series forecasting that addresses the challenge of information heterogeneity across variables. It uses a two‑stage training strategy with an unconditional diffusion backbone and introduces a lightweight state‑aware policy network that dynamically adjusts guidance strength based on real‑time variable reliability. The dynamic guidance is mathematically framed as local precision, allowing the model to focus on high‑confidence variables and suppress anomalous noise, leading to competitive performance and robustness on real‑world benchmarks.

By Zhente Zhang, Zhengwei Ni, Wei Fan
arXiv Machine Learning
Sep 11

RDDMPI: Residual Denoising Diffusion Model for Probabilistic Multivariate Time Series Imputation

RDDMPI introduces a residual denoising diffusion model for multivariate time series imputation. By decomposing the missing signal into a baseline reconstruction and a residual uncertainty component, the method conditions the diffusion process on both the completed signal and its latent representation, using a reliability-aware mechanism to balance baseline influence. Experiments on benchmark datasets show that this approach improves reconstruction accuracy and uncertainty quantification compared to prior diffusion-based methods.

By Ramiro Valdes Jara, David Chapman, Adam Meyers
arXiv AI
Sep 17

Diff-SPORT: Diffusion-based Sensor Placement Optimization and Reconstruction of Turbulent flows in urban environments

Diff-SPORT is a diffusion-based framework that integrates a generative diffusion prior, maximum a posteriori inference, and Shapley-value attribution to achieve high-fidelity reconstruction of turbulent flows and optimal sensor placement in urban environments. By training the diffusion prior once for a domain, it enables non-linear sensor placement and near-real-time flow reconstruction from sparse measurements, outperforming state-of-the-art methods and running orders of magnitude faster than RANS or LES simulations. The approach also generalizes to experimental passive scalar concentration data, demonstrating up to 57% lower reconstruction error than random sensor placement under extreme sparsity and providing compact, physically interpretable sensor configurations.

By Abhijeet Vishwasrao, Sai Bharath Chandra Gutha, Andres Cremades, Klas Wijk, Aakash Patil, H. D. Lim, Christina Vanderwel, Catherine Gorle, Beverley J McKeon, Hossein Azizpour, Ricardo Vinuesa
arXiv Machine Learning
Oct 2

Proper Scoring Rule-based Diffusion for Probabilistic Weather Forecasting

The paper introduces a new diffusion-based approach for probabilistic weather forecasting that incorporates auxiliary conditional denoising tasks. By predicting future states from both clean and corrupted contexts, the model reduces ambiguity and learns conditional distributions across noise levels using a proper scoring rule. Experiments demonstrate improved one‑step forecasting, especially at longer horizons, and better calibration in high‑dimensional global weather scenarios.

By Joonhyeong Park, Giung Nam, Hyungi Lee, Kyunghyun Cho, Byoungwoo Park, Juho Lee
arXiv Machine Learning
Aug 28

STITCH-OPE: Trajectory Stitching with Guided Diffusion for Off-Policy Evaluation

STITCH-OPE is a model‑based generative framework that uses denoising diffusion to perform off‑policy evaluation (OPE) in high‑dimensional, long‑horizon settings. It generates synthetic trajectories for a target policy by guiding a diffusion model trained on behavior data, subtracting the behavior policy’s score to avoid over‑regularization and stitching partial trajectories to extend horizon length. The authors provide theoretical variance‑reduction guarantees and demonstrate improved mean squared error, correlation, and regret on D4RL and OpenAI Gym benchmarks.

By Hossein Goli, Michael Gimelfarb, Nathan Samuel de Lara, Haruki Nishimura, Masha Itkina, Florian Shkurti