AlphaPre-CVPR2025

Official implementation of "AlphaPre: Amplitude-Phase Disentanglement Model for Precipitation Nowcasting"

Introduction

AlphaPre is an amplitude-phase disentanglement model for precipitation nowcasting that separately learns positional and intensity changes by leveraging phase and amplitude variations in the frequency domain.

This repository contains the part of training and inference code for using AlphaPre to make predictions (5 --> 20) on SEVIR datasets.

Code

Environment

conda env create -f env.yaml
conda activate alphapre

Optional Accelerate Env

We apply the HuggingFace Accelerator in our code to utilize multi-gpus. One can config the accelerator env before runing code.

• config the accelerate: accelerate config
• apply accelerate to run code: accelerate launch *.py

Resource

pretrained checkpoint: GoogleDrive

Datasets

All four datasets in our paper are from DiffCast. You can find the datasets as follows:

• SEVIR
• Metnet
• Shanghai_Radar
• CIKM_Radar

Evaluation

Before evaluation, you need to download the pretrained checkpoint and put it in resources/

# Note: Config the dataset path in `dataset/get_dataset.py` before running.
python run.py --eval --ckpt_milestone resources/AlphaPre_sevir128.pt  

Training

python run.py

You can check the experimental configuration by

python run.py -h

Acknowledgement

We refer to implementations of the following repositories and sincerely thank their contribution for the community:

• OpenSTL
• DiffCast

Citation

@InProceedings{Lin2025AlphaPre,
    author    = {Lin, Kenghong and Zhang, Baoquan and Yu, Demin and Feng, Wenzhi and Chen, Shidong and Gao, Feifan and Li, Xutao and Ye, Yunming},
    title     = {AlphaPre: Amplitude-Phase Disentanglement Model for Precipitation Nowcasting},
    booktitle = {Proceedings of the Computer Vision and Pattern Recognition Conference (CVPR)},
    month     = {June},
    year      = {2025},
    pages     = {17841-17850}
}