STPF-Net: Short-Term Precipitation Forecast Based on a Recurrent Neural Network

Abstract

Accurate and timely precipitation forecasts are critical in modern society, influencing both economic activity and daily life. While deep learning methods leveraging remotely sensed radar data have become prevalent for precipitation nowcasting, longer-term forecasting remains challenging. This is due to accumulated errors in deep learning models and insufficient information about precipitation systems over longer time horizons. To address these challenges, we introduce the Short-Term Precipitation Forecast Network (STPF-Net), a recurrent neural network designed for longer-term precipitation prediction. STPF-Net uses a multi-tier structure with varying temporal resolutions to mitigate the accumulated errors during longer forecasts. Additionally, its transformer-based module incorporates larger spatial contexts, providing more complete information about precipitation systems. We evaluated STPF-Net on radar data from southeastern China, training separate models for 6 and 12 h forecasts. Quantitative results demonstrate STPF-Net achieved superior accuracy and lower errors compared to benchmark deep learning and numerical weather prediction models. Visualized case studies indicate reasonably coherent 6 h predictions from STPF-Net versus other methods. For 12 h forecasts, while STPF-Net outperformed other models, it still struggled with storm initiation over longer forecasting time.