STAMP-GAN: A Spatiotemporal Attention-Modulated Generative Adversarial Network for Precipitation Nowcasting
Highlights
- STAMP-GAN improves regional hourly precipitation sequence prediction.
- Attention-modulated ConvLSTM captures multiscale spatiotemporal evolution.
- Dual-branch temporal discrimination strengthens long-lead forecast structure.
- Hybrid intensity-aware losses improve heavy-precipitation representation.
Abstract
1. Introduction
2. Materials and Methods
2.1. Study Area
2.2. Datasets
2.2.1. ERA5 Reanalysis Dataset
2.2.2. Station Observation Dataset (CMA-S)
2.2.3. Digital Elevation Model Data
2.3. Data Preprocessing and Sample Construction
2.4. Problem Formulation
2.5. Overall Architecture of STAMP-GAN
2.6. STAMP-Net Generator
2.7. AM-ConvLSTM Temporal Evolution Module
2.8. Dual-Branch Temporal PatchGAN Discriminator
2.9. Loss Functions
2.10. Experimental Setup
2.10.1. Computational Configuration
2.10.2. Baseline Adaptation and Fair-Comparison Protocol
2.10.3. Robustness Experiment for the Last-Three-Frame Discriminator
2.11. Evaluation Metrics
3. Results
3.1. Ablation Study
3.1.1. Progressive Component Ablation
3.1.2. Removal-Based Component Ablation
3.1.3. Multi-Seed Robustness Analysis of the Last-Three-Frame Discriminator
3.1.4. Ablation of DEM Usage Strategies
3.2. Overall Comparison on the ERA5 Dataset
Categorical Verification Results on ERA5
3.3. Overall Comparison on the CMA-S Dataset
3.3.1. Categorical Verification Results
Station-Proximity Sensitivity Analysis for High-Intensity Precipitation
3.3.2. Continuous Verification Results
3.3.3. Lead-Time-Resolved Results
3.4. Neighborhood-Based Spatial Verification
3.5. Case Studies
4. Discussion
4.1. Summary and Interpretation of the Main Findings
4.2. Contributions of the Model Components
4.3. Role of DEM and Terrain Information
4.4. Comparison with Existing Approaches
4.5. Practical Implications
4.6. Limitations and Future Work
5. Conclusions
6. Implementation and Robustness Analyses
6.1. Dataset Split and Event Statistics
6.2. Baseline Implementation Details
6.3. Robustness Analysis of the Last-Three-Frame Discriminator
6.4. CMA-S High-Threshold Event Counts
6.5. Paired Block-Bootstrap Uncertainty Analysis
Supplementary Materials
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Wilson, J.W.; Crook, N.A.; Mueller, C.K.; Sun, J.; Dixon, M. Nowcasting thunderstorms: A status report. Bull. Am. Meteorol. Soc. 1998, 79, 2079–2099. [Google Scholar] [CrossRef] [Scilit]
- Zhang, Y.; Long, M.; Chen, K.; Xing, L.; Jin, R.; Jordan, M.I.; Wang, J. Skilful nowcasting of extreme precipitation with NowcastNet. Nature 2023, 619, 526–532. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Wang, Z.; Yao, C.; Dong, J.; Yang, H. Precipitation characteristic and urban flooding influence of “7·20” extreme rainstorm in Zhengzhou. J. Hohai Univ. (Nat. Sci.) 2022, 50, 17–22. (In Chinese) [Google Scholar] [CrossRef]
- Yu, P. Temporal and spatial distribution characteristics of short-time heavy rainfall in Zhejiang Province. Clim. Environ. Res. 2022, 27, 397–407. (In Chinese) [Google Scholar] [CrossRef]
- Ayzel, G.; Heistermann, M.; Winterrath, T. Optical flow models as an open benchmark for radar-based precipitation nowcasting (rainymotion v0.1). Geosci. Model Dev. 2019, 12, 1387–1402. [Google Scholar] [CrossRef] [Scilit]
- Bowler, N.E.; Pierce, C.E.; Seed, A.W. STEPS: A probabilistic precipitation forecasting scheme which merges an extrapolation nowcast with downscaled NWP. Q. J. R. Meteorol. Soc. 2006, 132, 2127–2155. [Google Scholar] [CrossRef] [Scilit]
- Pulkkinen, S.; Nerini, D.; Pérez Hortal, A.A.; Velasco-Forero, C.; Seed, A.; Germann, U.; Foresti, L. Pysteps: An open-source Python library for probabilistic precipitation nowcasting (v1.0). Geosci. Model Dev. 2019, 12, 4185–4219. [Google Scholar] [CrossRef] [Scilit]
- Naz, F.; She, L.; Sinan, M.; Shao, J. Enhancing radar echo extrapolation by ConvLSTM2D for precipitation nowcasting. Sensors 2024, 24, 459. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Sun, J.; Xue, M.; Wilson, J.W.; Zawadzki, I.; Ballard, S.P.; Onvlee-Hooimeyer, J.; Joe, P.; Barker, D.M.; Li, P.W.; Golding, B.; et al. Use of NWP for nowcasting convective precipitation: Recent progress and challenges. Bull. Am. Meteorol. Soc. 2014, 95, 409–426. [Google Scholar] [CrossRef] [Scilit]
- Ye, Y.; Gao, F.; Cheng, W.; Liu, C.; Zhang, S. MSSTNet: A multi-scale spatiotemporal prediction neural network for precipitation nowcasting. Remote Sens. 2023, 15, 137. [Google Scholar] [CrossRef] [Scilit]
- Niu, D.; Zang, Z.; Huang, J.; Xu, L.; Che, H.; Tang, Y. Two-stage spatiotemporal context refinement network for precipitation nowcasting. Remote Sens. 2021, 13, 4285. [Google Scholar] [CrossRef] [Scilit]
- Shi, X.; Chen, Z.; Wang, H.; Yeung, D.Y.; Wong, W.K.; Woo, W.C. Convolutional LSTM network: A machine learning approach for precipitation nowcasting. In Proceedings of the Advances in Neural Information Processing Systems; Curran Associates: Red Hook, NY, USA, 2015; Volume 28. [Google Scholar]
- Shi, X.; Gao, Z.; Lausen, L.; Wang, H.; Yeung, D.Y.; Wong, W.K.; Woo, W.C. Deep learning for precipitation nowcasting: A benchmark and a new model. In Proceedings of the Advances in Neural Information Processing Systems; Curran Associates: Red Hook, NY, USA, 2017; Volume 30. [Google Scholar]
- Zhou, Z.; Siddiquee, M.M.R.; Tajbakhsh, N.; Liang, J. UNet++: A nested U-Net architecture for medical image segmentation. In Proceedings of the Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support; Springer: Berlin/Heidelberg, Germany, 2018; pp. 3–11. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Le Guen, V.; Thome, N. Disentangling physical dynamics from unknown factors for unsupervised video prediction. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA, 13–19 June 2020; pp. 11471–11481. [Google Scholar] [CrossRef] [Scilit]
- Liu, Z.; Lin, Y.; Cao, Y.; Hu, H.; Wei, Y.; Zhang, Z.; Lin, S.; Guo, B. Swin Transformer: Hierarchical vision transformer using shifted windows. In Proceedings of the IEEE/CVF International Conference on Computer Vision, Montreal, QC, Canada, 10–17 October 2021; pp. 9992–10002. [Google Scholar] [CrossRef] [Scilit]
- Ravuri, S.; Kavukcuoglu, K.; Willson, M.; Kangin, D.; Lam, R.; Mirowski, P.; Fitzsimons, M.; Athanassiadou, M.; Kashem, S.; Madge, S.; et al. Skilful precipitation nowcasting using deep generative models of radar. Nature 2021, 597, 672–677. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Klemmer, K.; Xu, T.; Acciaio, B.; Neill, D.B. SPATE-GAN: Improved generative modeling of dynamic spatio-temporal patterns with an autoregressive embedding loss. Proc. AAAI Conf. Artif. Intell. 2022, 36, 4523–4531. [Google Scholar] [CrossRef] [Scilit]
- Glawion, L.; Polz, J.; Kunstmann, H.; Fersch, B.; Chwala, C. Global spatio-temporal ERA5 precipitation downscaling to km and sub-hourly scale using generative AI. npj Clim. Atmos. Sci. 2025, 8, 219. [Google Scholar] [CrossRef] [Scilit]
- Gao, Z.; Shi, X.; Han, B.; Wang, H.; Jin, X.; Liang, D.; Yeung, D.Y. PreDiff: Precipitation nowcasting with latent diffusion models. Adv. Neural Inf. Process. Syst. 2023, 36, 78621–78656. [Google Scholar] [CrossRef] [Scilit]
- Asperti, A.; Merizzi, F.; Paparella, A.; Pedrazzi, G.; Angelinelli, M.; Colamonaco, S. Precipitation nowcasting with generative diffusion models. Appl. Intell. 2025, 55, 187. [Google Scholar] [CrossRef] [Scilit]
- Lam, R.; Sanchez-Gonzalez, A.; Willson, M.; Wirnsberger, P.; Fortunato, M.; Alet, F.; Ravuri, S.; Ewalds, T.; Eaton-Rosen, Z.; Hu, W.; et al. Learning skillful medium-range global weather forecasting. Science 2023, 382, 1416–1421. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Andrychowicz, M.; Espeholt, L.; Li, D.; Merchant, S.; Merose, A.; Zyda, F.; Agrawal, S.; Kalchbrenner, N. Deep learning for day forecasts from sparse observations. arXiv 2023, arXiv:2306.06079. [Google Scholar]
- Fang, Y.; Ma, H.; Cao, L. Analysis on typical and atypical spatiotemporal characteristics of Mei-yu precipitation over Zhejiang Province in 2024. Torrential Rain Disasters 2025, 45, 576–587. (In Chinese) [Google Scholar] [CrossRef]
- Hersbach, H.; Bell, B.; Berrisford, P.; Hirahara, S.; Horányi, A.; Muñoz-Sabater, J.; Nicolas, J.; Peubey, C.; Radu, R.; Schepers, D.; et al. The ERA5 global reanalysis. Q. J. R. Meteorol. Soc. 2020, 146, 1999–2049. [Google Scholar] [CrossRef] [Scilit]
- Shepard, D. A two-dimensional interpolation function for irregularly-spaced data. In Proceedings of the 1968 23rd ACM National Conference; Association for Computing Machinery: New York, NY, USA, 1968; pp. 517–524. [Google Scholar] [CrossRef] [Scilit]
- NASA JPL. NASA Shuttle Radar Topography Mission Global 1 Arc Second. Available online: https://www.earthdata.nasa.gov/data/catalog/lpcloud-srtmgl1-003 (accessed on 1 June 2025). [CrossRef] [Scilit]
- Wang, Q.; Wu, B.; Zhu, P.; Li, P.; Zuo, W.; Hu, Q. ECA-Net: Efficient channel attention for deep convolutional neural networks. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA, 13–19 June 2020; pp. 11534–11542. [Google Scholar]
- Lin, T.Y.; Goyal, P.; Girshick, R.; He, K.; Dollár, P. Focal loss for dense object detection. In Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy, 22–29 October 2017; pp. 2980–2988. [Google Scholar]
- Curcio, F.; Castro, P.; Fonseca, A.; Castro, R.; Franco, R.; Ogasawara, E.; Stepanenko, V.; Porto, F.; Ferro, M.; Bezerra, E. Towards a spatiotemporal fusion approach to precipitation nowcasting. arXiv 2025, arXiv:2505.19258. [Google Scholar] [CrossRef] [Scilit]
- Brooks, H.E.; Flora, M.L.; Baldwin, M.E. A rose by any other name: On basic scores from the 2 × 2 table and the plethora of names attached to them. Artif. Intell. Earth Syst. 2024, 3, e230104. [Google Scholar] [CrossRef] [Scilit]
- Willmott, C.J.; Matsuura, K. Advantages of the mean absolute error (MAE) over the root mean square error (RMSE) in assessing average model performance. Clim. Res. 2005, 30, 79–82. [Google Scholar] [CrossRef] [Scilit]
- Gurjar, S.; Goel, N.K.; Arora, M.; Goel, M.K. Short-range monsoon rainfall forecasting using multi-model ensembles and analysis of threshold-based rainfall verification. Theor. Appl. Climatol. 2026, 157, 155. [Google Scholar] [CrossRef] [Scilit]
- Nurmi, P. Recommendations on the Verification of Local Weather Forecasts; ECMWF Technical Memorandum 430; European Centre for Medium-Range Weather Forecasts: Reading, UK, 2003. [Google Scholar]
- Gupta, H.V.; Kling, H.; Yilmaz, K.K.; Martinez, G.F. Decomposition of the mean squared error and NSE performance criteria: Implications for improving hydrological modelling. J. Hydrol. 2009, 377, 80–91. [Google Scholar] [CrossRef] [Scilit]

















