2023

GraphCast: Learning Skillful Medium-Range Global Weather Forecasting

R. Lam, et al.

Observation and Context

Traditional global weather forecasting relies on physics-based Numerical Weather Prediction (NWP) systems, such as the European Centre for Medium-Range Weather Forecasts (ECMWF) High RESolution (HRES) deterministic model. While these models are highly precise and have improved steadily with expanding supercomputer capabilities, they possess a critical conceptual flaw: their core physics routines cannot directly absorb large archives of historical weather data to fix systematic simulation biases or discover unrecognized atmospheric patterns. Furthermore, running these models requires massive computing clusters and takes approximately an hour to output a single 10-day medium-range projection.

Hypothesis

The researchers engineered GraphCast, a purely machine learning-based weather prediction (MLWP) architecture. They hypothesized that a specialized Graph Neural Network (GNN) configured in an “encode-process-decode” topology could ingest large-scale historical reanalysis datasets and effectively learn the intricate, multi-layered partial differential equations governing planetary fluid dynamics. They posited that GraphCast could generate medium-range global forecasts with equal or superior skill to state-of-the-art NWP systems while using a fraction of the traditional computational runtime and energy.

Experiment and Methodology

To rigorously evaluate the framework, the authors devised the following experimental protocol:

  • Training and Optimization: The network was trained using 39 years (1979–2017) of the ECMWF’s ERA5 historical reanalysis archive. It optimized an autoregressive Mean Squared Error (MSE) loss function over a sequential curriculum that incremented up to 12 simulation steps. Training was completed using batch parallelism across 32 Cloud TPU v4 devices over a 4-week duration.
  • Architecture: GraphCast models 227 target variables mapped onto a global 0.25° regular grid. The encoder projects these parameters onto a uniform internal “multi-mesh” graph built from an icosahedron refined six times. The processor executes long-range message passing across 16 unshared GNN layers. The decoder returns the localized latent representations back to a geographical latitude-longitude grid to yield a residual physical change forecast.
  • Verification: The system’s predictive performance was tested out-of-sample on data from 2018 onward. The models were validated against both operational HRES forecasts and baseline machine learning implementations (like Pangu-Weather) across 1,380 verification metrics evaluated by Root Mean Squared Error (RMSE) and Anomaly Correlation Coefficient (ACC).

Results and Data

The testing phase revealed significant skill gains and unprecedented operational performance:

  • Widespread Accuracy Gains: GraphCast significantly outperformed the top deterministic HRES operational system on 90.3% of the 1,380 total verification targets. When excluding high-altitude stratospheric boundaries (50 hPa), GraphCast’s superiority rose to 96.9% of metrics.
  • Unparalleled Computational Efficiency: GraphCast produced a full, high-resolution 10-day global weather forecast in less than 60 seconds using a single Google Cloud TPU v4 device, bypassing the massive cluster hardware traditionally required.
  • ML Benchmark Dominance: When matched against competing high-resolution machine learning baselines (Pangu-Weather) utilizing identical 00z/12z initializations, GraphCast achieved better absolute RMSE across 99.2% of the designated comparison fields.
  • Severe Event Tracking: The architecture exhibited superior track precision over HRES for extreme weather, achieving lower median tropical cyclone track deviations from 18 hours out to 4.75 days. It also registered a 10% to 25% structural error reduction when modeling Vertically Integrated Water Vapor Transport (IVT) across western North American atmospheric rivers.

Conclusion and Climate Impact

The experiment demonstrated that data-driven Graph Neural Networks are highly capable of modeling complex planetary dynamics, marking a distinct milestone in modern meteorology where machine learning matches or exceeds centuries of explicit physical formulation.

Because GraphCast decouples intense computational overhead from final inference times, accurate global medium-range forecasting can now be processed cost-effectively without requiring high-performance supercomputing centers. This democratized access allows local jurisdictions and key economic sectors — such as agriculture, grid energy coordination, and logistics — to run instantaneous, localized risk assessments. Furthermore, its enhanced capacity for classifying extreme heatwaves, deep freezes, and cyclone trajectories translates directly into earlier early-warning distributions, protecting vulnerable populations and infrastructure under a changing climate.

Citation

Lam, R., Sanchez-Gonzalez, A., Willson, M., Wirnsberger, P., Fortunato, M., Alet, F., et al. (2023). GraphCast: Learning skillful medium-range global weather forecasting. arXiv preprint arXiv:2212.12794v2 [cs.LG].