Source-linked AI summary

West-WRF AI 2-km: High-Resolution Prediction of Integrated Vapor Transport and Precipitation

Nazak Rouzegari, Vesta Afzali Gorooh, Agniv Sengupta, Phu Nguyen, Kuo-Lin Hsu, Amir AghaKouchak, Soroosh Sorooshian, F. Martin Ralph, Luca Delle Monache

arXiv:2609.25512v1physics.ao-phcs.AI

TL;DR

Precipitation forecasting is difficult where models inadequately represent topography, small-scale atmospheric processes, and localized convective dynamics. The paper develops West-WRF AI 2-km, and reports its clearest advantages for localized terrain-sensitive extremes and intense IVT at longer lead times.

  • Problem

    Precipitation forecasting remains difficult where models inadequately represent topography, small-scale atmospheric processes, and localized convective dynamics.

    Source: §1 · p. 2
  • Method

    The study develops West-WRF AI 2-km, building on the previous CW3E West-WRF AI 6-km framework.

    Source: §1 · p. 3
  • Results

    The 2-km configuration’s advantage is concentrated over complex terrain and the coast, with value for localized terrain-sensitive extreme AR precipitation and intense IVT at longer lead times.

    Source: §4 · pp. 25, 28
  • Takeaways & Limitations

    West-WRF AI 2-km’s greatest value is for localized precipitation extremes and intense AR-related moisture transport, rather than uniform improvement.

    Source: §4 · pp. 25, 28

    The model is evaluated deterministically to isolate resolution refinement; probabilistic prediction is a logical next step.

    Source: §4 · p. 27

Abstract

from arXiv · show

We introduce a stretched-grid artificial intelligence (AI) weather forecasting model with 2-km resolution over the western United States and part of the Northeast Pacific and approximately 31-km resolution elsewhere globally. Forecasting over the western U.S. is challenging because complex topography and atmospheric rivers (ARs) strongly influence orographic precipitation. West-WRF AI 2-km builds on a global model pretrained with a 40-year European Centre for Medium-Range Weather Forecasts Reanalysis v5 (ERA5) dataset and is fine-tuned with the Center for Western Weather and Water Extremes (CW3E) 2-km regional reanalysis to produce autoregressive 6-hourly forecasts of precipitation and integrated vapor transport (IVT). Forecasts are evaluated over winters 2020-2023 using gridded precipitation observations, rain gauges, and AR Reconnaissance dropsondes and are benchmarked against coarser-resolution AI forecasts and regional and global numerical weather prediction (NWP) systems. West-WRF AI 2-km reproduces observed precipitation-intensity distributions, retains fine-scale spectral variability, and produces sharper narrow coastal precipitation bands and localized, terrain-sensitive extremes. Its broader-scale performance remains comparable to coarser-resolution configurations while preserving large-scale skill despite higher resolution. Dropsonde verification shows lower errors and improved categorical skill at the most extreme IVT threshold. Overall, West-WRF AI 2-km provides its greatest value for localized precipitation extremes and intense AR-related moisture transport.

1. Introduction

Forecasting precipitation over the western U.S. is difficult because complex terrain and small-scale atmospheric processes shape local precipitation. The paper introduces West-WRF AI 2-km to focus on AR-related IVT and orographic precipitation while retaining global coverage.

1. Introduction

  • Orographic precipitation is difficult to simulate because it spans large-scale moisture transport, mesoscale turbulence, and cloud microphysics, which coarse-resolution models struggle to capture.These processes contribute to sharp spatial contrasts in rainfall and snowfall.Source: §1 · p. 2
  • Regional NWP can represent fine-scale processes better but is computationally expensive and parameterization-dependent, while coarse global models may underestimate extreme precipitation during ARs.Coarse global AI models also have limited representation of extremes, complex terrain, and regional AR-related precipitation over the western U.S.Source: §1 · p. 3
  • Stretched-grid models retain coarser global coverage while increasing resolution over a target region, preserving large-scale context while resolving local terrain and mesoscale processes.The introduction describes this approach as particularly suitable for AR prediction.Source: §1 · p. 3
  • The study develops West-WRF AI 2-km, with approximately 2-km resolution over the western U.S. and Northeast Pacific and approximately 31-km resolution elsewhere globally.It focuses on precipitation and IVT during AR events.Source: §1 · pp. 3–4
  • West-WRF AI 2-km is described as the first AI weather forecasting system evaluated specifically for AR prediction at approximately 2-km resolution over the region.The system focuses on AR-related IVT and orographic precipitation over the western U.S.Source: §1 · p. 4

