| { |
| "model_name": "Spherical Fourier Neural Operators", |
| "model_type": "sfno", |
| "architectures": [ |
| "SFNO" |
| ], |
| "framework": "PyTorch", |
| "domain": "climate-and-atmosphere", |
| "task": "global-weather-forecasting", |
| "implementation": { |
| "entry_point": "model/sfno.py", |
| "scope": "configurable wrapper around torch_harmonics.examples.models.sfno.SphericalFourierNeuralOperator for deterministic single-state 6-hour forecasting" |
| }, |
| "architecture": { |
| "family": "Spherical Fourier Neural Operator", |
| "spectral_operator": "spherical harmonic transform", |
| "input_format": "B C H W", |
| "prediction": "one atmospheric state to the next 6-hour state; longer forecasts use autoregressive rollout", |
| "activation": "GELU", |
| "normalization": "InstanceNorm", |
| "repository_default_config": { |
| "purpose": "connectivity validation with synthetic ERA5 data", |
| "img_size": [ |
| 32, |
| 64 |
| ], |
| "scale_factor": 2, |
| "in_channels": 6, |
| "out_channels": 6, |
| "embed_dim": 16, |
| "num_layers": 2, |
| "use_mlp": true, |
| "mlp_ratio": 2.0, |
| "drop_rate": 0.0, |
| "drop_path_rate": 0.0, |
| "hard_thresholding_fraction": 1.0, |
| "residual_prediction": false, |
| "positional_embedding": "none", |
| "bias": false |
| }, |
| "paper_reference_config": { |
| "grid_resolution_degrees": 0.25, |
| "grid_size": [ |
| 721, |
| 1440 |
| ], |
| "channels": "26 or 73", |
| "embed_dim": 256, |
| "num_layers": "approximately 8", |
| "note": "the weather model's exact internal spectral downsampling factor is not disclosed" |
| } |
| }, |
| "data": { |
| "dataset": "ERA5", |
| "variables": [ |
| "10m_u_component_of_wind", |
| "10m_v_component_of_wind", |
| "2m_temperature", |
| "mean_sea_level_pressure", |
| "geopotential_500", |
| "temperature_850" |
| ], |
| "time_step_hours": 6, |
| "input_length": 1, |
| "output_length": 1, |
| "channels": 6, |
| "spatial_size": [ |
| 32, |
| 64 |
| ], |
| "storage_format": "HDF5 fields with T C H W layout", |
| "train_years": [ |
| 1951, |
| 1952 |
| ], |
| "validation_years": [ |
| 1953 |
| ], |
| "test_years": [ |
| 1954 |
| ] |
| }, |
| "configuration_sources": [ |
| "README.md", |
| "conf/config.yaml", |
| "model/sfno.py", |
| "scripts/train.py", |
| "scripts/inference.py", |
| "scripts/fake_data.py", |
| "configuration.json" |
| ] |
| } |
|
|
