Precipitation Downscaling with Spatiotemporal Video Diffusion

Prakhar Srivastava, Ruihan Yang, Gavin Kerrigan, Gideon Dresdner, Jeremy McGibbon +2 more
2/12/2026

No official code found. Best implementation: mandt-lab/STVD (12 stars, MIT, last commit 2024-11-14).

Abstract

In climate science and meteorology, high-resolution local precipitation (rain and snowfall) predictions are limited by the computational costs of simulation-based methods. Statistical downscaling, or super-resolution, is a common workaround where a low-resolution prediction is improved using statistical approaches. Unlike traditional computer vision tasks, weather and climate applications require capturing the accurate conditional distribution of high-resolution given low-resolution patterns to assure reliable ensemble averages and unbiased estimates of extreme events, such as heavy rain. This work extends recent video diffusion models to precipitation super-resolution, employing a deterministic downscaler followed by a temporally-conditioned diffusion model to capture noise characteristics and high-frequency patterns. We test our approach on FV3GFS output, an established large-scale global atmosphere model, and compare it against six state-of-the-art baselines. Our analysis, capturing CRPS, MSE, precipitation distributions, and qualitative aspects using California and the Himalayas as examples, establishes our method as a new standard for data-driven precipitation downscaling.

View on arXivView PDF

Code Implementations(1)

mandt-lab/STVDCommunity67%
MIT

Official code for NeurIPS '24 paper 'Precipitation Downscaling with Spatiotemporal Video Diffusion'

122PythonOct 30, 20241 years agoMIT

Similar Papers

Cite this paper

@article{srivastava2026precipitation,
  title  = {Precipitation Downscaling with Spatiotemporal Video Diffusion},
  author = {Prakhar Srivastava and Ruihan Yang and Gavin Kerrigan and Gideon Dresdner and Jeremy McGibbon and Christopher S. Bretherton and Stephan Mandt},
  year   = {2026},
  eprint = {2312.06071},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2312.06071},
  journal = {NEURIPS 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2312.06071.svg)](https://codeofpaper.com/paper/2312.06071)

Discussion