2. Methodology and Data Sources

This section describes the data, staged training, and stretched-grid forecasting architecture behind West-WRF AI 2-km, along with the observations and benchmark systems used for evaluation. It also defines the verification scope, selected atmospheric-river events, skill metrics, and spatial-block bootstrap procedure.

  • Fine-tuning uses 2012–2020 reanalysis, with winters 2020–2023 held out; benchmarks include West-WRF AI 6-km, IFS 9-km, and West-WRF 3-km/9-km.Source: a. Training Datasets · p. 4, f. Benchmark Models · pp. 7–8
  • The workflow pretrains a 31-km global model on 40 years of ERA5, fine-tunes it with CW3E 2-km reanalysis, then trains with an eight-step autoregressive rollout.At inference, the model runs for up to 24 six-hour steps (144 hours).Source: b. Training Framework · p. 5
  • The Anemoi Graph Transformer encodes 172 input channels on an icosahedral graph, applies processor-mesh message passing, and decodes forecasts to the latitude–longitude grid.Source: c. Stretched-grid Graph Transformer · p. 6
  • Precipitation is evaluated against daily PRISM gridded observations and quality-controlled hourly CNRFC gauges, including 526 valid stations across the common model domain.Source: e. Observational Datasets · p. 7
  • IVT verification uses about 2,360 AR Reconnaissance dropsonde profiles, an independent reference derived directly from observed moisture and wind profiles rather than reanalysis.Source: e. Observational Datasets · p. 7
  • Evaluation covers lead times up to 120 hours and compares both the 2-km domain and its surrounding 31-km global-grid region.Source: g. Model Evaluation · p. 8

AR-Based Precipitation Events

  • AR-related events use Ralph et al. (2019)’s scale and rank ≥ 2 San Francisco cases from Baño-Medina et al. (2025), spanning November 2020–March 2023.The events’ maximum IVT ranges from approximately 498–1095 kg m⁻¹ s⁻¹.Source: 1) AR-BASED PRECIPITATION EVENTS · p. 8
  • Verification assesses RMSE, CC, ETS, RAPSD, FSS, POD, FAR, and CSI using forecast and observation fields.Source: 2) METRICS · p. 9

Confidence Intervals

  • 90% confidence intervals use paired 33 × 33-cell spatial-block bootstrap resampling for 1,000 iterations, with block size based on PRISM’s 132-km decorrelation distance.The 5th and 95th percentiles define the intervals; the same resampled blocks are used across forecast systems.Source: 2) METRICS · p. 9
  • ETS assesses event detection using a fixed NWS precipitation threshold of 0.25 mm day⁻¹ and wet-day climatological percentile thresholds P50, P90, and P98 for November–March; FSS evaluates spatial skill as a function of neighborhood size.Source: 2) METRICS · pp. 9–11
  • RAPSD standardizes precipitation fields, computes a 2-D Fourier power spectrum, radially averages it, and converts wavenumber to spatial wavelength using grid spacing.Source: 2) METRICS · p. 10
  • IVT event thresholds are 250, 500, and 750 kg m⁻¹ s⁻¹, while gauge-based precipitation events use each station’s 2010–2023 climatological 95th percentile.POD, FAR, and CSI assess detection, false alarms, and combined detection skill, respectively.Source: 2) METRICS · p. 11

3. Results

The results evaluate West-WRF AI 2-km precipitation and IVT forecasts using gridded observations, neighborhood-scale verification, case studies, gauges, and AR Reconnaissance dropsondes. They examine forecast skill across precipitation structure and intensity, lead times, and extreme events.

a. Verification Against Gridded Observations

Results

  • At the 600-km window, West-WRF AI 2-km ranked second in FSS behind West-WRF 3-km, exceeding IFS 9-km by 11.6% at 2 days and 20.1% at 4 days.AI 2-km ranked among the two best-performing forecasts across lead times and window sizes.Source: a. Verification Against Gridded… · p. 17, Figure 5 · p. 18
  • For the AR Scale 2 case, West-WRF AI 2-km had the lowest spatial RMSE at 5.52 mm, 33.3% below IFS 9-km and 18.7% below West-WRF 3-km.It captured the narrow coastal band and localized maxima with strong spatial agreement to PRISM despite a domain-mean overestimate of 7.5%.Source: a. Verification Against Gridded… · pp. 18–19, Figure 6 · p. 19
  • For the AR Scale 3 case, West-WRF AI 2-km’s spatial RMSE was 9.64 mm, approximately 28.0% below the 6-km forecast’s 13.38 mm.The 2-km forecast resolved both localized high-intensity cores with sharper gradients and peak magnitudes closer to PRISM, whereas the 6-km forecast smoothed them.Source: a. Verification Against Gridded… · pp. 19–20, Figure 7 · p. 20

