Downscaling & super-resolution★ Highlight

Residual Corrective Diffusion Modeling for Km-scale Atmospheric Downscaling

Morteza Mardani, Noah Brenowitz, Yair Cohen, Jaideep Pathak, Chieh-Yu Chen, Cheng-Chin Liu, Arash Vahdat, Mohammad Amin Nabian, Tao Ge, Akshay Subramaniam, Karthik Kashinath, Jan Kautz et al.

Communications Earth & Environment · 24 September 2023 · NVIDIA

Paper PDF

Why it matters. CorrDiff introduces a residual diffusion approach to km-scale downscaling that underpins NVIDIA's later work, and it is a significant generative downscaling result.

In one sentence. Two-step UNet plus diffusion residual model downscales 25 km reanalysis to 2 km over Taiwan, recovering spectra and sharpening fronts and typhoon structures.

Abstract

The state of the art for physical hazard prediction from weather and climate requires expensive km-scale numerical simulations driven by coarser resolution global inputs. Here, a generative diffusion architecture is explored for downscaling such global inputs to km-scale, as a cost-effective machine learning alternative. The model is trained to predict 2km data from a regional weather model over Taiwan, conditioned on a 25km global reanalysis. To address the large resolution ratio, different physics involved at different scales and prediction of channels beyond those in the input data, we employ a two-step approach where a UNet predicts the mean and a corrector diffusion (CorrDiff) model predicts the residual. CorrDiff exhibits encouraging skill in bulk MAE and CRPS scores. The predicted spectra and distributions from CorrDiff faithfully recover important power law relationships in the target data. Case studies of coherent weather phenomena show that CorrDiff can help sharpen wind and temperature gradients that co-locate with intense rainfall in cold front, and can help intensify typhoons and synthesize rain band structures. Calibration of model uncertainty remains challenging. The prospect of unifying methods like CorrDiff with coarser resolution global weather models implies a potential for global-to-regional multi-scale machine learning simulation.

Labels

Labels and summary are generated by Claude and reviewed weekly. Wrong? Edit this entry.