自前のハードウェアでAI天気予報を実行:ECMWFのAIFS 2.0を起動
European Centre for Medium-Range Weather Forecasts
Hugging Face
Anemoi
PyTorch
ECMWFはデータ駆動型天気予報モデルであるAIFSをオープンソース化しましたが、実行には通常、NVIDIAのAmpereアーキテクチャ以降のGPUが必要です。互換性パッチにより、AIFS Single 2.0が任意のGPUまたはCPUで、ローカルおよびHugging Face Jobs経由で実行可能になりました。このガイドでは、オープンウェイトを研究、アプリケーション、教育に活用する方法を説明します。
ECMWFのAIFSは、データ駆動型のアプローチを採用したAIベースの全球中期天気予報モデルであり、従来の物理ベースのモデルと比較して約1000分の1のエネルギー消費で動作します。その重みはHugging Face上で公開されており、2025年2月25日から物理ベースのIFSと並行して運用されています。モデルファミリーには、AIFS Single(決定論的、15日予報)とAIFS ENS(アンサンブル、51メンバー)が含まれます。2026年5月のv2アップデートでは、波浪と降雪の予報が追加され、成層圏の表現が改善されました。高価なGPUなしでAIFSを利用できるようにするため、互換性パッチがflash-attnをPyTorchのSDPAに置き換え、任意のGPU(CUDAまたはApple MPS経由)またはCPUでの実行を可能にします。このガイドでは、Hugging Face Jobs(例:t4-smallフレーバーを使用)またはローカルでAIFS Single 2.0を実行する方法を、セットアップと実行の手順とともに示しています。ECMWFは、AIFSが主要な物理ベースのモデルと比較して熱帯低気圧の進路予測を最大20%改善し、洪水警報システムにデータを供給していることを報告しています。
出典: Habr — хаб ИИ —
原文
