Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Uncertainty-Calibrated Spatiotemporal Field Diffusion with Sparse Supervision

About

Physical fields are typically observed only at sparse, time-varying sensor locations, making forecasting and reconstruction ill-posed and uncertainty-critical. We present SOLID, a mask-conditioned diffusion framework that learns spatiotemporal dynamics from sparse observations alone: training and evaluation use only observed target locations, requiring no dense fields and no pre-imputation. Unlike prior work that trains on dense reanalysis or simulations and only tests under sparsity, SOLID is trained end-to-end with sparse supervision only. SOLID conditions each denoising step on the measured values and their locations, and introduces a dual-masking objective that (i) emphasizes learning in unobserved void regions while (ii) upweights overlap pixels where inputs and targets provide the most reliable anchors. This strict sparse-conditioning pathway enables posterior sampling of full fields consistent with the measurements, achieving up to an order-of-magnitude improvement in probabilistic error and yielding calibrated uncertainty maps (\r{ho} > 0.7) under severe sparsity.

Kevin Valencia, Xihaier Luo, Shinjae Yoo, David Keetae Park• 2026

Related benchmarks

TaskDatasetResultRank
Spatiotemporal Field ReconstructionNavier-Stokes (Full)
CRPS0.0968
30
Spatiotemporal Field ReconstructionNavier-Stokes 10% Subset
CRPS0.1005
30
Spatiotemporal forecastingAirDelhi AD-B
CRPS25.713
10
Showing 3 of 3 rows

Other info

Follow for update