Nature重磅:改写AI气象预测,可在台式电脑上运行,速度快千倍

2025-04-04 10:30:35 科技

图片

在全球变暖、自然极端天气频发的当下,准确的气象预测对于人类生命健康与社会发展至关重要

例如,在农业方面,为播种、灌溉和收割等农事活动提供参考,减少自然灾害对农作物的影响,提高产量和经济效益;在航空领域,帮助航空公司合理规划航班,避免因恶劣天气导致的延误和事故;在公共安全领域,为应急管理部门的工作提供参考,减少人员伤亡和财产损失……

从科学研究的角度来看,气象预测能够帮助科学家更好地理解大气的物理过程和气候变化的趋势。分析气候系统的复杂性,探索极端天气事件的成因,以及评估人类活动对气候的影响。

然而,现有的人工智能(AI)气象预测领域面临诸多挑战。传统数值天气预测(NWP)系统依赖复杂流程和超级计算机,难以快速改进,且改进成本(běn)高(gāo)昂(áng)。AI 模(mó)型(xíng)虽(suī)具(jù)潜(qián)力(lì),但(dàn)多(duō)依(yī)赖(lài) NWP 初(chū)始(shǐ)状(zhuàng)态(tài),性(xìng)能(néng)受(shòu)限(xiàn),且(qiě)在(zài)数(shù)据(jù)处(chù)理(lǐ)上(shàng)面(miàn)临(lín)数(shù)据(jù)缺(quē)失(shī)和(hé)融(róng)合(hé)难(nán)题(tí),影(yǐng)响(xiǎng)其(qí)预(yù)测(cè)性(xìng)能(néng)和(hé)实(shí)际(jì)应(yīng)用(yòng)。

谷(gǔ)歌(gē) GenCast、GraphCast、NeuralGCM 等(děng) AI 气(qì)象(xiàng)预(yù)测(cè)技(jì)术(shù)取(qǔ)得了显著进展,但它们在数据依赖性、预测时效和分辨率、模型复杂性和计算成本以及实际应用的局限性等方面仍存在一些缺点。

作为新一代 AI 气象预测系统,由来自剑桥大学和艾伦图灵研究所的研究团队开发的 Aardvark Weather 有望带来气象预测的范式转变(biàn)。

Aardvark Weather 是(shì)首(shǒu)个可用在台式电脑训练和运行的单一 AI 模型取代天气预报流程所有步骤的系统,速度快数千倍,可处理来自卫星、气象站和气象气球的多模态复杂数据,生成 10 天全球预报。

相关研(yán)究(jiū)论(lùn)文以(yǐ)“End-to-end data-driven weather prediction”为(wèi)题(tí),已(yǐ)发(fā)表(biǎo)在(zài)权(quán)威(wēi)科(kē)学(xué)期(qī)刊(kān) Nature 上(shàng)。

图(tú)片(piàn)

论(lùn)文链(liàn)接(jiē):

https://www.nature.com/articles/s41586-025-08897-0

研(yán)究(jiū)团(tuán)队(duì)表(biǎo)示(shì),这(zhè)将(jiāng)为(wèi)缺(quē)乏(fá)超(chāo)级(jí)计(jì)算(suàn)机(jī)、复(fù)杂(zá)基(jī)础(chǔ)设(shè)施(shī)和(hé)专(zhuān)业(yè)知(zhī)识(shi)的(de)发(fā)展(zhǎn)中(zhōng)国(guó)家(jiā)带(dài)来(lái)真(zhēn)正(zhèng)的(de)价值,以及在任何地方发挥作用,提高效率和准确性,甚至减少天气预报的大量碳足迹。

AI气象预测:更便宜、更准确、更高效

传统的数值天气预报(NWP)需 3 步,包括:收集信息估计大气当前状态、用复杂计算模型产生预报、处理预报确保其在特定位置可用。

过去两年,AI 虽用于流程第二步,使其速度更快、更准确,但第(dì)一(yī)步(bù)尚(shàng)未(wèi)取(qǔ)得(de)更(gèng)大(dà)进(jìn)展(zhǎn),仍(réng)需(xū)消(xiāo)耗(hào)大(dà)量(liàng)资(zī)源(yuán)。NWP 依(yī)赖(lài)数(shù)据(jù)同(tóng)化(huà)、流(liú)体(tǐ)力(lì)学(xué)方(fāng)程(chéng)求(qiú)解(jiě)、后处理等多个步骤,需超级计算机支持,这使得其运行成本高昂,且难以快速迭代和改进。生成全球预报需约1000节点小时,且依赖高分辨率模型。

