Probabilistic Global Weather
Predict a range of possible global weather futures.
Task & evaluation
- Objective
- Produce 15-day global ensemble weather forecasts at 0.25° resolution.
- Input
- Two global context frames separated by six hours.
- Output
- 56 ensemble members at 12-hour leads through 15 days.
- Data
- Atmospheric snapshots, weather analyses and cyclone records.
- Evaluation
- Accuracy and uncertainty measured with CRPS; rainfall and storms checked separately.
- Research challenge
- Keep fifteen-day forecasts coherent while representing uncertainty and rare events.
Understanding the task
A useful weather forecast needs more than a single best guess. This task trains a model to produce many possible futures, testing whether their uncertainty stays meaningful over fifteen days.
Forecast the storm and its uncertainty
An emergency planner needs to know the chance of damaging winds, not just the average wind speed. An ensemble represents this uncertainty with multiple possible futures. Its members should cover plausible outcomes without spreading so widely that the forecast stops being informative.
This task uses the WeatherNext 2 model family to generate 56 futures from two atmospheric snapshots six hours apart. Predictions cover the globe at twelve-hour intervals out to fifteen days. A plausible storm must also hold together across places and times: individually reasonable temperatures do not guarantee a coherent weather system.
From two weather snapshots to many possible futures
One model generates many possible weather trajectories. The diagram shows their information flow, not a measured weather event.
Atmosphere at 13 pressure levels + surface fields
Learn accuracy and uncertainty together
Report 12 h → 360 h
Storms / spectra / rainfall checked separately
Learn coarse patterns before long forecasts
Open a stage to see how the learning problem changes. Bars show the share of training updates; a fine-grid step costs more than a coarse-grid step.
410 updates · ERA5 1°, 12 h
Learn one-step weather evolution at coarse resolution. 70.7% of updates occur here, establishing a starting representation.
102 updates · ERA5 1°, 6 h
Predict at shorter intervals while building on what the model has already learned.
32 updates · ERA5 0.25°, 6 h
Move to the fine global grid. Resolution and mesh refinement change; finer detail increases the work per update.
8 updates · HRES AR1
Adapt to HRES fields using one autoregressive step. This is a change in data source as well as in curriculum stage.
28 updates · HRES AR2 → AR8
Continue from the model’s own forecasts so it learns to cope with accumulated errors. Training reaches eight successive steps; evaluation asks for a much longer forecast.
Source: Training stages.
Follow an ensemble into its probability score
Synthetic temperature anomalies at one grid cell. These nine illustrative members are not WeatherNext output; the benchmark requires 56 members on the full globe.
1 · A slice through the rollout
2 · From members to CRPS
CRPS = meanᵢ |xᵢ − y| − ½ meanᵢⱼ |xᵢ − xⱼ| = ∫ (F(z) − 1{y ≤ z})² dz.
Try collapsing the spread, then adding drift. A concentrated forecast can be confidently wrong; excessive spread also costs score. A single outcome cannot establish calibration, which requires many forecast–observation pairs.
From one location to the globe. This local score is in °C. The task divides each field’s CRPS by its frozen training standard deviation, aggregates native-cell scores over levels and leads and then ten variables, and reports reward = 1 / (1 + normalized CRPS). Spatial structure, extremes, cyclone tracks and precipitation are separate checks; a good marginal score does not certify them.
A small training sample of a large atmosphere
Training uses 32 weather windows from 2017–2020, with four dates in 2021 for development and eight dates in 2022 for evaluation. Inputs describe conditions near the surface and at thirteen pressure levels, allowing the model to connect local weather with circulation higher in the atmosphere.
The native grid is detailed, but the sample of dates is small. It cannot establish year-round skill or comprehensive coverage of rare extremes. The input sources also differ from the original paper’s operational setup. The test panel helped select the starter recipe, so its scores are development evidence rather than an untouched confirmation.
Reward useful uncertainty
The continuous ranked probability score, or CRPS, compares the forecast distribution with what happened. It balances closeness to the observation with ensemble spread. A tightly clustered forecast can be confidently wrong; an excessively broad forecast pays a penalty too.
Errors are normalized so variables with different units can contribute fairly. Lower normalized CRPS gives higher reward through 1 / (1 + error). Rainfall, cyclone tracks and spatial structure are checked separately, because a good average probability score can still hide an important failure in a storm.
Learn short steps, then learn to live with errors
The approximately 184-million-parameter model starts from random weights. Training first teaches weather evolution on a coarse grid, then introduces finer detail and different input fields. Later stages ask it to continue from its own predictions, rather than always receiving a correct observed state.
This matters because small mistakes change the next input and can grow throughout a long forecast. Training on self-generated states exposes the model to that difficulty. Its learning objective compares two possible forecasts with the same observation, encouraging both accuracy and meaningful diversity. The starter uses a very short training schedule compared with a full-scale operational effort.
A small gain needs a careful interpretation
The extended starter reduces normalized CRPS from about 0.480 to 0.469, an improvement of roughly 2.2%; reward rises from 0.676 to 0.681. A matched development comparison also lowers ten-day temperature, wind and pressure errors.
The extension includes more training computation, so it does not isolate an architectural improvement. It also uses one trained model, whereas the paper combines four. These results describe progress within this small task, not evidence of paper-level forecasting skill.
| Ten-day diagnostic | Before extension | After extension |
|---|---|---|
| Wind-speed RMSE · m/s | 9.927 | 9.682 |
| 2 m temperature RMSE · °C | 4.913 | 4.779 |
| Sea-level pressure RMSE · hPa | 15.31 | 14.33 |
Better averages can hide a worse storm track
One-day cyclone-position error increases slightly, from about 438 to 441 km, even as the main score improves. At longer leads, errors remain large. This is precisely why storm tracks are examined separately: a forecast can improve across many grid cells while placing a dangerous event in the wrong location.
Many reported comparisons share the same dates and weather fields. They are related observations, not thousands of independent tests. More dates and repeated training runs would be needed to assess how dependable the improvement is.
Detailed results
| Diagnostic | 562 updates → 580 updates | Interpretation |
|---|---|---|
| Native / derived CRPS panels | 2,827 of 2,969 lower with extended model | Broad improvement, not 2,969 independent trials. |
| One-day cyclone position error | 438.06 → 440.86 km; 9 matched initializations | Worsens despite better aggregate CRPS. |
| Three-day cyclone position error | 1,225.04 → 1,221.87 km; 7 matched initializations | Small improvement on a different subset. |
| Five-day cyclone position error | 1,661.27 → 1,653.07 km; 7 matched initializations | Errors remain large; do not compare unmatched storm subsets. |
Compute and resources
Main training recipe
Reported measurement
- Time
- ≈21.9 h across timed stages
- Hardware
- 4 × RTX PRO 6000
- GPU allocation
- ≈87.5 GPU-h
Combined allocation for the initial training stages and extended rollout learning.
GPU allocation by stage
- Initial stages
- 41.14 GPU-h
- Extended rollout training
- 46.37 GPU-h
Sum of measured stages, equivalent to about 21.9 hours on four GPUs. It is not one uninterrupted run; preparation and evaluation are excluded.
Resource details
- Independent evaluation
- 5.04 + 13.81 GPU-hoursRecorded allocations on RTX PRO 6000 and RTX A6000 respectively; different hardware types are kept separate.
- CPU evaluation
- 347 CPU-hoursAllocated CPU time for output checks, scoring and paired comparisons; this is not elapsed wall time.
Keep uncertainty and physical structure together
The next question is whether longer-rollout training improves both uncertainty and storm evolution. Compare forecast spread, late-lead drift and cyclone behavior together. Choosing methods on development dates and confirming them on fresh dates would make the next claimed improvement more convincing.
References and provenance
Skillful joint probabilistic weather forecasting from marginals
Explains the learning objective for producing ensembles that capture uncertainty across the globe.
WeatherBench 2 data guide
Documents the ERA5 and HRES atmospheric data used in weather-model evaluation.
International Best Track Archive for Climate Stewardship
Provides observed tropical cyclone tracks for assessing predicted storm trajectories.
Task definition and experimental evidence
Documents the model and evaluation used for the measurements on this page, distinct from the full published forecasting system.
Code links require access to the project’s GitHub repository.
Leaderboard
SimulatedDesign preview · leaderboard values and research histories are simulated.
| # | Method | Type | Forecast reward ↑ | Normalized CRPS ↓ | Illustrative event score ↑ | Research time | Run |
|---|---|---|---|---|---|---|---|
| 1 | Agent search | 0.6589 | 0.5176 | 0.6841 | 76 h | ||
| 2 | Agent search | 0.6577 | 0.5205 | 0.6828 | 60 h | ||
| 3 | WeatherNext 2 (11-stage AR) | Reference | 0.6550 | 0.5267 | 0.6800 | — | — |
How to read these leaderboard metrics
These leaderboard values and research histories are simulated design examples. The leaderboard is ranked by Forecast reward. The research curves show relative improvement in this metric over WeatherNext 2 (11-stage AR); only accepted new-best results update the retained trajectory. A positive gain means a better result, with the sign adjusted for metrics where lower is better. The task protocol is described in Task & evaluation above. These preview columns and trajectories are not measured benchmark results and must not be compared with the recorded experiments. Leaderboard rows and search traces are simulated interface examples, not measured submissions. The recorded 580-update starter has normalized CRPS 0.469183 and reward 0.680651; historical recipe selection used this test panel. Extreme-event diagnostics are separate from the reward and do not define a single official aggregate “extreme-event skill”.
- Forecast rewardPrimary metric · Higher is better
- The task-specific aggregate forecast score. Its component metrics, reference normalization and evaluation horizon determine how improvements should be interpreted.
- Normalized CRPSLower is better
- Continuous ranked probability score, assessing the forecast distribution against the observed outcome. It evaluates probabilistic forecasts rather than only their mean.
- Illustrative event scoreHigher is better
- Simulated interface value only. The actual task reports separate event diagnostics; there is no documented single aggregate event score.
AI4AI research progress
Research trajectory
Runtime–performance frontier
How improvement is measured
Relative improvement over WeatherNext 2 (11-stage AR) in the primary metric; only retained results update the best-so-far curve.
Accepted new-best experiments raise the trajectory. The frontier includes evaluated candidates; trial, partial and stopped experiments are excluded.