zhangrenchao commited on
Commit
e2e33d8
·
verified ·
1 Parent(s): b72f6a3

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +86 -0
README.md ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Aerosol Forecasting
9
+ - Air Quality
10
+ - Vision Transformer
11
+ - Relay Forecasting
12
+ frameworks: PyTorch
13
+ ---
14
+
15
+ <p align="center"><strong><span style="font-size: 30px;">AI-GAMFS</span></strong></p>
16
+
17
+ # Model Introduction
18
+
19
+ AI-GAMFS is a machine-learning global aerosol-meteorology forecasting system producing five-day forecasts at three-hour intervals.
20
+
21
+ Paper: Advancing operational global aerosol forecasting with machine learning
22
+ https://doi.org/10.1038/s41586-026-10234-y
23
+
24
+ # Model Description
25
+
26
+ The model was proposed by teams from the Chinese Academy of Meteorological Sciences, National Meteorological Center, NASA, and collaborators. It was trained with 54 MERRA-2 aerosol and meteorological variables from 1980–2021. Vision Transformer, U-Net, and 3/6/9/12-hour relay models support global AOD, aerosol-component, and air-quality forecasts.
27
+
28
+ # Use Cases
29
+
30
+ | Use Case | Description |
31
+ | :---: | :--- |
32
+ | Global aerosols | Forecast AOD, optical components, and surface concentrations. |
33
+ | Dust and smoke | Track regional pollution transport. |
34
+ | Coupled weather | Jointly forecast aerosols and meteorology. |
35
+ | ModelScope/OneCode execution | Validate data, training, inference, aerosol metrics, and visualization. |
36
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
37
+
38
+ # Usage Instructions
39
+
40
+ Use a GPU or DCU when available; CPU supports the default smoke configuration.
41
+
42
+ ```bash
43
+ hf download OneScience-Group/AI-GAMFS --local-dir ./AI-GAMFS
44
+ cd AI-GAMFS
45
+ ```
46
+
47
+ ### Environment Dependencies
48
+
49
+ **Hardware Requirements**
50
+
51
+ - A GPU or DCU is recommended.
52
+ - A CPU can be used for connectivity validation with the default small-sample configuration.
53
+ - DCU users should install DTK 25.04.2 or a compatible OneScience-recommended version first.
54
+
55
+ **DCU Environment**
56
+
57
+ ```bash
58
+ # Activate DTK and Conda first
59
+ conda create -n onescience311 python=3.11 -y
60
+ conda activate onescience311
61
+ pip install onescience[earth-dcu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
62
+ ```
63
+
64
+ **GPU Environment**
65
+
66
+ ```bash
67
+ # Activate Conda first
68
+ conda create -n onescience311 python=3.11 -y libstdcxx-ng=12 libgcc-ng=12 gcc_linux-64=12 gxx_linux-64=12
69
+ conda activate onescience311
70
+ pip install onescience[earth-gpu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
71
+ ```
72
+
73
+ ```bash
74
+ python scripts/fake_data.py
75
+ python scripts/train.py
76
+ torchrun --standalone --nproc_per_node=2 scripts/train.py
77
+ python scripts/inference.py
78
+ python scripts/result.py
79
+ ```
80
+ Training optimizes four relay models. Inference produces 40 three-hourly forecasts and evaluation reports finite AOD RMSE and correlation.
81
+ ## Trained Weights
82
+ No weights are bundled under `weight/`. The paper does not provide a directly loadable official pretrained-weight link.
83
+ # Citation and License
84
+ This repository is an independent engineering reproduction of the public AI-GAMFS specifications.
85
+
86
+ The original paper is licensed under CC BY-NC-ND 4.0; official code, model weights, and related data retain their respective terms.