b. Verification Against Gauge-Based Observations

  • For 6-hour accumulated precipitation forecasts verified against gauges, both West-WRF AI forecasts retained CC ≈ 0.61 at 120 hours, versus 0.53 for IFS 9-km; RMSE rises and CC falls with lead time.Source: b. Verification Against Gauge-Based… · p. 21, Figure 8 · p. 22
  • At 24 hours, gauge-based POD for 95th-percentile precipitation events was highest for West-WRF AI 2-km at 0.516, and CSI was 0.324 for AI 2-km and 0.320 for AI 6-km.At 120 hours, CSI was 0.186 for AI 6-km and 0.135 for AI 2-km, versus 0.105 for IFS 9-km and 0.084 for West-WRF 3-km; the AI forecasts’ FAR performance traded off by lead time.Source: b. Verification Against Gauge-Based… · pp. 21–22

c. AR Reconnaissance Dropsonde IVT

  • Dropsonde-verified IVT RMSE increased with lead time for the evaluated forecasts.The passage identifies 2,360 AR Reconnaissance dropsonde samples and lead times of 24–120 hours.Source: c. AR Reconnaissance Dropsonde IVT · p. 23

3. Results

4. Discussion and Conclusions

West-WRF AI 2-km’s benefits are concentrated in localized, terrain-sensitive extremes and intense AR-related IVT, while broader-scale performance remains comparable to coarser configurations. The discussion also identifies verification, training, and domain boundaries that qualify the comparison.

4. Discussion and Conclusions

  • The 2-km configuration better represents upper-percentile precipitation intensity than 6-km, without the stronger upper-tail overestimation seen in West-WRF 3-km.Differences between the AI configurations are small near the median and grow at upper percentiles.Source: §4 · pp. 25–26
  • For the December 31, 2022 case, 2-km sharpened localized cores and had lower spatial RMSE than 6-km, which smoothed and displaced the most intense core.The reported spatial RMSEs were 9.64 mm for 2-km and 13.38 mm for 6-km.Source: §4 · p. 26
  • Gauge verification found AI forecasts competitive with or better than dynamical forecasts across much of the five-day period, but neither AI configuration was uniformly best for station-specific extremes.The 120-hour reversal involves rare station-specific 95th-percentile events with limited effective sample size.Source: §4 · pp. 26–27
  • The configurations differ in both regional resolution and outer-domain setup, so controlled experiments are needed to isolate the effect of inner-grid refinement.Reanalysis initialization may advantage West-WRF AI relative to dynamical forecasts initialized from operational analyses; operational-analysis testing would better represent real-time performance.Source: §4 · pp. 26–27
  • At short IVT lead times, lower-percentile precipitation intensity, and large-neighborhood skill, 2-km performs comparably to—not clearly better than—6-km.The value of refinement depends on forecast variable, spatial scale, lead time, verification method, and application.Source: §4 · pp. 26, 28
  • Pointwise verification can penalize displaced sharp features twice, while neighborhood verification tolerates small displacements; MSE training favors spatially averaged predictions under positional uncertainty.The displacement sensitivity of sharper 2-km features grows with lead time.Source: §4 · pp. 26–27
  • The study is a deterministic baseline; the 2-km refinement and fine-tuning are limited to the CW3E reanalysis domain, and ERA5 training versus operational-analysis initialization remains unevaluated.The authors identify ensemble and probabilistic prediction, wider 2-km coverage, and operational-analysis initialization as future directions.Source: §4 · pp. 27–28
  • The clearest 2-km benefits are for localized, terrain-sensitive extreme AR precipitation and intense IVT, while large-scale skill shows no evident reduction.The two configurations remain comparable for domain-mean quantities, broader spatial scales, lower precipitation thresholds, and short-lead IVT.Source: §4 · p. 28
Loading 2609.25512v1…