Back to Search Start Over

Utilization of a Lightweight 3D U-Net Model for Reducing Execution Time of Numerical Weather Prediction Models

Authors :
Hyesung Park
Sungwook Chung
Source :
Atmosphere, Vol 16, Iss 1, p 60 (2025)
Publication Year :
2025
Publisher :
MDPI AG, 2025.

Abstract

Conventional weather forecasting relies on numerical weather prediction (NWP), which solves atmospheric equations using numerical methods. The Korea Meteorological Administration (KMA) adopted the Met Office Global Seasonal Forecasting System version 6 (GloSea6) NWP model from the UK and runs it on a supercomputer. However, due to high task demands, the limited resources of the supercomputer have caused job queue delays. To address this, the KMA developed a low-resolution version, Low GloSea6, for smaller-scale servers at universities and research institutions. Despite its ability to run on less powerful servers, Low GloSea6 still requires significant computational resources like those of high-performance computing (HPC) clusters. We integrated deep learning with Low GloSea6 to reduce execution time and improve meteorological research efficiency. Through profiling, we confirmed that deep learning models can be integrated without altering the original configuration of Low GloSea6 or complicating physical interpretation. The profiling identified “tri_sor.F90” as the main CPU time hotspot. By combining the biconjugate gradient stabilized (BiCGStab) method, used for solving the Helmholtz problem, with a deep learning model, we reduced unnecessary hotspot calls, shortening execution time. We also propose a convolutional block attention module-based Half-UNet (CH-UNet), a lightweight 3D-based U-Net architecture, for faster deep-learning computations. In experiments, CH-UNet showed 10.24% lower RMSE than Half-UNet, which has fewer FLOPs. Integrating CH-UNet into Low GloSea6 reduced execution time by up to 71 s per timestep, averaging a 2.6% reduction compared to the original Low GloSea6, and 6.8% compared to using Half-UNet. This demonstrates that CH-UNet, with balanced FLOPs and high predictive accuracy, offers more significant execution time reductions than models with fewer FLOPs.

Details

Language :
English
ISSN :
20734433
Volume :
16
Issue :
1
Database :
Directory of Open Access Journals
Journal :
Atmosphere
Publication Type :
Academic Journal
Accession number :
edsdoj.91892be87bb648789b60026f9907c31f
Document Type :
article
Full Text :
https://doi.org/10.3390/atmos16010060