Two-Stage Spatiotemporal Context Refinement Network for Precipitation Nowcasting

• Published in: Remote sensing (Basel, Switzerland) Vol. 13; no. 21; p. 4285
• Main Authors: Niu, Dan, Huang, Junhao, Zang, Zengliang, Xu, Liujia, Che, Hongshu, Tang, Yuanqing
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.11.2021
• Subjects: Accuracy; Algorithms; China; Coders; Context; data collection; Datasets; Deep learning; Feature extraction; image analysis; Image quality; Lead time; Learning algorithms; Machine learning; Methods; Neural networks; Nowcasting; Precipitation; precipitation nowcasting; prediction; Predictions; Radar; Radar echoes; Radar imaging; Rain; Rainfall; refinement network; Remote sensing; RNN; spatiotemporal prediction; Weather forecasting; China
• DOI: 10.3390/rs13214285
• ISSN: 2072-4292

Precipitation nowcasting by radar echo extrapolation using machine learning algorithms is a field worthy of further study, since rainfall prediction is essential in work and life. Current methods of predicting the radar echo images need further improvement in prediction accuracy as well as in presenting the predicted details of the radar echo images. In this paper, we propose a two-stage spatiotemporal context refinement network (2S-STRef) to predict future pixel-level radar echo maps (deterministic output) more accurately and with more distinct details. The first stage is an efficient and concise spatiotemporal prediction network, which uses the spatiotemporal RNN module embedded in an encoder and decoder structure to give a first-stage prediction. The second stage is a proposed detail refinement net, which can preserve the high-frequency detailed feature of the radar echo images by using the multi-scale feature extraction and fusion residual block. We used a real-world radar echo map dataset of South China to evaluate the proposed 2S-STRef model. The experiments showed that compared with the PredRNN++ and ConvLSTM method, our 2S-STRef model performs better on the precipitation nowcasting, as well as at the image quality evaluating index and the forecasting indices. At a given 45 dBZ echo threshold (heavy precipitation) and with a 2 h lead time, the widely used CSI, HSS, and SSIM indices of the proposed 2S-STRef model are found equal to 0.195, 0.312, and 0.665, respectively. In this case, the proposed model outperforms the OpticalFlow method and PredRNN++ model.