anaprietonem
commited on
Commit
•
f11903e
1
Parent(s):
245670d
Update README.md
Browse files
README.md
CHANGED
@@ -27,10 +27,6 @@ are available to the public under ECMWF’s open data policy. (https://www.ecmwf
|
|
27 |
AIFS is based on a graph neural network (GNN) encoder and decoder, and a sliding window transformer processor,
|
28 |
and is trained on ECMWF’s ERA5 re-analysis and ECMWF’s operational numerical weather prediction (NWP) analyses.
|
29 |
|
30 |
-
<div style="display: flex; justify-content: center;">
|
31 |
-
<img src="aifs_diagram.png" alt="High-level AIFS diagram" style="width: 50%;"/>
|
32 |
-
</div>
|
33 |
-
|
34 |
<div style="display: flex;">
|
35 |
<img src="encoder_graph.jpeg" alt="Encoder graph" style="width: 50%;"/>
|
36 |
<img src="decoder_graph.jpeg" alt="Decoder graph" style="width: 50%;"/>
|
@@ -110,6 +106,10 @@ aifs_forecast.to_xarray()
|
|
110 |
AIFS is trained to produce 6-hour forecasts. It receives as input a representation of the atmospheric states
|
111 |
at \\(t_{−6h}\\), \\(t_{0}\\), and then forecasts the state at time \\(t_{+6h}\\).
|
112 |
|
|
|
|
|
|
|
|
|
113 |
The full list of input and output fields is shown below:
|
114 |
|
115 |
| Field | Level type | Input/Output |
|
@@ -130,7 +130,7 @@ the forcing variables, like orography, are min-max normalised.
|
|
130 |
of 260,000 steps. The LR is increased from 0 to \\(10^{-4}\\) during the first 1000 steps, then it is annealed to a minimum
|
131 |
of \\(3 × 10^{-7}\\).
|
132 |
- **Fine-tuning I**: The pre-training is then followed by rollout on ERA5 for the years 1979 to 2018, this time with a LR
|
133 |
-
of \\(6 × 10^{-7}\\). As in [Lam et al. [2023]](
|
134 |
rollout every 1000 training steps up to a maximum of 72 h (12 auto-regressive steps).
|
135 |
- **Fine-tuning II**: Finally, to further improve forecast performance, we fine-tune the model on operational real-time IFS NWP
|
136 |
analyses. This is done via another round of rollout training, this time using IFS operational analysis data
|
@@ -162,39 +162,29 @@ state.
|
|
162 |
|
163 |
<!-- This section describes the evaluation protocols and provides the results. -->
|
164 |
|
165 |
-
|
166 |
-
|
167 |
-
|
168 |
-
|
169 |
-
|
170 |
-
|
171 |
-
|
172 |
-
|
173 |
-
#### Factors
|
174 |
-
|
175 |
-
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
176 |
|
177 |
-
|
178 |
-
|
179 |
-
|
180 |
-
|
181 |
-
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
182 |
-
|
183 |
-
{{ testing_metrics | default("[More Information Needed]", true)}}
|
184 |
-
|
185 |
-
### Results
|
186 |
-
|
187 |
-
{{ results | default("[More Information Needed]", true)}}
|
188 |
-
|
189 |
-
#### Summary
|
190 |
-
|
191 |
-
{{ results_summary | default("", true) }}
|
192 |
-
|
193 |
-
## Model Examination [optional]
|
194 |
|
195 |
-
<!-- Relevant interpretability work for the model goes here -->
|
196 |
|
197 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
198 |
|
199 |
## Technical Specifications
|
200 |
|
|
|
27 |
AIFS is based on a graph neural network (GNN) encoder and decoder, and a sliding window transformer processor,
|
28 |
and is trained on ECMWF’s ERA5 re-analysis and ECMWF’s operational numerical weather prediction (NWP) analyses.
|
29 |
|
|
|
|
|
|
|
|
|
30 |
<div style="display: flex;">
|
31 |
<img src="encoder_graph.jpeg" alt="Encoder graph" style="width: 50%;"/>
|
32 |
<img src="decoder_graph.jpeg" alt="Decoder graph" style="width: 50%;"/>
|
|
|
106 |
AIFS is trained to produce 6-hour forecasts. It receives as input a representation of the atmospheric states
|
107 |
at \\(t_{−6h}\\), \\(t_{0}\\), and then forecasts the state at time \\(t_{+6h}\\).
|
108 |
|
109 |
+
<div style="display: flex; justify-content: center;">
|
110 |
+
<img src="aifs_diagram.png" alt="High-level AIFS diagram" style="width: 80%;"/>
|
111 |
+
</div>
|
112 |
+
|
113 |
The full list of input and output fields is shown below:
|
114 |
|
115 |
| Field | Level type | Input/Output |
|
|
|
130 |
of 260,000 steps. The LR is increased from 0 to \\(10^{-4}\\) during the first 1000 steps, then it is annealed to a minimum
|
131 |
of \\(3 × 10^{-7}\\).
|
132 |
- **Fine-tuning I**: The pre-training is then followed by rollout on ERA5 for the years 1979 to 2018, this time with a LR
|
133 |
+
of \\(6 × 10^{-7}\\). As in [Lam et al. [2023]](doi: 10.21957/slk503fs2i) we increase the
|
134 |
rollout every 1000 training steps up to a maximum of 72 h (12 auto-regressive steps).
|
135 |
- **Fine-tuning II**: Finally, to further improve forecast performance, we fine-tune the model on operational real-time IFS NWP
|
136 |
analyses. This is done via another round of rollout training, this time using IFS operational analysis data
|
|
|
162 |
|
163 |
<!-- This section describes the evaluation protocols and provides the results. -->
|
164 |
|
165 |
+
AIFS is evaluated against ECMWF IFS (Integrated Forecast System) for 2022. The results of such evaluation are summarized in
|
166 |
+
the scorecard below that compares different forecast skill measures across a range of
|
167 |
+
variables. For verification, each system is compared against the operational ECMWF analysis from which the forecasts
|
168 |
+
are initialised. In addition, the forecasts are compared against radiosonde observations of geopotential, temperature
|
169 |
+
and windspeed, and SYNOP observations of 2 m temperature, 10 m wind and 24 h total precipitation. The definition
|
170 |
+
of the metrics, such as ACC (ccaf), RMSE (rmsef) and forecast activity (standard deviation of forecast anomaly,
|
171 |
+
sdaf) can be found in e.g Ben Bouallegue et al. ` [2024].
|
|
|
|
|
|
|
|
|
172 |
|
173 |
+
<div style="display: flex; justify-content: center;">
|
174 |
+
<img src="aifs_v021_scorecard.png" alt="Scorecard comparing forecast scores of AIFS versus IFS (2022)" style="width: 80%;"/>
|
175 |
+
</div>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
176 |
|
|
|
177 |
|
178 |
+
. Forecasts are initialised on 00 and 12
|
179 |
+
UTC. Shown are relative score changes as function of lead time (day 1 to 10) for northern extra-tropics (n.hem),
|
180 |
+
southern extra-tropics (s.hem), tropics and Europe. Blue colours mark score improvements and red colours score
|
181 |
+
degradations. Purple colours indicate an increased in standard deviation of forecast anomaly, while green colours
|
182 |
+
indicate a reduction. Framed rectangles indicate 95% significance level. Variables are geopotential (z), temperature
|
183 |
+
(t), wind speed (ff), mean sea level pressure (msl), 2 m temperature (2t), 10 m wind speed (10ff) and 24 hr total
|
184 |
+
precipitation (tp). Numbers behind variable abbreviations indicate variables on pressure levels (e.g., 500 hPa), and
|
185 |
+
suffix indicates verification against IFS NWP analyses (an) or radiosonde and SYNOP observations (ob). Scores
|
186 |
+
shown are anomaly correlation (ccaf), SEEPS (seeps, for precipitation), RMSE (rmsef) and standard deviation of
|
187 |
+
forecast anomaly (sdaf, see text for more explanation).
|
188 |
|
189 |
## Technical Specifications
|
190 |
|