microsoft / aurora

Implementation of the Aurora model for atmospheric forecasting
https://microsoft.github.io/aurora
Other
249 stars 31 forks source link

Implementation of loss function implementation. #50

Open KennyWu opened 2 weeks ago

KennyWu commented 2 weeks ago

I am reading the pretraining and finetuning loss function and am trying to understand the implementation of this loss: Image

I think I have a general idea of implementation, but would like to confirm in case I am not missing any details. The idea I have is as follows:

For each variable calculate the MAE loss Multiply each variable loss by specific variable weight. Sum loss of variables grouping by Surface variables and Atmospheric Variables separately Multiply Surface and Atmospheric losses by alpha and beta weights respectively Finally sum losses together and multiply by (dataset weight/(tot # variables in dataset)