LLM2D
ORBIT-2:面向天气和气候降尺度的_exacale视觉基础模型的扩展-scaling
ORBIT-2: Scaling Exascale Vision Foundation Models for Weather and Climate Downscaling
作者: Xiao Wang, Jong-Youl Choi, Takuya Kurihaya, Isaac Lyngaas, Hong-Jun Yoon, Ming Fan, Nasik Muhammad Nafi, Aristeidis Tsaris, Ashwin M. Aji, Maliha Hossain, Mohamed Wahib, Dali Wang, Peter Thornton, Prasanna Balaprakash, Moetasim Ashfaq, Dan Lu
发布日期: 5/9/2025
arXiv ID: oai:arXiv.org:2505.04802v1

摘要

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的高准确性。