在 NVIDIA GPU 上運(yùn)行的一個(gè)擴(kuò)散模型向天氣預(yù)報(bào)工作者展示了加速計(jì)算如何實(shí)現(xiàn)新的用途并提升能效。

當(dāng)臺灣地區(qū)的天氣預(yù)報(bào)工作人員近距離看到在一臺計(jì)算機(jī)上模擬的臺風(fēng)時(shí)，都感嘆于這能夠如此大幅節(jié)省預(yù)報(bào)任務(wù)的時(shí)間和能源消耗。

許多其他領(lǐng)域的用戶在親眼見證生成式 AI 在降低總體擁有成本方面的驚人表現(xiàn)時(shí)，都會(huì)有同樣的感受。

揭秘風(fēng)暴預(yù)測中的 AI

追蹤臺風(fēng)是證明生成式 AI 強(qiáng)大能力的絕佳案例。過去，這項(xiàng)工作首先需要利用多個(gè) CPU 集群執(zhí)行復(fù)雜的算法來創(chuàng)建分辨率為 25 公里的大氣模型。

現(xiàn)在有了 CorrDiff，情況大不相同。CorrDiff 是 NVIDIA Earth-2 上的一個(gè)生成式 AI 模型，包含一整套適用于天氣與氣候研究的服務(wù)和軟件。

通過使用一類為當(dāng)今文本生成圖像服務(wù)提供支持的擴(kuò)散模型，CorrDiff 可以將模型分辨率從 25 公里提高到 2 公里，每次推理比傳統(tǒng)的方法計(jì)算速度快 1000 倍，能源效率提高 3000 倍。

CorrDiff 使成本降至 1/50 ，

能耗降至 1/25

CorrDiff 是 NVIDIA AI 平臺上一個(gè)亮眼的存在。即使一年需要重新訓(xùn)練一次該模型，并且使用 1000 次預(yù)測的統(tǒng)計(jì)組來提高預(yù)測的準(zhǔn)確性，它也同樣表現(xiàn)出色。與在同樣條件下使用傳統(tǒng)方法相比，使用 CorrDiff 每年成本可降低至其 1/50，能耗降至其 1/25。

這意味著，過去在一組 CPU 集群上運(yùn)行、能源消耗需花費(fèi)近 300 萬美元才能完成的工作，現(xiàn)在使用一個(gè)包含單個(gè) NVIDIA Hopper GPU 的系統(tǒng)僅需大約 6 萬美元即可完成。這一成本的大幅降低表明生成式 AI 和加速計(jì)算能夠有效地提高能效和降低總體擁有成本。

該技術(shù)還能幫助天氣預(yù)報(bào)工作者更精確地預(yù)測臺風(fēng)登陸地點(diǎn)，從而挽救更多生命。

臺灣災(zāi)害防救科技中心主任陳宏宇表示：“NVIDIA 的 CorrDiff 生成式 AI 模型讓利用 AI 生成分辨率達(dá)公里級的天氣預(yù)報(bào)成為可能，可以幫助臺灣更好地為應(yīng)對臺風(fēng)做好準(zhǔn)備?！?/p>

通過使用 CorrDiff，臺灣地區(qū)的天氣預(yù)報(bào)工作者每年預(yù)計(jì)可節(jié)省近 1 吉瓦時(shí)的電量。如果全球近 200 個(gè)地區(qū)級氣象數(shù)據(jù)中心都采用這項(xiàng)技術(shù)，以實(shí)現(xiàn)更加可持續(xù)的計(jì)算，那么節(jié)省的能源將非?？捎^。

提供商業(yè)性天氣預(yù)報(bào)服務(wù)的企業(yè)也可以采用 CorrDiff 來加快速度并節(jié)省成本。

廣闊的節(jié)能前景

NVIDIA Earth-2 可使上述能力惠及全球各地，其融合了 AI、物理模擬和數(shù)據(jù)觀測，能夠幫助政府和企業(yè)應(yīng)對氣候變化等全球性問題，這將有助于應(yīng)對氣候變化的影響。預(yù)計(jì)到 2050 年，極端天氣災(zāi)害每年會(huì)導(dǎo)致百萬人喪生，并造成 1.7 萬億美元的損失。

加速計(jì)算和生成式 AI 能夠?yàn)樵S多應(yīng)用帶來更高水平的性能和能效提升。