Accurate and timely rainfall nowcasting is crucial for disaster mitigation and water resource management. Despite recent advances in deep learning, precipitation prediction remains challenging due to limitations in effectively leveraging diverse multimedia data sources. We introduce M3R, a Meteorology-informed MultiModal attention-based architecture for direct Rainfall prediction that synergistically combines visual NEXRAD radar imagery with numerical Personal Weather Station (PWS) measurements, using a comprehensive pipeline for temporal alignment of heterogeneous meteorological data. With specialized multimodal attention mechanisms, M3R novelly leverages weather station time series as queries to selectively attend to spatial radar features, enabling focused extraction of precipitation signatures. Experimental results for three spatial areas of 100 km * 100 km centered at NEXRAD radar stations demonstrate that M3R outperforms existing approaches, achieving substantial improvements in accuracy, efficiency, and precipitation detection capabilities. Our work establishes new benchmarks for multimedia-based precipitation nowcasting and provides practical tools for operational weather prediction systems. The source code is available at https://github.com/Sanjeev97/M3Rain
Figures & tables
Fig. 1: Rainfall Event Sample (Sequence 5655) of the LA dataset.
Fig. 2: Illustration of Uniform Frame Sampling for Visual Embedding.
Fig. 3: Illustration of M3R Model with Vision Encoder, TS Encoder, MultiModal Attention Block, TS Decoder and MLP Head.
Methods
RMSE ( ↓ )
MAE ( ↓ )
R 2 ( ↑ )
CC ( ↑ )
CSI 0.1 ( ↑ )
CSI 5 ( ↑ )
CSI 10 ( ↑ )
LA
AL
MS
LA
AL
MS
LA
AL
MS
LA
AL
MS
LA
AL
MS
LA
AL
MS
LA
AL
MS
Transformer [ 19 ]
3.79
4.06
3.68
0.74
1.03
0.64
0.01
0.004
0.09
0.12
0.06
0.30
0.060
0.071
0.070
0.174
0.029
0.221
0.022
0.024
0.172
DLinear [ 22 ]
3.27
3.10
3.49
0.81
0.83
0.77
0.06
0.08
0.12
0.25
0.28
0.35
0.047
0.051
0.042
0.211
0.125
0.234
0.114
0.037
0.154
PatchTST [ 14 ]
3.63
3.14
3.56
0.45
0.36
0.41
0.07
0.06
0.10
0.26
0.24
0.32
0.366
0.359
0.372
0.226
0.145
0.340
0.133
0.071
0.250
iTransformer [ 12 ]
3.43
3.41
3.54
0.49
0.41
0.41
0.08
0.02
0.13
0.28
0.15
0.36
0.365
0.349
0.355
0.233
0.074
0.339
0.125
0.066
0.213
Diffcast-SimVP [ 21 ]
4.70
3.13
4.14
0.97
0.60
0.65
0.05
0.09
0.08
0.21
0.31
0.29
0.074
0.092
0.073
0.218
0.114
0.377
0.105
0.100
0.284
TABLE I: Comparison showing our model is best or second best in both time series forecasting and binary rain classification over three spatial domains of interest (denoted by LA, AL and MS).
Fig. 4: Comparison: (a) Zero (b) High-intensity and (c) Medium-intensity precipitation showing M3R’s consistent performance across all scenarios for LA.
Method
Size
FLOPs
Training
Inference
Memory
Time
Memory
Time
AlphaPre
4.06 M
133.6 G
6111 MB
13 hr
1415 MB
35 s
Diffcast-PhyDnet
3.09 M
18.3 T
1154 MB
12 hr
426 MB
34 s
Diffcast-SimVP
10.85 M
18.3 T
4848 MB
33 hr
2238 MB
68 s
M3R (Ours)
3.22 M
0.14 G
770 MB
1 hr
269 MB
5 s
TABLE II: Efficiency Analysis on Jackson (MS) Station
Methods
RMSE
MAE
R2
CC
CSI 0.1
CSI 10
TS Encoder only
3.79
0.74
0.01
0.12
0.060
0.022
M3R w/o TS Decoder
3.08
0.36
0.13
0.36
0.340
0.091
M3R (Ours)
2.95
0.33
0.21
0.45
0.395
0.150
TABLE III: Ablation study results on Lake Charles (LA) Station
Deep learning has significantly improved the accuracy of precipitation nowcasting. However, most existing multimodal models typically use simple channel concatenation or interpolation methods for data fusion, which often overlook the feature differences between different modalities. This paper therefore proposes a novel precipitation nowcasting optimisation framework called FusionCast. This framework incorporates three types of data: historical precipitable water vapour (PWV) data derived from global navigation satellite system (GNSS) inversions, historical radar based quantitative precipitation estimation (QPE), and forecasted radar QPE serving as a future prior. The FusionCast model comprises two core modules: the future prior radar QPE processing Module, which forecasts future radar data; and the Radar PWV Fusion (RPF) module, which uses a gate mechanism to efficiently combine features from various sources. Experimental results show that FusionCast significantly improves nowcasting performance.
Henan Wang, Shengwu Xiong, Yifang Zhang +4
School of Computer Science and Artificial Intelligence, Wuhan University of Technology · School of Earth and Space Science and Technology, Wuhan University
Precipitation nowcasting (short-term forecasting) is still often performed using numerical solvers for physical equations, which are computationally expensive and make limited use of the large volumes of available weather data. Deep learning models have shown strong potential for precipitation nowcasting, offering both accuracy and computational efficiency. Among these models, convolutional neural networks (CNNs) are particularly effective for image-to-image prediction tasks. The SmaAt-UNet is a lightweight CNN based architecture that has demonstrated strong performance for precipitation nowcasting. This paper introduces the Multimodal Advection-Guided Small Attention GNet (MAD-SmaAt-GNet), which extends the core SmaAt-UNet by (i) incorporating an additional encoder to learn from multiple weather variables and (ii) integrating a physics-based advection component to ensure physically consistent predictions. We show that each extension individually improves rainfall forecasts and that their combination yields further gains. MAD-SmaAt-GNet reduces the mean squared error (MSE) by 8.9% compared with the baseline SmaAt-UNet for four-step precipitation forecasting up to four hours ahead. Additionally, experiments indicate that multimodal inputs are particularly beneficial for short lead times, while the advection-based component enhances performance across both short and long forecasting horizons.
Samuel van Wonderen, Siamak Mehrkanoon
Department of Information and Computing Sciences, Utrecht University, Utrecht, The Netherlands
Precipitation nowcasting is a vital spatio-temporal prediction task for meteorological applications but faces challenges due to the chaotic property of precipitation systems. Existing methods predominantly rely on single-source radar data to build either deterministic or probabilistic models for extrapolation. However, the single deterministic model suffers from blurring due to MSE convergence. The single probabilistic model, typically represented by diffusion models, can generate fine details but suffers from spurious artifacts that compromise accuracy and computational inefficiency. To address these challenges, this paper proposes a novel coarse-to-fine Vision Mamba Unet and residual Diffusion (VMU-Diff) based precipitation nowcasting framework. It realizes precipitation nowcasting through a two-stage process, i.e., a deterministic model-based coarse stage to predict global motion trends and a probabilistic model-based fine stage to generate fine prediction details. In the coarse prediction stage, rather than single-source radar data, both radar and multi-band satellite data are taken as input. A spatial-temporal attention block and several Vision mamba state-space blocks realize multi-source data fusion, and predict the future echo global dynamics. The fine-grained stage is realized by a spatio-temporal refine generator based on residual conditional diffusion models. It first obtains spatio-temporal residual features based on coarse prediction and ground truth, and further reconstructs the residual via conditional Mamba state-space module. Experiments on Jiangsu SWAN datasets demonstrate the improvements of our method over state-of-the-art methods, particularly in short-term forecasts.
Chunlei Shi, Hao Li, Yufeng Zhu +6
Department of Automation, Southeast University, Nanjing 210096, China