RainODE: Continuous-Time Precipitation Forecasting with Latent Neural ODEs
- URL: http://arxiv.org/abs/2606.29855v1
- Date: Mon, 29 Jun 2026 06:48:02 GMT
- Title: RainODE: Continuous-Time Precipitation Forecasting with Latent Neural ODEs
- Abstract summary: RainODE is a framework that models precipitation evolution in latent space using a Neural ODE.<n>This formulation enables derivative-consistent temporal dynamics and captures the dominant large-scale advective motion of precipitation systems.<n>A source modeling module based on a Brownian Bridge restores residual intensity variations while preserving fine-grained structures.
- Score: 31.6943694848236
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: 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.
Related papers
- Why SGD is not Brownian Motion: A New Perspective on Stochastic Dynamics [92.39053980710702]
Gradient Descent (SGD) is commonly modeled as a Langevin process, assuming that minibatch noise acts as Brownian motion.<n>This approximation relies on a continuous-time limit and a sqrt(eta) noise scaling that does not match the discrete SGD update at finite learning rate.<n>We propose an alternative formulation of SGD as deterministic dynamics in a fluctuating loss landscape induced by minibatch sampling.
arXiv Detail & Related papers (2026-05-21T15:50:40Z) - Forecasting as Rendering: A 2D Gaussian Splatting Framework for Time Series Forecasting [79.37674445572462]
Time series forecasting (TSF) remains a challenging problem due to the intricate entanglement of intraperiod-fluctuations and interperiod-trends.<n>Treating reshaped tensors as static images results in a topological mismatch.<n> relying on uniform fixed-size representations allocates modeling capacity inefficiently.<n>TimeGS is a novel framework that fundamentally shifts the forecasting paradigm from regression to 2D generative rendering.
arXiv Detail & Related papers (2026-02-10T14:13:36Z) - STLDM: Spatio-Temporal Latent Diffusion Model for Precipitation Nowcasting [22.29494904927957]
Precipitation nowcasting is a critical prediction task for society to prevent severe damage owing to extreme weather events.<n>We present a simple yet effective model architecture termed STLDM, a diffusion-based model that learns the latent representation from end to end alongside both the Varitemporal Autoencoder and the conditioning network.
arXiv Detail & Related papers (2025-12-24T11:34:44Z) - Structured Noise Modeling for Enhanced Time-Series Forecasting [0.0]
This work introduces a forecast-blur-denoise framework that improves temporal fidelity through structured noise modeling.<n> Experiments across electricity, traffic, and solar datasets show consistent gains in multi-horizon accuracy and stability.<n>This framework contributes to more trustworthy AI systems used in forecasting-driven decision support across energy, infrastructure, and other time-critical domains.
arXiv Detail & Related papers (2025-11-24T19:44:46Z) - Nowcast3D: Reliable precipitation nowcasting via gray-box learning [43.74533335638689]
Extreme precipitation nowcasting demands high-dimensional fidelity and extended lead times, yet existing approaches remain limited.<n>We introduce a gray-box, fully three-temporal nowcasting framework that directly processes radar motions and couples physically constrained neural operators with datadriven learning.<n>The framework achieves more accurate forecasts up to three-hour lead time in a blind evaluation by 160 meteorologists.
arXiv Detail & Related papers (2025-11-06T18:44:35Z) - RainDiff: End-to-end Precipitation Nowcasting Via Token-wise Attention Diffusion [64.49056527678606]
We propose a Token-wise Attention integrated into not only the U-Net diffusion model but also the radar-temporal encoder.<n>Unlike prior approaches, our method integrates attention into the architecture without incurring the high resource cost typical of pixel-space diffusion.<n>Our experiments and evaluations demonstrate that the proposed method significantly outperforms state-of-the-art approaches, robustness local fidelity, generalization, and superior in complex precipitation forecasting scenarios.
arXiv Detail & Related papers (2025-10-16T17:59:13Z) - Elucidated Rolling Diffusion Models for Probabilistic Weather Forecasting [52.6508222408558]
We introduce Elucidated Rolling Diffusion Models (ERDM)<n>ERDM is the first framework to unify a rolling forecast structure with the principled, performant design of Elucidated Diffusion Models (EDM)<n>On 2D Navier-Stokes simulations and ERA5 global weather forecasting at 1.5circ resolution, ERDM consistently outperforms key diffusion-based baselines.
arXiv Detail & Related papers (2025-06-24T21:44:31Z) - Hierarchical Implicit Neural Emulators [20.09615751270837]
We introduce a multiscale implicit neural emulator that enhances long-term prediction accuracy.<n>Our approach leverages predictions several steps ahead in time at increasing compression rates for next-timestep refinements.<n> Experiments on turbulent fluid dynamics show that our method achieves high short-term accuracy and produces long-term stable forecasts.
arXiv Detail & Related papers (2025-06-05T00:28:26Z) - DiffCast: A Unified Framework via Residual Diffusion for Precipitation Nowcasting [20.657502066923023]
Precipitation nowcasting is an important task to predict the radar echoes sequences based on current observations, which can serve both meteorological science and smart city applications.
Previous studies address the problem either from the perspectives of deterministic modeling or probabilistic modeling.
We propose to decompose and model the chaotic evolutionary precipitation systems from the perspective of global deterministic motion and local variations with residual mechanism.
arXiv Detail & Related papers (2023-12-11T11:26:32Z) - Learning Robust Precipitation Forecaster by Temporal Frame Interpolation [65.5045412005064]
We develop a robust precipitation forecasting model that demonstrates resilience against spatial-temporal discrepancies.
Our approach has led to significant improvements in forecasting precision, culminating in our model securing textit1st place in the transfer learning leaderboard of the textitWeather4cast'23 competition.
arXiv Detail & Related papers (2023-11-30T08:22:08Z) - Stochastically forced ensemble dynamic mode decomposition for
forecasting and analysis of near-periodic systems [65.44033635330604]
We introduce a novel load forecasting method in which observed dynamics are modeled as a forced linear system.
We show that its use of intrinsic linear dynamics offers a number of desirable properties in terms of interpretability and parsimony.
Results are presented for a test case using load data from an electrical grid.
arXiv Detail & Related papers (2020-10-08T20:25:52Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.