Graph Machine Learning
AnemoI
English
anaprietonem commited on
Commit
f11903e
1 Parent(s): 245670d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -35
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]](https://doi.org/10.48550/arXiv.2212.12794) 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,39 +162,29 @@ state.
162
 
163
  <!-- This section describes the evaluation protocols and provides the results. -->
164
 
165
- ### Testing Data, Factors & Metrics
166
-
167
- #### Testing Data
168
-
169
- <!-- This should link to a Dataset Card if possible. -->
170
-
171
- {{ testing_data | default("[More Information Needed]", true)}}
172
-
173
- #### Factors
174
-
175
- <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
176
 
177
- {{ testing_factors | default("[More Information Needed]", true)}}
178
-
179
- #### Metrics
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
- {{ model_examination | default("[More Information Needed]", true)}}
 
 
 
 
 
 
 
 
 
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