MetMamba: Regional Weather Forecasting with Spatial-Temporal Mamba Model

Haoyu Qin, Yungang Chen, Qianchuan Jiang, Pengchao Sun, Xiancai Ye, Chao Lin

arXiv preprint · 12 August 2024

In one sentence. MetMamba uses a Mamba state-space backbone for regional forecasting, outperforming attention and neural operator backbones and enabling coupled limited-area training with a global host.

Abstract

Deep Learning based Weather Prediction (DLWP) models have been improving rapidly over the last few years, surpassing state of the art numerical weather forecasts by significant margins. While much of the optimization effort is focused on training curriculum to extend forecast range in the global context, two aspects remains less explored: limited area modeling and better backbones for weather forecasting. We show in this paper that MetMamba, a DLWP model built on a state-of-the-art state-space model, Mamba, offers notable performance gains and unique advantages over other popular backbones using traditional attention mechanisms and neural operators. We also demonstrate the feasibility of deep learning based limited area modeling via coupled training with a global host model.

Labels

Labels and summary are generated by Claude and reviewed weekly. Wrong? Edit this entry.