Add pipeline tag, library name, paper link and Github repo link
Browse filesThis PR improves the model card by:
- Setting the pipeline tag to `time-series-forecasting`
- Setting the library_name to `torch` since the model uses PyTorch code
- Adding links to the paper and github
README.md
CHANGED
|
@@ -1,3 +1,77 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
pipeline_tag: time-series-forecasting
|
| 4 |
+
library_name: torch
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
# OneForecast: A Universal Framework for Global and Regional Weather Forecasting
|
| 8 |
+
|
| 9 |
+
This repository contains the code for the model described in the paper [OneForecast: A Universal Framework for Global and Regional Weather Forecasting](https://huggingface.co/papers/2502.00338).
|
| 10 |
+
|
| 11 |
+
Github repository: https://github.com/YuanGao-YG/OneForecast.
|
| 12 |
+
|
| 13 |
+
Accurate weather forecasts are important for disaster prevention, agricultural planning, etc. Traditional numerical weather prediction (NWP) methods offer physically interpretable high-accuracy predictions but are computationally expensive and fail to fully leverage rapidly growing historical data. In recent years, deep learning models have made significant progress in weather forecasting, but challenges remain, such as balancing global and regional high-resolution forecasts, excessive smoothing in extreme event predictions, and insufficient dynamic system modeling. To address these issues, this paper proposes a global-regional nested weather forecasting framework (OneForecast) based on graph neural networks. By combining a dynamic system perspective with multi-grid theory, we construct a multi-scale graph structure and densify the target region to capture local high-frequency features. We introduce an adaptive messaging mechanism, using dynamic gating units to deeply integrate node and edge features for more accurate extreme event forecasting. For high-resolution regional forecasts, we propose a neural nested grid method to mitigate boundary information loss. Experimental results show that OneForecast performs excellently across global to regional scales and short-term to long-term forecasts, especially in extreme event predictions.
|
| 14 |
+
|
| 15 |
+
# Quick Start
|
| 16 |
+
|
| 17 |
+
### Installation
|
| 18 |
+
|
| 19 |
+
- cuda 11.8
|
| 20 |
+
|
| 21 |
+
```
|
| 22 |
+
# git clone this repository
|
| 23 |
+
git clone https://github.com/YuanGao-YG/OneForecast.git
|
| 24 |
+
cd OneForecast
|
| 25 |
+
|
| 26 |
+
# create new anaconda env
|
| 27 |
+
conda env create -f environment.yml
|
| 28 |
+
conda activate oneforecast
|
| 29 |
+
```
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
### Inference
|
| 33 |
+
|
| 34 |
+
1. Global Forecasts Inference
|
| 35 |
+
|
| 36 |
+
(1) Preparing the test data as follows:
|
| 37 |
+
|
| 38 |
+
```
|
| 39 |
+
./data/
|
| 40 |
+
|--global
|
| 41 |
+
| |--test
|
| 42 |
+
| | |--2020.h5
|
| 43 |
+
| |--mean.npy
|
| 44 |
+
| |--std.npy
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
(2) Inference with 1-step supervised pretrained ckpt:
|
| 48 |
+
```
|
| 49 |
+
sh inference.sh
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
+
(3) Inference with finetuned pretrained ckpt:
|
| 53 |
+
```
|
| 54 |
+
sh inference_finetune.sh
|
| 55 |
+
```
|
| 56 |
+
|
| 57 |
+
2. Regional Forecasts Inference
|
| 58 |
+
|
| 59 |
+
(1) Preparing the test data as follows:
|
| 60 |
+
|
| 61 |
+
```
|
| 62 |
+
./data/
|
| 63 |
+
|--global
|
| 64 |
+
| |--test
|
| 65 |
+
| | |--2020.h5
|
| 66 |
+
| |--mean.npy
|
| 67 |
+
| |--std.npy
|
| 68 |
+
|--regional
|
| 69 |
+
| |--test
|
| 70 |
+
| | |--2020.h5
|
| 71 |
+
```
|
| 72 |
+
|
| 73 |
+
(2) Inference with 1-step supervised pretrained ckpt:
|
| 74 |
+
|
| 75 |
+
```
|
| 76 |
+
sh inference_nng.sh
|
| 77 |
+
```
|