English

STCast: Adaptive Boundary Alignment for Global and Regional Weather Forecasting

Machine Learning 2026-04-21 v4 Artificial Intelligence Atmospheric and Oceanic Physics

Abstract

To gain finer regional forecasts, many works have explored the regional integration from the global atmosphere, e.g., by solving boundary equations in physics-based methods or cropping regions from global forecasts in data-driven methods. However, the effectiveness of these methods is often constrained by static and imprecise regional boundaries, resulting in poor generalization ability. To address this issue, we propose Spatial-Temporal Weather Forecasting (STCast), a novel AI-driven framework for adaptive regional boundary optimization and dynamic monthly forecast allocation. Specifically, our approach employs a Spatial-Aligned Attention (SAA) mechanism, which aligns global and regional spatial distributions to initialize boundaries and adaptively refines them based on attention-derived alignment patterns. Furthermore, we design a Temporal Mixture-of-Experts (TMoE) module, where atmospheric variables from distinct months are dynamically routed to specialized experts using a discrete Gaussian distribution, enhancing the model's ability to capture temporal patterns. Beyond global and regional forecasting, we evaluate our STCast on extreme event prediction and ensemble forecasting. Experimental results demonstrate consistent superiority over state-of-the-art methods across all four tasks. Code: https://github.com/chenhao-zju/STCast

Keywords

Cite

@article{arxiv.2509.25210,
  title  = {STCast: Adaptive Boundary Alignment for Global and Regional Weather Forecasting},
  author = {Hao Chen and Tao Han and Jie Zhang and Song Guo and Lei Bai},
  journal= {arXiv preprint arXiv:2509.25210},
  year   = {2026}
}

Comments

This paper has already been accepted by CVPR 2026 (Highlight)