From Drops to Grid: Noise-Aware Spatio-Temporal Neural Process for Rainfall Estimation
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Rainfall estimation | Gridded Rainfall Products (test) | CSI48.9 | 30 | |
| Rainfall estimation | Denmark PWS holdout 2025 | CSI45.1 | 8 | |
| Rainfall estimation | Denmark SYNOP Stations 2025 | CSI47.2 | 7 | |
| Rainfall estimation | Europe PWS holdout 2025 | CSI47.6 | 7 | |
| Rainfall estimation | SYNOP Stations Europe 2025 | CSI40 | 7 | |
| Rainfall estimation | PWS (holdout) | CSI53.2 | 6 | |
| Rainfall estimation | SYNOP stations research-quality | CSI0.551 | 5 | |
| Rainfall estimation | SYNOP Stations | CSI55.1 | 5 |