arXiv:2606.29855cs.CV2026-06被引 1

用连续时间模型提升降水预报精度与时间分辨率

RainODE: Continuous-Time Precipitation Forecasting with Latent Neural ODEs

论文配图:RainODE: Continuous-Time Precipitation Forecasting with Latent Neural ODEs
图 1 · 摘自论文原文
  • 将降水演变建模为隐空间中的神经微分方程,实现连续时间动态
  • 在SEVIR和RAPID数据集上,多时长预测均优于现有方法
  • 结合确定性动力学与随机修正模块,保持大尺度运动一致性

降水预报不仅要求高精度,还需高时间分辨率。然而,观测限制和密集离散建模的计算成本制约了分辨率提升。为此,我们提出RainODE,将降水预报重构为连续时间动力系统,利用神经微分方程在隐空间建模降水演化,实现导数一致的时序动态并捕捉主导的大尺度平流运动。但纯确定性微分方程难以刻画局部强度变化(如局地增长、衰减、亚网格变异性),常导致预测过平滑。为此,我们引入基于布朗桥的随机源建模模块,精细修正残差强度变化,恢复细粒结构,同时保持平流一致性。通过确定性连续动力学与随机精修的结合,RainODE支持任意时间点推断且预测清晰。在SEVIR及新提出的雷达降水综合数据集RAPID上的实验表明,其在多种时间间隔和降水类型下均具稳定提升。代码已开源。

原文摘要 · Abstract (English)

In precipitation forecasting, not only accuracy but also temporal resolution is critical. However, increasing temporal resolution is constrained by observational limitations and the computational cost of dense discrete modeling. To overcome this limitation, we reformulate precipitation forecasting as a continuous-time dynamical system and propose RainODE, a framework that models precipitation evolution in latent space using a Neural ODE. This formulation enables derivative-consistent temporal dynamics and captures the dominant large-scale advective motion of precipitation systems. Nevertheless, a purely deterministic ODE struggles to represent non-advective intensity changes such as localized growth, decay, and sub-grid variability, often leading to over-smoothed predictions. To address this issue, we introduce a stochastic source modeling module based on a Brownian Bridge formulation, which refines residual intensity variations and restores fine-grained structures while preserving advective consistency. By combining deterministic continuous dynamics with stochastic refinement, RainODE enables arbitrary-time inference while maintaining sharp predictions. Experiments on SEVIR and the newly introduced Radar-based Precipitation Integrated Dataset (RAPID) demonstrate consistent improvements across multiple temporal intervals and precipitation regimes. The code is available at https://github.com/SeongYE/RainODE.

降水预报神经微分方程连续时间随机建模

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