在这项研究中,Aardvark Weather 通过其端到端的数据驱动方法,显著降低了气象预测的计算成本。它还能够在几秒钟内生成全球天气预报,相较于传统方法需要数小时甚至数天的计算时间,极大地提高了预测效率。

此外,Aardvark Weather 在多个气象变量和预测时效上展现出与传统 NWP 系统相当甚至更优的准确性,尤其是在处理小范围极端天气事件时表现出色。这种高效、经济且准确的特性,使其成为未来气象预测领域的一个重要发展方向。

具体而言,Aardvark 的速度更快。它是首个完全数据驱动的端到端天气预报系统,比传统 NWP 更快、成本更低,其生成速度比现有系统快几个数量级,并且不依赖 NWP 预报产品。

在 4 个 NVIDIA A100 GPU 上,Aardvark 由观测数据生成完整预报大约只需要一秒钟。相比之下,高分辨率预报(HRES)仅执行数据同化和预报就需要大约 1000 个节点小时,这还未考虑下游本地模型和处理。

而且,Aardvark 是端到(dào)端(duān)的(de)。Aardvark 学(xué)习端到端模型提供了额外的功能,即能够优化系统,以在任意感兴趣的变量或区域上实现性能最大化。与传统 NWP 系统相比,机器学习系统不仅速度(dù)更(gèng)快(kuài)、计(jì)算(suàn)成(chéng)本更低,而且改进和维护起来也容易得多。

该系统的简单性,不仅使其更容易被已经运行 NWP 的用户部署和维护,还为发展中国家的一些地区提供了运行定制 NWP 的潜力,这些地区的机构通常缺乏运行(xíng)传(chuán)统(tǒng)系(xì)统(tǒng)的(de)资(zī)源(yuán)和(hé)专(zhuān)业(yè)知(zhī)识(shi)。

此(cǐ)外(wài),端(duān)到(dào)端(duān)预(yù)测(cè)还(hái)展(zhǎn)示(shì)了(le)能(néng)够(gòu)针(zhēn)对(duì)特(tè)定(dìng)区(qū)域和(hé)变(biàn)量(liàng)优(yōu)化(huà)定(dìng)制(zhì)模(mó)型(xíng)以(yǐ)最(zuì)大(dà)化(huà)预测能力的潜力,这引起了农业、可再生能源、保险和金融等多个领域终端用户的极大兴趣。

那么,Aardvark Weather 是如何做到这些的呢?

Aardvark Weather 是一个深度学习模型,能够从多种观测数据源中摄取信息,并生成全球网格化预测和本地站点预测,三个主要模块组成:编码器(encoder)、处理器(processor)和解码器(decoder)。它在训练阶段利用高质量的再分析(xī)数(shù)据(jù),而(ér)在(zài)部(bù)署(shǔ)阶(jiē)段(duàn)完(wán)全独(dú)立(lì)于(yú)传(chuán)统(tǒng) NWP 产(chǎn)品(pǐn)。

图(tú)片(piàn)

图(tú)|三(sān)个(gè)模(mó)块(kuài)各(gè)自(zì)的(de)功(gōng)能(néng)

首(shǒu)先(xiān),编(biān)码(mǎ)器(qì)模(mó)块(kuài)整(zhěng)合(hé)多(duō)种(zhǒng)来(lái)源(yuán)的(de)观(guān)测(cè)数(shù)据(jù),包(bāo)括网内和网外观测数据,并生成网格(gé)化(huà)初(chū)始(shǐ)状(zhuàng)态(tài)。网(wǎng)内(nèi)观(guān)测(cè)数(shù)据(jù)是(shì)规(guī)则(zé)网(wǎng)格(gé)上(shàng)的(de)数(shù)据(jù)模(mó)态(tài),而(ér)网(wǎng)外(wài)模(mó)态(tài)则(zé)是(shì)一(yī)组(zǔ)经(jīng)纬(wěi)度(dù)位(wèi)置(zhì)上(shàng)的(de)数据。

为此,他们利用了深度学习在处理离网数据和缺失数据方面的最新进展。这一过程采用非递归方法进行状态估计,以避免传统数据同化系统中的复杂性。

对于编码器模块,他们通过掩(yǎn)码(mǎ)通道区分缺失数据与真实观测数据,使用 SetConv 层处理非网格化数据,并利用 ViT 提取特征,最终输出 24 个大气变量的初始估计值,训练过程中采用 VLW-RMSE 损失函数,并通过 AdamW 优化器进行优化。

其次,处理器模块以编码器估算出的初始大气状态作为输入,会在 24 小时内生成网格预报。通过自回归方式生成未来 24 小时的天气预测。对于更长的预测时间,处理器模块会将其自身的预测作为输入,逐步推算未来的天气状态。

