EAAI Journal 2026 Journal Article
More realistic and accurate precipitation nowcasting with Conditional Rectified Flow Transformers
- Yunlong Zhou
- Chen Zhao
- Fanfan Ji
- Renlong Hang
- Qingshan Liu
- Xiao-Tong Yuan
Precipitation nowcasting plays a critical role in disaster prevention and daily life but remains challenging due to the intricate spatiotemporal dynamics of atmospheric processes. In response to these challenges, recent research has shown that diffusion models can generate visually realistic precipitation results. However, challenges such as accurately predicting precipitation positions and improving inference speed remain unresolved. To address these issues, we propose a novel Conditional Rectified Flow Transformers (CRFT) architecture to improve precipitation nowcasting, which is designed to deliver both predictive accuracy and visual realism. At its core, CRFT features an efficient latent space predictor powered by OmniFormer blocks, which integrate spatial, temporal, and spatiotemporal Transformers to holistically capture the atmosphere dynamics. We explore five variants of spatiotemporal dynamic information interactions for OmniFormer and demonstrate that integrating triple Transformers achieves the best performance. Additionally, we significantly reduce inference time by employing a rectified flow approach, achieving a reduction in inference steps by 98. 4% compared to existing methods, enabling high-quality 20-frame predictions within 2 s. Evaluated on three benchmark datasets, CRFT outperforms state-of-the-art (SOTA) models in both accuracy and quality across multiple metrics, offering an accurate and efficient solution for real-world nowcasting. The code is publicly available at https: //github. com/RuntimeWarning/CRFT.