2605.14606
2026-05-15
cs.CV
MambaRain: Multi-Scale Mamba-Attention Framework for 0-3 Hour Precipitation Nowcasting
Chunlei Shi, Cui Wu, Xiang Xu, Hao Li, Ni Fan, Xue Han, Yongchao Feng, Yufeng Zhu, Boyu Liu, Zengliang Zang, Hongbin Wang, Yanlan Yang, Dan Niu
发表机构
*
School of Automation, Southeast University(东南大学自动化学院)
;
Nanjing XinDa Institute of Meteorological Science and Technology(南京新达气象科学与技术研究所)
;
Beijing Leninainfo Technology Co., Ltd.(北京 Leninainfo 技术有限公司)
;
China CEC Engineering Corporation(中国 CEC 工程公司)
;
School of Mathematical Sciences, Tongji University(同济大学数学科学学院)
;
State Key Laboratory of Virtual Reality Technology and Systems, Beihang University(北京航空航天大学虚拟现实技术与系统国家重点实验室)
;
College of Meteorology and Oceanography, National University of Defense Technology(国防科技大学气象学与海洋学学院)
;
Key Laboratory of Transportation Meteorology of China Meteorological Administration, Nanjing Innovation Institute for Atmospheric Sciences(中国气象局交通运输气象重点实验室,南京大气科学创新研究院)
AI总结
本文提出了一种名为MambaRain的多尺度编码-解码框架,用于0-3小时的降水临近预报。该方法结合了Mamba模型的线性复杂度长期时间建模能力和自注意力机制对空间相关性的显式捕捉,有效解决了现有方法在长时段预测中性能下降的问题。通过引入混合架构和频谱损失函数,MambaRain在保持计算效率的同时提升了预报精度,尤其在2-3小时的困难预测区间表现突出。