Earth-2 - NVIDIAのオープンソースAI気象予報モデル群

地球2とは何ですか？

NVIDIA Earth-2は、NVIDIAが発表した世界初の完全オープンソースAI気象予報モデルファミリーです。Atlas、StormScope、HealDAという3つのコアアーキテクチャに基づき、それぞれ15日間のグローバル中期予報、0～6時間先のキロメートルレベルのナウキャスト、そして第2レベルの初期条件生成を実現します。物理方程式を解くためにスーパーコンピューティングに依存する従来の数値気象予報とは異なり、Earth-2はエンドツーエンドのAIパイプラインを採用することで、処理速度を1000倍向上させ、エネルギー消費量を大幅に削減します。これにより、世界中の気象機関、企業、開発者は、オープンソースモデルに基づいて独立かつ制御可能な高精度予報システムを構築できるようになり、気象AIが産業展開の時代に本格的に突入したことを示しています。

地球2の主な機能

• 地球2号機の中期天気予報Atlasアーキテクチャをベースとしたこのシステムは、最大15日間のグローバルな天気予報を提供し、気温、気圧、風速、湿度など70種類以上の気象変数を予測できます。主要な業界ベンチマークテストでは、Google DeepMindのGenCastなど、業界をリードするオープンモデルを上回る精度を実現しています。
• 地球2のナウキャスティングStormScopeのアーキテクチャに基づき、生成型AI技術を用いて衛星データとレーダーデータを直接処理し、0～6時間先の短期的な異常気象を予測するとともに、数分以内にキロメートル単位の解像度で局地的な暴風雨予報を生成することができる。
• Earth-2 グローバルデータ同化HealDAアーキテクチャをベースとしたこのシステムは、気象予報における初期条件生成という最も時間のかかる問題を解決します。従来のスーパーコンピュータでは数時間かかっていた処理を、GPU上では数秒で完了できます。中間モデルと組み合わせることで、AIによって完全に駆動される、現在利用可能な最も強力な予報パイプラインを構築できます。
• AIによるダウンスケーリング（Earth-2 CorrDiff）生成型AIによるダウンスケーリング手法は、条件付き拡散モデルを用いることで実現されており、従来の手法に比べて500倍高速かつ10,000倍のエネルギー効率を実現しています。この手法は、粗粒度の予報を街路レベルまで精緻化し、気象現象の確率分布をより正確に評価するためのデータセットを多数生成することを可能にします。
• 視覚化サービスNVIDIA OmniverseとOpenUSD 3Dフレームワークをベースとしており、地球規模の気候シミュレーションと地理空間データセットの集約と可視化をサポートし、クラウド上での高解像度インタラクティブ探索を可能にします。

地球2の技術原理

• アトラスアーキテクチャAtlasアーキテクチャは、Earth-2中期予報モデルの中核を成すものです。深層学習ネットワークを用いて地球規模の時空間系列データを処理し、従来の数値気象予報における複雑な物理方程式の解法をエンドツーエンドのAIパイプラインに置き換えています。これにより、高い精度を維持しながら予報効率を大幅に向上させ、70種類以上の気象変数について最大15日間の地球規模予報を実現できます。
• StormScopeアーキテクチャ StormScopeアーキテクチャは、Earth-2近未来予測モデルの技術的基盤です。このアーキテクチャは、生成型AI技術を用いて衛星画像やレーダー画像から直接学習・推論を行い、従来の物理方程式に頼ることなく大気運動をシミュレーションします。これにより、モデルは嵐システムの動的な変化をより迅速かつ正確に捉えることができ、嵐の力学シミュレーションにおいて従来の物理的手法を凌駕する初のAIアーキテクチャとなっています。
• HealDAアーキテクチャHealDAアーキテクチャは、Earth-2グローバルデータ同化モデルの中核となる革新的な技術です。GPUの並列計算能力を活用することでデータ同化プロセスを完全に再構築し、従来スーパーコンピュータで数時間かかっていた初期大気条件生成タスクをわずか数秒で完了させます。これにより、観測データと予測モデルを効率的に連携させることが可能になり、AIを活用したエンドツーエンドの予測パイプライン構築の基盤が築かれます。
• CorrDiffテクノロジーCorrDiff技術は、条件付き拡散モデルに基づいた生成型ダウンスケーリングを実現します。生成型AIの超解像能力を活用し、低解像度の気象予報データを高解像度の局所スケールに精緻化することで、物理的な整合性を維持しながら、処理速度を500倍、エネルギー効率を1万倍向上させ、街路レベルでの高精度な予報を可能にします。

Earth-2プロジェクトの住所

• プロジェクト公式サイト：https://www.nvidia.com/en-us/high-performance-computing/earth-2/
• GitHubリポジトリ：https://github.com/NVIDIA/earth2studio
• ハギングフェイスモデルライブラリ：https://huggingface.co/collections/nvidia/earth-2

Earth-2の応用シナリオ

• 天気予報サービス気象機関は、現地のデータを利用してオープンソースモデルであるEarth-2を微調整し、国の地理的特性に適合した、完全に独立して制御可能な国家気象予報システムを構築することができる。
• 異常気象警報FourCastNetのようなモデルを用いることで、台風、豪雨、熱波といった異常気象を3週間前に予測することができ、1,000個のアンサンブル予報メンバーを生成することで、災害予防と軽減のための貴重な対応時間を確保できます。
• クリーンエネルギー管理風力発電所と日射量の変化を正確に予測し、風力発電所と太陽光発電所の発電量予測を最適化し、再生可能エネルギーの消費能力を向上させる。
• 農業における精密播種このシステムは、農家が灌漑、施肥、播種、病害虫防除を正確に計画できるように支援し、それによって気象災害による損失を減らし、作物の収量と品質を向上させる。
• 航空および船舶の安全飛行ルートの気象状況をリアルタイムで監視することで、晴天時の乱気流や雷雨などの危険な気象状況を事前に回避し、飛行経路や離着陸時間を最適化して、定時運航率と飛行安全性を向上させることができます。