摘要
arXiv:2505.04802v1 地方类型: cross
摘要:稀疏观测和粗分辨率气候模型限制了有效的区域决策,强调了稳健降尺度的必要性。然而,现有的AI方法在变量和地理区域的一般化方面存在困难,并且受到Vision Transformer (ViT) 自注意力二次复杂性的制约。我们引入了ORBIT-2,这是一种用于全球高分辨率气候降尺度的可扩展基础模型。ORBIT-2包含两项关键技术创新:(1) 简轻残差ViT(Reslim),一种带有残差学习和贝叶斯正则化的轻量级架构,用于高效、稳健的预测;和(2) 块级序列尺度算法(TILES),该算法将自注意力复杂性从二次降低到线性,从而实现长序列处理和大量并行。ORBIT-2可在32,768个GPU上扩展至100亿参数,实现最高1.8 ExaFLOPS持续吞吐量和92-98%的强扩展效率。它支持将分辨率细化至0.9公里的全球分辨率,并且可以处理多达42亿个令牌的序列。在7公里分辨率的基准测试中,ORBIT-2在观测数据的R²得分范围内达到0.98到0.99的高准确性。