??? ??? ??? ???? ?? ?? ?????

Reading Time: 5 minutes

?? ?? ?? ?? ??? ??? ?? ?? ??(NWP) ??? ??? ? ?? ??? ??? ???? ????. NVIDIA ?????(FourCastNet)? ?? ??? ?? ?? ??? ??? ?? ??? ? ???? ? ? ?? ??? ? ??? ?? ?????, ?? ??? NWP ?? ?????? ?? ??? ????.

?? ??? ?1??? ?? ????? ????? ?? ?? ??? ???? ???? ?? ?? ?? ??? ?????. ML ?? ?? ??? ??? ??? ???, ??? ??? ??? ??? GPU? ????? ?????.

??? ??? ??? ??? ??? ????? ??? ?? ??? ? ? ????:

??? ??? ??? ??? ? ?????
??? ? ???? ? ?????
? ?? ???? ????? ???? ??? ?? ? ??, ??? ? ?? ???? ?? ?? ? ?????

? ?????? ?? ??? ?? ???(SFNO), ?? ??? ???, ??? ???, ?? ??? ??(SHT)? ??? SFNO? ???? ??? ?? ?????. ??? ?? ??? ??? ICML ??? ?? ??? ?? ???? ?????: ?? ??? ?? ???: ???? ??? ?? ??? ?????.

??? ???

*?? 1. 5??? ?? SFNO? ???. SFNO? ?? ?? ???? ??? ?? ?? ??? ?? ???? ????*.

????? ??? ? ?? ??? ??? ?? ??? ?? ???? ?? ??? ????? ?? ??? ???? ??? ????? ?? ?????.

?? ??? ????? ???? ???? ??????:

??? ???? ?? ???? ??? ??? ???? ????.
?? ?? ???? ?????? ?? ?? ??? ??? ?? ??? ?????.

?? ??? ????? ?? ???? ??? ??? ?? ??????. ??? ??? ?? ???? ???? ??? ???? ?? ?????.

?? ML ?? ???? ??? ?? ??? ??? ?? ??? ?? ???? ?? ?? ??? ?? ???? ????? ??? ??? ?????. ??? ??? ?? ??? ?? ??????, ??? ??? ?? ??? ???? ?? ??? ?????.

?? ??, ???? ??? ??? ??? ? ?? ?? ??? ??? ????? ????? ???? ????. ?? ???? ??? ???? ?? ???? ???? ?? ?????, ?? ???? ??? ???? ?? ??? ??? ??? ?? ???? ??? ????? ??? ???? ???. ? ?? ??? ?? ??? ?? ?? ? ? ?? ??? ???? ???.

??? ??? ??? ?? ?? ?? ?? ?? ??? ? ??? ???? ??? ??? ? ????.

U-Net? ?? ??? ??? ????? ? ??? ??? ? ???, ??? ?? ??? ??? ??? ??? ?????. ??? ??? ?? ????, ? ??? ??? ???? ?? ?????.

U-Net? ?? ?? ???? ?? ??? ??? ???? ??? ???? ??? ?? ?????. ?? ???? ? ??? ???? ?? ???? ??????. ?? ?? ?? ??? ?? ???? ??, ? ??? ?? ??? ?? ??? ? ?? ???? ?????.

??? ??? ?? ???(FNO)? ?? ?? ?? ?? ???? ? ??? ?? ??? ???? PDE? ? ???? ????? ?? ??? ?????? ?????.

??? ???? FNO? ???? ???? ????, ?? ??? ?? ??? ????. ??? ???? ???? ??? ???? ????? ??? ???? ??? ?? ?? ???? ?? ? ???? ???? ????? ?????. ??? ????? ????? ? ???? ???? ??? ???? ?????(?? 2).

?? ??? FNO? ?? ???? ???? ????.

?? 2? ??? ??? ?? ???(AFNO)? ??? ?? ??? ?? ??? ?? ???(SFNO)? ???? ?????. ?? ??? ? ?? ??? ???? ????? ???? ???? ???? ?????.

?? ??? ?? ???

???? ??? ?? ???? ??? ???? ?? ?? ??? ?? ???? ??? ?? ???? ?? ??? ?? ???(SFNO)? ??????. ?? ?? ?? ?? ???? ???? ??? ??????.

??? ????? FNO? ?? ?? ?????. FFT? ?? ??? ????? ??? ??? ???? ??? ??? ??? ???? ??? ?? ?????.

????? ?? ?? ???? ??? ?? ????? ?? ?? ??? ??? ???? ?? ??? ??(SHT)? ?????.

?? ?? ??? ?? ????

SFNO? ???? ???? ?? ??? SHT? ??????. ?? ?? ?? SHT? ?? PyTorch ?????? ?? ????(torch-harmonics)? ??????. ? ?????? ?? ??? ?? ?? ??? ?? CPU?? ??? ?? ? ?? GPU?? SHT ??? ????? ????, ?? ???? ???? ?? ??? ? ????:

pip install torch-harmonics

?? ????(torch-harmonics)? PyTorch? ???? ?????. ???? SHT? ?? ML ????? ??? ?? ??? ? ????. ?? ??? SHT? ????? ?? ?? ??? ?????:

import torch
import torch_harmonics as th
 
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
# parameters
nlat = 512
nlon = 2*nlat
batch_size = 32
signal = torch.randn(batch_size, nlat, nlon)
 
# create SHT instance
sht = th.RealSHT(nlat, nlon).to(device).float()
 
# execute transform
coeffs = sht(signal)

?? ???(torch-harmonics)? ????? ?? ?? ??? ?? ??? ?? ??? ???? ???? ???? ???? ?? ????(?? 3). ? ????? SHT? ISHT? ???? ?? ???? ??? ???? ??? ?????.

*?? 3. ?? ???? ??? ?? ?? ??? ?? ??? ??(??). ??? ??? ???? ??? ???? ??(???).*

?? ?? ?? ?? ??? ?? ???

??? ?? ?? ?? ??(ECMWF)?? ???? ERA5 ??? ??? SFNO? ???????. ? ??? ??? ?? 44??? ?? ?? ??? ?? ? ???? ?? ??? ?????. ?? 2? ?? SFNO? ????? ??? ??? ??? ???, ?? 1? ?? ?? ?? ?? ?? ??? ???? ???? ??? ??? ????? ????? ?? ? ? ????(?? 1).

??? ??? ?? ?? ?? ?? ?? ??? ??? ?? ?? ?????. ? ??? ??? ????? ???? ML ?? ??? ?? ??? ?? ?? ??? ??? ?? ? ?? ??? ?? ??? ?? ? ?? ????.

1460?? ?? ?? ??? ???? 1??? SFNO ?? ???? ?? NVIDIA RTX A6000?? 13? ?? ??? ? ????. ?? ??? ?? ?? ?? ???? ? ? ?? ?? ?????.

??? ?? ? ?? ?? ??? ?? NWP? ???? ??? ????? ???? ? ??? ??? ?? ?? ??? ????? ??? ? ?? ?? ????, ???? ???? ? ????? ??? ?? ? ?? ???? ??? ? ?? ????.