Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Generalizing Weather Forecast to Fine-grained Temporal Scales via Physics-AI Hybrid Modeling

About

Data-driven artificial intelligence (AI) models have made significant advancements in weather forecasting, particularly in medium-range and nowcasting. However, most data-driven weather forecasting models are black-box systems that focus on learning data mapping rather than fine-grained physical evolution in the time dimension. Consequently, the limitations in the temporal scale of datasets prevent these models from forecasting at finer time scales. This paper proposes a physics-AI hybrid model (i.e., WeatherGFT) which generalizes weather forecasts to finer-grained temporal scales beyond training dataset. Specifically, we employ a carefully designed PDE kernel to simulate physical evolution on a small time scale (e.g., 300 seconds) and use a parallel neural networks with a learnable router for bias correction. Furthermore, we introduce a lead time-aware training framework to promote the generalization of the model at different lead times. The weight analysis of physics-AI modules indicates that physics conducts major evolution while AI performs corrections adaptively. Extensive experiments show that WeatherGFT trained on an hourly dataset, effectively generalizes forecasts across multiple time scales, including 30-minute, which is even smaller than the dataset's temporal resolution.

Wanghan Xu, Fenghua Ling, Wenlong Zhang, Tao Han, Hao Chen, Wanli Ouyang, Lei Bai• 2024

Related benchmarks

TaskDatasetResultRank
Weather forecastingWeatherBench South Pacific 1h horizon
z150 RMSE123
8
Weather forecastingWeatherBench South Pacific 24-hour horizon (test)
z150 (m s-2)65
8
Weather forecastingWeatherBench South Pacific 60-hour horizon (test)
Error Z (150hPa)26
8
Weather forecastingWeatherBench South Pacific 12h horizon (test)
RMSE (z=150, m/s^2)97
8
Weather forecastingWeatherBench South Pacific 3h horizon
z150 RMSE (m s^-2)175
8
Weather forecastingWeatherBench South Pacific 6h horizon (test)
z150 RMSE (m/s^2)141
8
Showing 6 of 6 rows

Other info

Follow for update