该模块由 10 个独立的 ViT 串联构成,每个 ViT 负责预测未来 1 天的气象状态残差(residual,即当前预测与真实状态的差异),并通过叠加残差逐步生成 1 至 10 天的全球网格化预报。

例如,第一天的 ViT 以初始状态为输入预测第 1 天残差,第二天的 ViT 则以第 1 天修正后的状态为输入,依此类推。

最后,解码器模块收到处理器模块输出的网格化预测数据,任务特定的解码器模块将网格化数据转换为具体的本地气象预测。

该模块采用 U-Net 为主架构,对全球预报数据进行空间特征提取,捕捉局部气象细节,随后通过 SetConv 层将网格数据插值到任意目标站点坐标,并结合多层感知机(MLP)融入地形辅助信息(如海拔、坡度)。

迈向新一代端到端气象预测系统

虽(suī)然(rán) AI 气(qì)象(xiàng)预(yù)测(cè)工(gōng)具(jù)正(zhèng)在(zài)快(kuài)速(sù)发(fā)展(zhǎn),但(dàn)这(zhè)仍(réng)然(rán)是(shì)一(yī)项(xiàng)实(shí)验(yàn)技(jì)术(shù),需(xū)要(yào)在(zài)一(yī)段(duàn)时(shí)间(jiān)内(nèi)进(jìn)行(xíng)严(yán)格(gé)的(de)评(píng)估(gū)。要(yào)想(xiǎng)在(zài)实(shí)际(jì)中(zhōng)部(bù)署(shǔ) Aardvark Weather 这(zhè)样(yàng)的(de)端(duān)到(dào)端(duān)数(shù)据(jù)驱(qū)动(dòng)模(mó)型(xíng),必(bì)须(xū)考(kǎo)虑(lǜ)当(dāng)前(qián)模(mó)型(xíng)的(de)局(jú)限(xiàn)性(xìng)。

与(yǔ)所(suǒ)有(yǒu)当前的 AI-NWP 系统一样,Aardvark Weather 尚未以 IFS 的分辨率运行。需要进一步研究以提高网格分辨率,并通过扩散等方式生成预报(bào)集合(hé)。

此(cǐ)外(wài),Aardvark Weather 在(zài)观(guān)测(cè)数(shù)据(jù)的(de)使(shǐ)用(yòng)上(shàng)也(yě)存(cún)在(zài)一(yī)些(xiē)局(jú)限(xiàn)性(xìng)。增加更多的观测方(fāng)式(shì)很可能会提高预报能力。同时,还需要考虑如何将那些尚未有训练数据的新仪器的数据有效地整合到系统中。例如,可以通过对模拟数据进行训练来实现。

对于气象预测工具来说,飓风和洪水等极端天气尤其重要。不幸的是,这类罕见事件在训练数据中较少出现,这意味着人工智能系统在这些现象上可能会遇到更多困难。

另一个需要考虑的问题是应对观测数据漂移以及数据随时间的其他变化,比如气候变化,这可能会降低基于过去数据训练的模型的准确性。这可以通过定期使用最近几个月的数据对所有模块进行微调,从而适应仪器特性的变化。

对此,研究团队表示,他们将在未来的工作中通过扩展 Aardvark Weather 以支持更多其他预报变量,无论是在其网格化预报中,还是通过其解码器模块,都可以进一步增加其功能。例如,Aardvark Weather 可以支持多种解码器模块,以提供不同类型的终端用户预报,如飓风、洪水、严重对流、火灾和其他极端天气预警。

另一个令人兴奋的方向是,利用端到端系统进行更长时间的预报,来生成季节性预报产品

此外,增加更多的观测模式将允许对地球系统的其他组成部分进行建模。例如,用于空气质量预报的大气化学和(hé)用(yòng)于(yú)海(hǎi)洋(yáng)预(yù)报(bào)的(de)海(hǎi)洋(yáng)参(cān)数(shù)。

他(tā)们(men)设(shè)想(xiǎng),Aardvark Weather 将(jiāng)成(chéng)为(wèi)新(xīn)一(yī)代(dài)端(duān)到(dào)端(duān)气(qì)象(xiàng)预(yù)测(cè)系(xì)统(tǒng)中(zhōng)第(dì)一(yī)个(gè)能(néng)够(gòu)应(yīng)对(duì)这(zhè)些(xiē)多(duō)样(yàng)化(huà)任(rèn)务(wu)的(de)系(xì)统(tǒng)

作(zuò)者(zhě):锦(jǐn)鲤(lǐ)