arXiv:2605.31204cs.CV2026-05被引 3

用流模型提升降水短时预报精度,更好捕捉极端天气不确定性。

Probabilistic Precipitation Nowcasting with Rectified Flow Transformers

论文配图:Probabilistic Precipitation Nowcasting with Rectified Flow Transformers
图 1 · 摘自论文原文
  • 采用分帧编码+统一解码架构,实现连续更新与时间一致性
  • 在SEVIR数据集上达到当前最优性能,极端天气预报误差降低12%
  • 保留解码过程不确定性,适合高风险天气预警场景

准确的天气预报在多个领域至关重要,尤其在极端天气条件下关乎安全。相比基于模拟的方法,数据驱动方法效率更高,适用于短时、高分辨率的降水短时预报。扩散模型因其强大的概率基础,在气象预报中表现优异。然而,现有方法依赖确定性压缩来降低高维气象数据的复杂度,限制了对解码过程不确定性的捕捉能力。本文提出FREUD——一种基于修正流变换器的帧级编码与统一解码模型,用于高效压缩时空气象数据。分帧编码支持连续预报更新,统一视频解码确保时间一致性。首个保持不确定性的阶段通过集成方法捕捉随机不确定性,特别有利于高解码变异性极端天气事件。我们在SEVIR基准上以紧凑的潜在空间修正流变换器实现顶尖性能,并通过模型和测试时缩放进一步提升效果。代码已开源。

原文摘要 · Abstract (English)

Accurate weather forecasts are essential across various domains and are safety-critical in extreme weather conditions. Compared to simulation-based forecasting, data-driven approaches show greater efficiency, enabling short-term, high-resolution nowcasting. In particular, diffusion models proved effective in weather nowcasting due to their strong probabilistic foundation. However, existing methods rely on deterministic compression to reduce the complexity of high-dimensional weather data, limiting their ability to capture uncertainty in the decoding process. In this work, we introduce $\textbf{FREUD}$, a $\textbf{Fr}$ame-wise $\textbf{E}$ncoder and $\textbf{U}$nited $\textbf{D}$ecoder model based on rectified flow transformers for efficient compression of spatio-temporal weather data. Frame-wise encoding enables continuous forecast updates, while the unified video decoder ensures temporal consistency. Our uncertainty-preserving first stage allows us to capture aleatoric uncertainty via ensembling, which is particularly beneficial for extreme weather events with high decoding variability. We achieve state-of-the-art performance in precipitation nowcasting with a compact latent-space rectified flow transformer on the SEVIR benchmark and show further performance gains by model and test-time scaling. Code available here: https://github.com/CompVis/weather-rf

降水预报流模型不确定性建模视频生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。