AI天气预报:一张显卡就能跑,飓风预警提前一天
WeatherNext 的强项在气旋预测:平均比现有模型多给一天提前量。三天内的预报,准确度大约相当于过去模型两天的水平。
传统天气预报靠超算跑完整数值模拟,WeatherNext 用一张 H100 GPU 就能跑。这块卡大约等于 20 年前的超级计算机。训练它依然要不少算力和数据,可跑一次预测,比完整数值模拟便宜得多、快得多。
它不是生成式 AI,要回答的问题也相对固定。只要有人问过“佛罗里达这个月会有飓风吗”,所有人都能共享这个答案。
代码开源了,但门槛还在。有人问,能不能出个量化版、先在消费级显卡上跑。还有人更直白:希望 Google 先做一个正常工作的普通天气模型。
微软也在这条路上。Aurora 也能在本地运行,底层是注意力机制加 UNET。它的精度不如 WeatherNext,但门槛更低:有人用 Xeon v4 CPU 跑了起来。每处理一帧(代表 6 小时模拟时间)要 180 秒。
天气预报正在从超算中心挪到单张显卡上。Google 和微软都进场了,竞争才刚刚开始。