| Dataset | Variable | Physical Meaning | Temporal Resolution | Spatial Resolution |
|---|---|---|---|---|
| ERA5 | tp | Total precipitation | 1 h | 0.25° |
| d2m | 2 m dewpoint temperature | 1 h | 0.25° | |
| u10, v10 | 10 m wind components | 1 h | 0.25° | |
| msl | Mean sea-level pressure | 1 h | 0.25° | |
| tcc | Total cloud cover | 1 h | 0.25° | |
| rh* | Relative humidity | 1 h | 0.25° | |
| DEM | Terrain elevation | Static | 0.00028° | |
| CMA-S | PRE_1h | Hourly precipitation | 1 h | 0.05° |
| TEM | Air temperature | 1 h | 0.05° | |
| RHU | Relative humidity | 1 h | 0.05° | |
| WIN_S_INST | Wind speed | 1 h | 0.05° | |
| WIN_D_INST | Wind direction | 1 h | 0.05° | |
| PRS | Atmospheric pressure | 1 h | 0.05° | |
| DPT | Dewpoint temperature | 1 h | 0.05° | |
| DEM | Terrain elevation | Static | 0.00028° |
| Loss Component | Symbol | Weight |
|---|---|---|
| Weighted L1 regression | 1.0 | |
| Multi-threshold focal classification | 1.0 | |
| Occurrence-gate supervision | 0.5 | |
| Global adversarial (WGAN-GP) | 0.05 | |
| Last-3-frame adversarial (WGAN-GP) | 0.08 | |
| Temporal consistency | 0.1 | |
| Spatial-gradient consistency | 0.05 |
| Index | Configuration | CSI | POD | FAR | MBE | MAE | RMSE | CORR | KGE |
|---|---|---|---|---|---|---|---|---|---|
| 1 | Baseline | 0.5998 | 0.5359 | 0.3597 | −0.3660 | 0.4799 | 1.8165 | 0.4855 | 0.3804 |
| 2 | +STAMP-Net | 0.6193 | 0.8406 | 0.2984 | 0.0679 | 0.4502 | 1.1459 | 0.5431 | 0.3427 |
| 3 | +AM-ConvLSTM | 0.6232 | 0.8442 | 0.2961 | 0.0431 | 0.4365 | 1.2828 | 0.5558 | 0.4535 |
| 4 | +Dual-branch discriminator | 0.6333 | 0.8689 | 0.2560 | −0.0258 | 0.4274 | 1.0587 | 0.5449 | 0.4447 |
| 5 | +Hybrid loss | 0.6379 | 0.9350 | 0.2495 | 0.0026 | 0.3892 | 0.9260 | 0.6180 | 0.5866 |
| Configuration | CSI | POD | FAR | MBE | MAE | RMSE | CORR | KGE |
|---|---|---|---|---|---|---|---|---|
| Full STAMP-GAN | 0.6379 | 0.9350 | 0.2495 | 0.0026 | 0.3892 | 0.9260 | 0.6180 | 0.5866 |
| w/o dual-branch discriminator | 0.6272 | 0.8850 | 0.2717 | 0.0468 | 0.4256 | 1.0030 | 0.6092 | 0.5290 |
| w/o STAMP-Net | 0.6193 | 0.8276 | 0.2677 | −0.2748 | 0.3746 | 1.9649 | 0.5455 | 0.4638 |
| w/o AM-ConvLSTM | 0.6273 | 0.8444 | 0.2878 | −0.0140 | 0.4268 | 1.0230 | 0.5448 | 0.5272 |
| w/o hybrid loss | 0.6194 | 0.8907 | 0.3313 | 0.0394 | 0.4322 | 0.9885 | 0.5661 | 0.5362 |
| Forecast Range | Model | CSI at 0.5 mm h−1 ↑ | CSI at 5 mm h−1 ↑ | CSI at 10 mm h−1 ↑ |
|---|---|---|---|---|
| T1–T6 | STAMP-GAN w/o | 0.6294 ± 0.0052 | 0.2568 ± 0.0095 | 0.1304 ± 0.0062 |
| T1–T6 | Full STAMP-GAN | 0.6343 ± 0.0040 | 0.2931 ± 0.0070 | 0.1455 ± 0.0084 |
| T4–T6 | STAMP-GAN w/o | 0.5777 ± 0.0056 | 0.1699 ± 0.0056 | 0.0383 ± 0.0059 |
| T4–T6 | Full STAMP-GAN | 0.5811 ± 0.0068 | 0.1995 ± 0.0081 | 0.0604 ± 0.0067 |
| Configuration | CSI | POD | FAR | MBE | MAE | RMSE | CORR | KGE |
|---|---|---|---|---|---|---|---|---|
| Original DEM-conditioning configuration | 0.6267 | 0.9284 | 0.2695 | 0.0450 | 0.4331 | 0.9950 | 0.6021 | 0.5557 |
| Revised model | 0.6379 | 0.9350 | 0.2495 | 0.0026 | 0.3892 | 0.9260 | 0.6180 | 0.5866 |
| Revised model without DEM input | 0.6175 | 0.9177 | 0.2536 | 0.0450 | 0.4086 | 0.9496 | 0.5990 | 0.5726 |
| Index | Models | Year | CSI ↑ | POD ↑ | FAR ↓ | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| 0.5 | 5 | 10 | 0.5 | 5 | 10 | 0.5 | 5 | 10 | |||
| 1 | ConvLSTM | 2015 | 0.5133 | 0.2329 | 0.1075 | 0.8639 | 0.3757 | 0.1322 | 0.4602 | 0.5870 | 0.7651 |
| 2 | TrajGRU | 2017 | 0.5428 | 0.2509 | 0.1115 | 0.8564 | 0.3498 | 0.1358 | 0.4109 | 0.5724 | 0.7524 |
| 3 | U-Net++ | 2018 | 0.5554 | 0.2607 | 0.1026 | 0.8874 | 0.3755 | 0.1186 | 0.4024 | 0.5738 | 0.7319 |
| 4 | PhyDNet | 2020 | 0.5446 | 0.2777 | 0.1140 | 0.9175 | 0.4574 | 0.2294 | 0.4298 | 0.6067 | 0.7404 |
| 5 | Swin Transformer | 2021 | 0.5346 | 0.2731 | 0.1227 | 0.9237 | 0.4549 | 0.1774 | 0.4423 | 0.6118 | 0.7728 |
| 6 | DGMR | 2021 | 0.5987 | 0.2314 | 0.0955 | 0.7577 | 0.3047 | 0.1109 | 0.3320 | 0.5563 | 0.7455 |
| 7 | MetNet3-R | 2023 | 0.5340 | 0.2352 | 0.1012 | 0.8688 | 0.3049 | 0.1137 | 0.4260 | 0.5321 | 0.7009 |
| 8 | PreDiff-R | 2023 | 0.3995 | 0.0603 | 0.0080 | 0.8474 | 0.3815 | 0.1059 | 0.5695 | 0.9340 | 0.9914 |
| 9 | SpateGAN | 2023 | 0.4781 | 0.2673 | 0.1201 | 0.8610 | 0.4309 | 0.1600 | 0.4801 | 0.6058 | 0.7217 |
| 10 | GraphCast-R | 2023 | 0.4897 | 0.2699 | 0.1228 | 0.9181 | 0.4334 | 0.2133 | 0.5023 | 0.6724 | 0.7824 |
| 11 | Diffusion Nowcasting | 2025 | 0.5136 | 0.1913 | 0.0701 | 0.7795 | 0.4502 | 0.2096 | 0.4451 | 0.7642 | 0.9103 |
| 12 | Fusion-2025 | 2025 | 0.4834 | 0.2262 | 0.0925 | 0.8812 | 0.3016 | 0.1058 | 0.4948 | 0.5694 | 0.7685 |
| 13 | STAMP-GAN (Ours) | – | 0.6379 | 0.2981 | 0.1551 | 0.9350 | 0.5403 | 0.3655 | 0.2495 | 0.5201 | 0.6990 |
| Index | Models | Year | HSS ↑ | MAE ↓ | RMSE ↓ | MBE∼0 | Corr ↑ | ||
|---|---|---|---|---|---|---|---|---|---|
| 0.5 | 5 | 10 | |||||||
| 1 | ConvLSTM | 2015 | 0.4743 | 0.3677 | 0.1698 | 0.5414 | 0.9677 | 0.1219 | 0.5819 |
| 2 | TrajGRU | 2017 | 0.5355 | 0.3567 | 0.1812 | 0.4933 | 0.9546 | 0.2686 | 0.5914 |
| 3 | U-Net++ | 2018 | 0.4394 | 0.3668 | 0.1682 | 0.5222 | 1.0157 | 0.1924 | 0.5887 |
| 4 | PhyDNet | 2020 | 0.3997 | 0.3870 | 0.1861 | 0.4706 | 0.9567 | 0.0409 | 0.5871 |
| 5 | Swin Transformer | 2021 | 0.3717 | 0.3806 | 0.1946 | 0.4141 | 0.9467 | 0.1102 | 0.5783 |
| 6 | DGMR | 2021 | 0.5550 | 0.3310 | 0.1542 | 0.4610 | 0.9251 | 0.1054 | 0.5979 |
| 7 | MetNet3-R | 2023 | 0.5306 | 0.3404 | 0.1641 | 0.4849 | 0.9572 | 0.2489 | 0.5589 |
| 8 | PreDiff-R | 2023 | 0.0696 | 0.0213 | 0.0065 | 0.7801 | 1.6670 | 0.3794 | 0.4093 |
| 9 | SpateGAN | 2023 | 0.2821 | 0.3754 | 0.1954 | 0.5196 | 1.0047 | 0.1287 | 0.5793 |
| 10 | GraphCast-R | 2023 | 0.2520 | 0.3755 | 0.2017 | 0.6475 | 1.1280 | 0.3883 | 0.6009 |
| 11 | Diffusion Nowcasting | 2025 | 0.3675 | 0.2566 | 0.1164 | 0.5093 | 1.0126 | 0.1837 | 0.6033 |
| 12 | Fusion-2025 | 2025 | 0.5384 | 0.3239 | 0.1503 | 1.0835 | 2.1711 | 0.2414 | 0.6659 |
| 13 | STAMP-GAN (Ours) | – | 0.6158 | 0.4088 | 0.2386 | 0.3892 | 0.9260 | 0.0026 | 0.6180 |
| Index | Models | Year | CSI ↑ | POD ↑ | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0.5 | 5 | 10 | 15 | 20 | 0.5 | 5 | 10 | 15 | 20 | |||
| 1 | ConvLSTM | 2015 | 0.4045 | 0.1546 | 0.0591 | 0.0343 | 0.0165 | 0.7376 | 0.3070 | 0.1157 | 0.0663 | 0.0311 |
| 2 | TrajGRU | 2017 | 0.4099 | 0.1669 | 0.0845 | 0.0629 | 0.0421 | 0.8057 | 0.5112 | 0.2476 | 0.1883 | 0.1258 |
| 3 | U-Net++ | 2018 | 0.4511 | 0.2043 | 0.0996 | 0.0709 | 0.0411 | 0.8076 | 0.6146 | 0.3400 | 0.2497 | 0.1665 |
| 4 | PhyDNet | 2020 | 0.2455 | 0.1866 | 0.0945 | 0.0700 | 0.0363 | 0.8634 | 0.5337 | 0.2381 | 0.1262 | 0.0339 |
| 5 | Swin Transformer | 2021 | 0.1846 | 0.0766 | 0.0324 | 0.0119 | 0.0000 | 0.8306 | 0.4234 | 0.0630 | 0.0242 | 0.0000 |
| 6 | DGMR | 2021 | 0.4006 | 0.1952 | 0.0942 | 0.0618 | 0.0255 | 0.7159 | 0.4245 | 0.1799 | 0.0994 | 0.0306 |
| 7 | MetNet3-R | 2023 | 0.2925 | 0.1695 | 0.0845 | 0.0528 | 0.0185 | 0.7919 | 0.3015 | 0.1242 | 0.0651 | 0.0170 |
| 8 | PreDiff-R | 2023 | 0.3315 | 0.0463 | 0.0085 | 0.0040 | 0.0010 | 0.6585 | 0.1493 | 0.0300 | 0.0119 | 0.0023 |
| 9 | SpateGAN | 2023 | 0.3255 | 0.1885 | 0.0861 | 0.0442 | 0.0259 | 0.7755 | 0.5864 | 0.2819 | 0.1398 | 0.0305 |
| 10 | GraphCast-R | 2023 | 0.3123 | 0.1496 | 0.0857 | 0.0589 | 0.0219 | 0.8168 | 0.5613 | 0.2435 | 0.1147 | 0.0279 |
| 11 | Diffusion Nowcasting | 2025 | 0.1184 | 0.0258 | 0.0078 | 0.0052 | 0.0017 | 0.5561 | 0.3512 | 0.0539 | 0.0157 | 0.0024 |
| 12 | Fusion-2025 | 2025 | 0.4316 | 0.2045 | 0.0844 | 0.0659 | 0.0339 | 0.7603 | 0.4786 | 0.1905 | 0.1244 | 0.0685 |
| 13 | STAMP-GAN (Ours) | – | 0.4941 | 0.2134 | 0.1016 | 0.0715 | 0.0412 | 0.8252 | 0.6984 | 0.4714 | 0.3727 | 0.1948 |
| Index | Models | Year | FAR ↓ | HSS ↑ | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0.5 | 5 | 10 | 15 | 20 | 0.5 | 5 | 10 | 15 | 20 | |||
| 1 | ConvLSTM | 2015 | 0.5313 | 0.7681 | 0.8883 | 0.9275 | 0.9625 | 0.4449 | 0.2283 | 0.0999 | 0.0603 | 0.0303 |
| 2 | TrajGRU | 2017 | 0.5477 | 0.8057 | 0.8924 | 0.9181 | 0.9434 | 0.4595 | 0.2413 | 0.1383 | 0.1075 | 0.0750 |
| 3 | U-Net++ | 2018 | 0.4941 | 0.7897 | 0.8723 | 0.8999 | 0.9300 | 0.5153 | 0.2942 | 0.1607 | 0.1218 | 0.0678 |
| 4 | PhyDNet | 2020 | 0.7451 | 0.8136 | 0.8751 | 0.8931 | 0.9121 | 0.1693 | 0.2313 | 0.1529 | 0.1092 | 0.0443 |
| 5 | Swin Transformer | 2021 | 0.8142 | 0.9160 | 0.9616 | 0.9771 | 1.0000 | 0.0613 | 0.1324 | 0.0507 | 0.0000 | 0.0000 |
| 6 | DGMR | 2021 | 0.6067 | 0.8047 | 0.8604 | 0.8684 | 0.8842 | 0.4522 | 0.2826 | 0.1553 | 0.1056 | 0.0468 |
| 7 | MetNet3-R | 2023 | 0.6914 | 0.7949 | 0.8697 | 0.8867 | 0.8954 | 0.2965 | 0.2677 | 0.1369 | 0.0893 | 0.0339 |
| 8 | PreDiff-R | 2023 | 0.5996 | 0.9368 | 0.9880 | 0.9938 | 0.9983 | 0.0107 | 0.0163 | 0.0102 | 0.0061 | 0.0016 |
| 9 | SpateGAN | 2023 | 0.6577 | 0.7857 | 0.8460 | 0.8604 | 0.8867 | 0.3149 | 0.2696 | 0.1444 | 0.0757 | 0.0478 |
| 10 | GraphCast-R | 2023 | 0.6706 | 0.8317 | 0.8876 | 0.8990 | 0.9106 | 0.2644 | 0.2020 | 0.1359 | 0.0962 | 0.0380 |
| 11 | Diffusion Nowcasting | 2025 | 0.8249 | 0.9549 | 0.9834 | 0.9870 | 0.9913 | 0.0061 | 0.0048 | 0.0065 | 0.0074 | 0.0030 |
| 12 | Fusion-2025 | 2025 | 0.5038 | 0.7431 | 0.8766 | 0.9001 | 0.8971 | 0.4942 | 0.3006 | 0.1334 | 0.1138 | 0.0602 |
| 13 | STAMP-GAN (Ours) | – | 0.3370 | 0.6575 | 0.8020 | 0.8410 | 0.8705 | 0.5776 | 0.3098 | 0.1683 | 0.1230 | 0.0714 |
| Model | Evaluation Domain | CSI@15 | POD@15 | FAR@15 | CSI@20 | POD@20 | FAR@20 |
|---|---|---|---|---|---|---|---|
| U-Net++ | Full grid | 0.0709 | 0.2497 | 0.8999 | 0.0411 | 0.1665 | 0.9300 |
| U-Net++ | Within 5 km | 0.0862 | 0.2903 | 0.8905 | 0.0461 | 0.1649 | 0.9254 |
| STAMP-GAN | Full grid | 0.0715 | 0.3727 | 0.8411 | 0.0412 | 0.1948 | 0.8705 |
| STAMP-GAN | Within 5 km | 0.0734 | 0.3779 | 0.8385 | 0.0424 | 0.2898 | 0.8673 |
| Index | Models | Year | MAE ↓ | RMSE ↓ | MBE∼0 | Corr ↑ |
|---|---|---|---|---|---|---|
| 1 | ConvLSTM | 2015 | 1.2749 | 2.3277 | 0.9217 | 0.2844 |
| 2 | TrajGRU | 2017 | 1.2086 | 2.3593 | 0.8642 | 0.2914 |
| 3 | U-Net++ | 2018 | 0.7252 | 1.8717 | 0.4134 | 0.3388 |
| 4 | PhyDNet | 2020 | 0.7823 | 1.8839 | 0.5058 | 0.3504 |
| 5 | Swin Transformer | 2021 | 1.6529 | 2.7392 | 1.3466 | 0.2206 |
| 6 | DGMR | 2021 | 0.5033 | 2.3167 | 0.1292 | 0.3426 |
| 7 | MetNet3-R | 2023 | 0.6138 | 1.6774 | 0.2723 | 0.3232 |
| 8 | PreDiff-R | 2023 | 1.1596 | 2.0289 | 0.4999 | 0.0303 |
| 9 | SpateGAN | 2023 | 0.9853 | 1.9034 | 0.7170 | 0.3378 |
| 10 | GraphCast-R | 2023 | 1.8823 | 2.8406 | 1.4439 | 0.0257 |
| 11 | Diffusion Nowcasting | 2025 | 1.1282 | 2.1049 | 0.6532 | 0.2954 |
| 12 | Fusion-2025 | 2025 | 0.9013 | 1.9621 | 0.6022 | 0.3337 |
| 13 | STAMP-GAN (Ours) | – | 0.3809 | 1.5205 | -0.0464 | 0.3743 |
| Model | 0.5 mm h−1 | 5 mm h−1 | 10 mm h−1 | ||||||
|---|---|---|---|---|---|---|---|---|---|
| ConvLSTM | 0.697 | 0.828 | 0.864 | 0.384 | 0.615 | 0.718 | 0.181 | 0.363 | 0.460 |
| TrajGRU | 0.703 | 0.827 | 0.864 | 0.388 | 0.602 | 0.694 | 0.183 | 0.355 | 0.442 |
| U-Net++ | 0.713 | 0.840 | 0.877 | 0.399 | 0.609 | 0.701 | 0.170 | 0.317 | 0.384 |
| PhyDNet | 0.704 | 0.807 | 0.839 | 0.423 | 0.635 | 0.728 | 0.223 | 0.419 | 0.521 |
| Swin Transformer | 0.661 | 0.767 | 0.796 | 0.254 | 0.299 | 0.377 | 0.161 | 0.267 | 0.349 |
| DGMR | 0.748 | 0.881 | 0.922 | 0.360 | 0.570 | 0.663 | 0.156 | 0.276 | 0.319 |
| MetNet3-R | 0.695 | 0.796 | 0.829 | 0.369 | 0.589 | 0.685 | 0.165 | 0.305 | 0.357 |
| PreDiff-R | 0.429 | 0.675 | 0.722 | 0.400 | 0.505 | 0.531 | 0.167 | 0.342 | 0.429 |
| SpateGAN | 0.647 | 0.793 | 0.826 | 0.410 | 0.638 | 0.734 | 0.200 | 0.380 | 0.454 |
| GraphCast-R | 0.578 | 0.612 | 0.624 | 0.422 | 0.608 | 0.677 | 0.193 | 0.403 | 0.516 |
| Diffusion Nowcasting | 0.441 | 0.690 | 0.736 | 0.349 | 0.357 | 0.360 | 0.200 | 0.323 | 0.412 |
| Fusion-2025 | 0.651 | 0.766 | 0.797 | 0.353 | 0.569 | 0.665 | 0.152 | 0.262 | 0.298 |
| STAMP-GAN (Ours) | 0.771 | 0.896 | 0.934 | 0.439 | 0.663 | 0.759 | 0.227 | 0.424 | 0.527 |
| Model | 0.5 mm h−1 | 5 mm h−1 | 10 mm h−1 | ||||||
|---|---|---|---|---|---|---|---|---|---|
| ConvLSTM | 0.324 | 0.399 | 0.425 | 0.192 | 0.316 | 0.377 | 0.110 | 0.241 | 0.313 |
| TrajGRU | 0.352 | 0.421 | 0.447 | 0.205 | 0.303 | 0.344 | 0.118 | 0.226 | 0.268 |
| U-Net++ | 0.383 | 0.585 | 0.626 | 0.279 | 0.420 | 0.453 | 0.150 | 0.271 | 0.330 |
| PhyDNet | 0.270 | 0.364 | 0.391 | 0.220 | 0.300 | 0.325 | 0.112 | 0.266 | 0.325 |
| Swin Transformer | 0.168 | 0.218 | 0.245 | 0.308 | 0.363 | 0.417 | 0.203 | 0.246 | 0.317 |
| DGMR | 0.411 | 0.547 | 0.604 | 0.273 | 0.426 | 0.445 | 0.138 | 0.266 | 0.325 |
| MetNet3-R | 0.429 | 0.557 | 0.594 | 0.282 | 0.418 | 0.455 | 0.159 | 0.293 | 0.313 |
| PreDiff-R | 0.204 | 0.386 | 0.413 | 0.311 | 0.332 | 0.442 | 0.246 | 0.317 | 0.423 |
| SpateGAN | 0.395 | 0.504 | 0.534 | 0.263 | 0.395 | 0.453 | 0.160 | 0.325 | 0.406 |
| GraphCast-R | 0.234 | 0.242 | 0.245 | 0.192 | 0.238 | 0.287 | 0.133 | 0.156 | 0.191 |
| Diffusion Nowcasting | 0.231 | 0.273 | 0.282 | 0.282 | 0.291 | 0.314 | 0.176 | 0.220 | 0.288 |
| Fusion-2025 | 0.353 | 0.422 | 0.447 | 0.257 | 0.417 | 0.443 | 0.143 | 0.333 | 0.334 |
| STAMP-GAN (Ours) | 0.488 | 0.589 | 0.661 | 0.315 | 0.429 | 0.461 | 0.341 | 0.367 | 0.408 |
| Dataset | Split | Sliding Windows | Total Target Pixels | Dry | Light | Moderate | Heavy |
|---|---|---|---|---|---|---|---|
| ERA5 | Training | 6511 | 10,000,896 | 7,402,100 (74.0%) | 2,108,484 (21.1%) | 397,522 (4.0%) | 92,790 (0.9%) |
| ERA5 | Validation | 720 | 1,105,920 | 984,640 (89.0%) | 110,264 (10.0%) | 8610 (0.8%) | 2406 (0.2%) |
| ERA5 | Testing | 4381 | 6,729,216 | 5,545,091 (82.4%) | 955,083 (14.2%) | 188,350 (2.8%) | 40,692 (0.6%) |
| CMA-S | Training | 2514 | 96,537,600 | 78,000,060 (80.8%) | 13,916,164 (14.4%) | 3,142,361 (3.3%) | 1,479,015 (1.5%) |
| CMA-S | Validation | 709 | 27,225,600 | 25,510,728 (93.7%) | 1,375,212 (5.1%) | 231,255 (0.8%) | 108,405 (0.4%) |
| CMA-S | Testing | 4381 | 168,230,400 | 150,211,272 (89.3%) | 14,140,103 (8.4%) | 2,714,222 (1.6%) | 1,164,803 (0.7%) |
| Seed | Full STAMP-GAN | STAMP-GAN w/o | ||||
|---|---|---|---|---|---|---|
| CSI@0.5 | CSI@5 | CSI@10 | CSI@0.5 | CSI@5 | CSI@10 | |
| 42 | 0.6379 | 0.2981 | 0.1552 | 0.6272 | 0.2695 | 0.1392 |
| 123 | 0.6342 | 0.2811 | 0.1410 | 0.6280 | 0.2641 | 0.1308 |
| 2024 | 0.6314 | 0.2978 | 0.1530 | 0.6319 | 0.2517 | 0.1316 |
| 3407 | 0.6293 | 0.2946 | 0.1429 | 0.6368 | 0.2470 | 0.1220 |
| 5678 | 0.6386 | 0.2937 | 0.1352 | 0.6229 | 0.2518 | 0.1284 |
| Mean ± std. | 0.6343 ± 0.0040 | 0.2931 ± 0.0070 | 0.1455 ± 0.0084 | 0.6294 ± 0.0052 | 0.2568 ± 0.0095 | 0.1304 ± 0.0062 |
| Seed | Full STAMP-GAN | STAMP-GAN w/o | ||||
|---|---|---|---|---|---|---|
| CSI@0.5 | CSI@5 | CSI@10 | CSI@0.5 | CSI@5 | CSI@10 | |
| 42 | 0.5865 | 0.2027 | 0.0618 | 0.5809 | 0.1739 | 0.0450 |
| 123 | 0.5894 | 0.1948 | 0.0518 | 0.5850 | 0.1680 | 0.0369 |
| 2024 | 0.5792 | 0.2115 | 0.0653 | 0.5781 | 0.1752 | 0.0317 |
| 3407 | 0.5728 | 0.1981 | 0.0677 | 0.5715 | 0.1712 | 0.0440 |
| 5678 | 0.5776 | 0.1904 | 0.0554 | 0.5728 | 0.1612 | 0.0341 |
| Mean ± std. | 0.5811 ± 0.0068 | 0.1995 ± 0.0081 | 0.0604 ± 0.0067 | 0.5777 ± 0.0056 | 0.1699 ± 0.0056 | 0.0383 ± 0.0059 |
| Threshold | Lead | Observed | STAMP TP | STAMP FP | STAMP FN | Fusion TP | Fusion FP | Fusion FN |
|---|---|---|---|---|---|---|---|---|
| 10 | T1 | 185,525 | 120,823 | 231,873 | 64,702 | 132,866 | 480,435 | 52,659 |
| 10 | T2 | 185,572 | 116,476 | 262,062 | 69,096 | 70,451 | 382,335 | 115,121 |
| 10 | T3 | 185,644 | 93,741 | 418,981 | 91,903 | 26,871 | 178,107 | 158,773 |
| 10 | T4 | 185,728 | 86,748 | 578,753 | 98,980 | 11,601 | 76,916 | 174,127 |
| 10 | T5 | 185,751 | 79,552 | 479,610 | 106,199 | 4010 | 39,065 | 181,741 |
| 10 | T6 | 185,674 | 27,658 | 307,631 | 158,016 | 77 | 279 | 185,597 |
| 15 | T1 | 97,323 | 52,469 | 109,308 | 44,854 | 64,650 | 325,437 | 32,673 |
| 15 | T2 | 97,352 | 50,708 | 173,693 | 46,644 | 22,773 | 160,391 | 74,579 |
| 15 | T3 | 97,393 | 43,922 | 270,031 | 53,471 | 4955 | 45,853 | 92,438 |
| 15 | T4 | 97,444 | 33,812 | 286,863 | 63,632 | 1039 | 13,509 | 96,405 |
| 15 | T5 | 97,459 | 30,573 | 281,078 | 66,886 | 95 | 770 | 97,364 |
| 15 | T6 | 97,445 | 6292 | 99,084 | 91,153 | 11 | 4 | 97,434 |
| 20 | T1 | 39,460 | 13,388 | 29,698 | 26,072 | 19,980 | 158,063 | 19,480 |
| 20 | T2 | 39,473 | 10,953 | 50,077 | 28,520 | 2520 | 29,751 | 36,953 |
| 20 | T3 | 39,496 | 10,508 | 107,281 | 28,988 | 213 | 6311 | 39,283 |
| 20 | T4 | 39,517 | 5924 | 61,532 | 33,593 | 0 | 109 | 39,517 |
| 20 | T5 | 39,524 | 4728 | 55,872 | 34,796 | 0 | 0 | 39,524 |
| 20 | T6 | 39,521 | 651 | 19,686 | 38,870 | 0 | 0 | 39,521 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Ma, X.; Lu, Z.; Wang, F.; Feng, H.; Lu, B. STAMP-GAN: A Spatiotemporal Attention-Modulated Generative Adversarial Network for Precipitation Nowcasting. Remote Sens. 2026, 18, 3026. https://doi.org/10.3390/rs18173026
Ma X, Lu Z, Wang F, Feng H, Lu B. STAMP-GAN: A Spatiotemporal Attention-Modulated Generative Adversarial Network for Precipitation Nowcasting. Remote Sensing. 2026; 18(17):3026. https://doi.org/10.3390/rs18173026
Chicago/Turabian StyleMa, Xiaoxiao, Zhenyu Lu, Fang Wang, Hailin Feng, and Bingjian Lu. 2026. "STAMP-GAN: A Spatiotemporal Attention-Modulated Generative Adversarial Network for Precipitation Nowcasting" Remote Sensing 18, no. 17: 3026. https://doi.org/10.3390/rs18173026
APA StyleMa, X., Lu, Z., Wang, F., Feng, H., & Lu, B. (2026). STAMP-GAN: A Spatiotemporal Attention-Modulated Generative Adversarial Network for Precipitation Nowcasting. Remote Sensing, 18(17), 3026. https://doi.org/10.3390/rs18173026

