Aurora: Microsoft 向地球大气基础 AI 模型迈出的一大步

随着全球变暖的加剧，世界各地的社区都在努力应对其毁灭性的影响。无情的温室气体排放增加正在加剧极端天气事件、毁灭性的自然灾害和气候相关疾病的发生。天气预报系统是我们抵御这些挑战的第一道防线，但传统的方法依赖于人类专家，难以跟上。2023 年风暴 Ciarán 的破坏使我们清楚地认识到需要更好的预测模型。但是，地球的大气层具有复杂的天气模式、化学相互作用和各种影响，仍然是一个预测挑战。Microsoft最近在解决这些挑战方面取得了突破。他们开发了一个地球大气的AI模型，称为Aurora，可用于天气预报、空气污染监测等。这篇文章深入探讨了这一发展，探索Aurora、其应用和其对天气预报的影响。

介绍Aurora

Aurora是一种开创性的AI模型，旨在解决从极端天气事件到空气污染和短期、中期天气预报等各种预测挑战。Aurora的特点是其在超过一百万小时的多样化天气和气候模拟上进行了训练，提供了对不断变化的大气过程的深刻理解。这使得Aurora在预测任务中表现出色，甚至在数据有限或极端天气场景下也是如此。
使用称为视觉变换器的人工神经网络模型构建，Aurora被训练为理解驱动大气变化的复杂关系。作为一种通用模型，Aurora可以处理多种类型的输入和输出。它采用基于感知器架构的编码器-解码器模型，专门设计用于管理时间变化的输入和输出。
Aurora的训练过程包括两个步骤：预训练和微调。在预训练期间，Aurora从具有不同详细程度的多样化数据集中学习，涵盖了天气模式、空气压力等大气的各个方面。通过在这些不同类型的数据中最小化错误来提高其技能。在初始训练之后，Aurora经过两个阶段的微调。第一阶段专注于提高Aurora的短期预报能力。在第二阶段，它进一步完善了其长期预测的技能，使用了一种称为低秩适应（LoRA）的方法。

Aurora的关键特征

• 广泛的训练：Aurora的有效性在于其在超过一百万小时的多样化天气和气候模拟上进行的训练，这些模拟来自六个天气和气候模型。这种全面训练使Aurora更好地理解大气动力学。
• 性能和效率：以0.1°（约11公里，赤道附近）的高空间分辨率运行，Aurora捕捉大气过程的细节，使其比传统的数值天气预报系统更准确，并且计算成本仅为传统系统的一小部分。
• 快速速度：Aurora可以在不到一分钟的时间内生成5天的全球空气污染预报和10天的天气预报，超越了传统的模拟工具和最好的专用深度学习模型。
• 多模态能力：Aurora是一种多模态AI模型，能够处理和分析各种类型的数据，包括数值天气数据、卫星图像和气候模拟。
• 多样化预测：该模型可以预测广泛的大气变量，从温度和风速到空气污染水平和温室气体浓度。

Aurora的潜在应用

• 极端天气预报：Aurora在预测极端天气事件（如飓风、风暴和热浪）方面表现出色。其高分辨率功能使其能够精确跟踪和预测这些事件，为灾难预防和应对工作提供了至关重要的预警时间。
• 空气污染监测：Aurora可以生成准确的5天全球空气污染预报，有效跟踪污染物如二氧化氮。该功能在人口密集的地区尤其有价值，因为空气质量直接影响公共卫生。
• 气候变化分析：Aurora能够处理多样化的气候数据，使其成为研究长期气候趋势和评估气候变化对各种大气现象影响的宝贵工具。
• 农业规划：通过提供详细的天气预报，Aurora支持农业规划和决策。农民可以优化种植时间、灌溉和收获时间，降低因意外天气变化导致的作物歉收的风险。
• 能源部门优化：Aurora的准确天气预报有助于优化能源生产和分配。太阳能和风能等可再生能源可以从精确的预报中受益，确保高效的能源收获和电网管理。
• 环境保护：Aurora的详细预报和空气污染跟踪有助于环境保护工作。政策制定者和环境机构可以使用其数据来实施和监测旨在减少污染和减轻气候变化影响的法规。

Aurora与GraphCast

Aurora和GraphCast是两种著名的天气预报模型，每种都有其自身的优势和能力。GraphCast由Google DeepMind开发，以其在天气预报方面的技能而闻名。它是在ERA5数据集上预训练的，并以0.25度的分辨率运行，使其能够预测最多五天的天气。
另一方面，Aurora拥有更为多样化的训练数据集，并以0.1度的更高分辨率运行。这使得Aurora能够以更高的精度生成10天的全球天气预报。在一项比较研究中，Microsoft研究人员发现Aurora在91%的目标上都优于GraphCast，当两种模型都以0.25度的分辨率进行评估时。
两种模型之间的一个关键区别是其多功能性。虽然GraphCast仅专注于天气预报，但Aurora是一种基础模型，能够执行各种任务，包括天气预报、空气污染监测和气候变化分析。这种多功能性使Aurora成为一个更全面的工具，用于理解和预测大气现象。

结论

Aurora不仅仅是一种天气预报模型，它是对整个地球系统（而不仅仅是大气层）的建模迈出了开创性的步伐。其在有限数据下表现良好的能力可能会使准确的天气和气候信息的获取民主化，特别是惠及数据有限的地区，如发展中国家。Aurora为各个领域（包括农业、交通、能源和灾害预防）带来了巨大的希望，能够为决策者和全球公众提供及时、准确和实用的见解。随着技术和数据可用性的不断进步，Aurora有可能在天气和气候预测中发挥关键作用，提供及时、准确和实用的见解。