mosaic

Mosaic是一个面向气象预报的机器学习模型，采用基于HEALPix网格的块稀疏注意力机制，在原生分辨率下建模大气动力过程。其核心特点是通过共享键值的块稀疏注意力实现线性计算复杂度，避免传统压缩编码导致的高频混叠，并引入学习型函数扰动生成具备真实谱特性的概率集合预报。模型在1.5°分辨率下以214M参数量达到甚至超越6倍更细分辨率模型的谱保真度与预报技巧，单卡H100可在12秒内完成24成员×10天预报。适用于对谱特性敏感的中短期数值天气预报、气候模式后处理及地球系统AI研究场景。