mlsanityData to 14 Oct 2026

From Drops to Grid: Noise-Aware Spatio-Temporal Neural Process for Rainfall Estimation

arXiv 2605.05912 ↗First posted 7 May 2026cs.LG cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

High-resolution rainfall observations are crucial for weather forecasting, water management, and hazard mitigation. Traditional operational measurements are often biased and low-resolution, limiting their ability to capture local rainfall. Accurate high-resolution rainfall maps require integrating sparse surface observations, yet existing deep learning densification methods are hindered by rainfall's skewed, localized nature, noise, and limited spatio-temporal fusion. We present DropsToGrid, a Neural Process-based method that generates dense rainfall fields by fusing temporal sequences from noisy, irregularly distributed private weather stations with spatial context from radar. Leveraging multi-scale feature extraction, temporal attention, and multi-modal fusion, the model produces stochastic, continuous rainfall estimates and explicitly quantifies uncertainty. Evaluations on real-world datasets demonstrate that DropsToGrid outperforms both operational and deep learning baselines, generating accurate high-resolution rainfall maps with well-calibrated uncertainty, even when only few stations are available and in cross-regional scenarios.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 68 references that are in the map. First posted on arXiv 7 May 2026.

Builds on

18 of 68 references.

References · 18

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  2. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
  3. 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
  4. 2018Conditional Neural Processescited by 426
  5. 2019Attentive Neural Processescited by 283
  6. 2017Deep Learning for Precipitation Nowcasting: A Benchmark and A New Modelcited by 182
  7. 2022Earthformer: Exploring Space-Time Transformers for Earth System Forecastingcited by 152
  8. 2019Convolutional Conditional Neural Processescited by 142
  9. 2022Transformer Neural Processes: Uncertainty-Aware Meta Learning Via Sequence Modelingcited by 127
  10. 2023Deep Learning for Day Forecasts from Sparse Observationscited by 72
  11. 2023DiffCast: A Unified Framework via Residual Diffusion for Precipitation Nowcastingcited by 61
  12. 2024CasCast: Skillful High-resolution Precipitation Nowcasting via Cascaded Modellingcited by 56
6 more
  1. 2020Equivariant Learning of Stochastic Fields: Gaussian Processes and Steerable Conditional Neural Processescited by 29
  2. 2021Convolutional conditional neural processes for local climate downscalingcited by 29
  3. 2024Data driven weather forecasts trained and initialised directly from observationscited by 19
  4. 2024Fuxi-DA: A Generalized Deep Learning Data Assimilation Framework for Assimilating Satellite Observationscited by 18
  5. 2023Practical Equivariances via Relational Conditional Neural Processescited by 13
  6. 2024Neural general circulation models optimized to predict satellite-based precipitation observationscited by 6

Built on by

Not cited yet.