SSRF-Net: A Stagewise Scheduled Rainfall Forecasting Network With an Asymmetric Architecture

Deterministic deep learning models for precipitation nowcasting often face several limitations, including cumulative error in long-sequence predictions, over-smoothing, and a reduced ability to capture rare, high-impact rainfall due to data imbalance. To address these challenges, we propose the stag...

Full description

Saved in:
Bibliographic Details
Published in:IEEE transactions on geoscience and remote sensing Vol. 63; pp. 1 - 18
Main Authors: Luo, Wenjie, Li, Chaorong, Ling, Xudong, Deng, Chuanhu, Wang, Zhuo
Format: Journal Article
Language:English
Published: New York IEEE 2025
The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
Subjects:
ISSN:0196-2892, 1558-0644
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Deterministic deep learning models for precipitation nowcasting often face several limitations, including cumulative error in long-sequence predictions, over-smoothing, and a reduced ability to capture rare, high-impact rainfall due to data imbalance. To address these challenges, we propose the stagewise scheduled rainfall forecasting network (SSRF-Net), a convolutional framework for continuous multistep rainfall prediction that achieves lower floating-point operations (FLOPs) than competitive baselines under a standardized evaluation. Our framework introduces a multistage, sliding-window prediction mechanism trained with teacher forcing and scheduled sampling to mitigate error accumulation and stabilize training. We design an asymmetric encoder-decoder (E-D) architecture featuring a differential selective encoder (DSE) for selective feature compression and an additive fusion decoder (AFD) that progressively reconstructs details and alleviates over-smoothing. We further introduce an intensity-weighted Gaussian KL divergence loss that aligns sequence-level Gaussian summaries (means and variances) of predictions and ground truth via a KL term, prioritizing heavy-rain events without assuming pixelwise Gaussianity. Extensive experiments on the KNMI and SEVIR datasets show that SSRF-Net outperforms strong baselines, particularly for moderate to severe precipitation; on KNMI, it yields up to 41.8% higher per-frame critical success index (CSI) at the 30-mm/h threshold, with consistent gains on SEVIR.
Bibliography:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 14
ISSN:0196-2892
1558-0644
DOI:10.1109/TGRS.2025.3621627