MambaDS: Near-Surface Meteorological Field Downscaling with Topography Constrained Selective State Space Modeling

Zili Liu,Hao Chen,Lei Bai,Wenyuan Li,Wanli Ouyang,Zhengxia Zou,Zhenwei Shi

2024-08-20

Abstract:In an era of frequent extreme weather and global warming, obtaining precise, fine-grained near-surface weather forecasts is increasingly essential for human activities. Downscaling (DS), a crucial task in meteorological forecasting, enables the reconstruction of high-resolution meteorological states for target regions from global-scale forecast results. Previous downscaling methods, inspired by CNN and Transformer-based super-resolution models, lacked tailored designs for meteorology and encountered structural limitations. Notably, they failed to efficiently integrate topography, a crucial prior in the downscaling process. In this paper, we address these limitations by pioneering the selective state space model into the meteorological field downscaling and propose a novel model called MambaDS. This model enhances the utilization of multivariable correlations and topography information, unique challenges in the downscaling process while retaining the advantages of Mamba in long-range dependency modeling and linear computational complexity. Through extensive experiments in both China mainland and the continental United States (CONUS), we validated that our proposed MambaDS achieves state-of-the-art results in three different types of meteorological field downscaling settings. We will release the code subsequently.

Atmospheric and Oceanic Physics,Artificial Intelligence,Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The paper aims to address the issue of high-resolution near-surface meteorological forecasting, especially in the context of global warming and the frequent occurrence of extreme weather events. Accurate and fine-grained near-surface meteorological forecasts are becoming increasingly important for human activities. To this end, the paper proposes a new model—MambaDS, which is used to reconstruct high-resolution meteorological states of the target area from global-scale forecast results (i.e., downscaling). Existing downscaling methods are often structurally constrained and fail to effectively integrate terrain information, while MambaDS addresses these issues by introducing a selective state-space model.

Specifically, MambaDS improves existing methods through the following points:
1. **Multivariable Correlation Enhancement**: Introduces a channel attention mechanism to establish relationships between different meteorological variables.
2. **Terrain Constraint Layer**: Designs an efficient terrain constraint layer at the end of the model to ensure physical consistency while avoiding additional computational overhead.
3. **Long-Range Dependency Modeling**: Utilizes the advantages of the Mamba model to enhance the modeling capability of long-range dependencies while maintaining linear computational complexity.

Through extensive experimental validation conducted in mainland China and the contiguous United States (CONUS), MambaDS achieved state-of-the-art results in three different meteorological field downscaling settings.