mlsanityData to 7 Oct 2026

Uncertainty-Calibrated Spatiotemporal Field Diffusion with Sparse Supervision

arXiv 2603.04431 ↗First posted 17 Feb 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

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 (rho > 0.7) under severe sparsity.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 42 of its 66 references that are in the map. First posted on arXiv 17 Feb 2026.

Builds on

42 of 66 references.

References · 42

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  4. 2020Denoising Diffusion Implicit Modelscited by 9,655
  5. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
  6. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
  7. 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
  8. 2020Implicit Neural Representations with Periodic Activation Functionscited by 2,214
  9. 2020Rethinking Attention with Performerscited by 1,584
  10. 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 748
  11. 2021Solving Inverse Problems in Medical Imaging with Score-Based Generative Modelscited by 514
  12. 2018Conditional Neural Processescited by 421
30 more
  1. 2018Neural Processescited by 307
  2. 2022Scale-MAE: A Scale-Aware Masked Autoencoder for Multiscale Geospatial Representation Learningcited by 302
  3. 2019Attentive Neural Processescited by 280
  4. 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
  5. 2023Neural Operators for Accelerating Scientific Simulations and Designcited by 218
  6. 2020Spatial-Temporal Fusion Graph Neural Networks for Traffic Flow Forecastingcited by 212
  7. 2022Representing Long-Range Context for Graph Neural Networks with Global Attentioncited by 193
  8. 2020Adversarial Generation of Continuous Imagescited by 135
  9. 2023DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal Forecastingcited by 105
  10. 2023Score-based Diffusion Models in Function Spacecited by 80
  11. 2024On conditional diffusion models for PDE simulationscited by 56
  12. 2025Frequency-Aligned Knowledge Distillation for Lightweight Spatiotemporal Forecastingcited by 53
  13. 2021Neural Stochastic PDEs: Resolution-Invariant Learning of Continuous Spatiotemporal Dynamicscited by 46
  14. 2021Neural Point Process for Learning Spatiotemporal Event Dynamicscited by 42
  15. 2022Neural Implicit Flow: a mesh-agnostic dimensionality reduction paradigm of spatio-temporal datacited by 37
  16. 2023Learning Spatial-Temporal Implicit Neural Representations for Event-Guided Video Super-Resolutioncited by 33
  17. 2021Conditional Local Convolution for Spatio-temporal Meteorological Forecastingcited by 30
  18. 2024Deep Generative Data Assimilation in Multimodal Settingcited by 28
  19. 2025Learning Distributions of Complex Fluid Simulations with Diffusion Graph Networkscited by 24
  20. 2022End-to-end video instance segmentation via spatial-temporal graph neural networkscited by 16
  21. 2022Neur2SP: Neural Two-Stage Stochastic Programmingcited by 16
  22. 2022Contrastive Conditional Neural Processescited by 11
  23. 2024Neural P3M: A Long-Range Interaction Modeling Enhancer for Geometric GNNscited by 9
  24. 2021Super-resolution data assimilationcited by 8
  25. 2024FNP: Fourier Neural Processes for Arbitrary-Resolution Data Assimilationcited by 8
  26. 2025Generating Full-field Evolution of Physical Dynamics from Irregular Sparse Observationscited by 8
  27. 2025SCENT: Robust Spatiotemporal Learning for Continuous Scientific Data via Scalable Conditioned Neural Fieldscited by 3
  28. 2025NeuralDMD: Interpretable Neural Representation of Dynamics from Sparse and Noisy Measurementscited by 2
  29. 2025OmniField: Conditioned Neural Fields for Robust Multimodal Spatiotemporal Learningcited by 1
  30. 2025PMA-Diffusion: A Physics-guided Mask-Aware Diffusion Framework for TSE from Sparse Observationscited by 1

Built on by

Not cited yet.