生成AIで異常気象の確率を正確に予測、NVIDIAが拡散モデルを用いた極端現象の尤度推定手法を発表

要点

• NVIDIAなどの研究チームが、拡散モデルを用いて極端気象などの稀なイベントの発生確率を効率的かつ正確に推定する新手法を発表した。
• 誘導型拡散モデルによる過剰サンプリングの偏りを、誘導時と非誘導時の確率から計算する「オッズ比」を用いて補正する。
• 気候エミュレータ「NVIDIA cBottle」を用いた熱帯低気圧のリスク推定において、従来のモンテカルロ法に比べて標準誤差を大幅に削減することに成功した。
• この手法は、AI気候・気象ワークフロー向けのオープンソース推論プラットフォーム「NVIDIA Earth2Studio」で実装・公開されている。

リード文

NVIDIAのPeter Manshausen氏およびMike Pritchard氏らは2026年7月13日、AIを活用した気候エミュレータにおいて、発生確率が極めて低いものの社会的影響が大きい「極端現象（極端気象イベント）」の発生確率を正確かつ効率的に推定するための手法を発表した。本手法に関する研究論文「Towards accurate extreme event likelihoods from diffusion model climate emulators」が公開されており、拡散モデルを特定の気象状態に向けて誘導した際の偏りを補正するアプローチを提案している。

従来の予測手法における計算コストの課題

科学、工学、金融などの分野において、発生確率は低いが甚大な影響をもたらす「ローリスク・ハイインパクト（低確率・高影響）」なイベントの発生確率を推定することは極めて重要である。しかし、従来の「モンテカルロ・サンプリング（乱数を用いてシミュレーションを何度も繰り返し、特定の現象が起こる確率を統計的に割り出す手法）」と呼ばれる力まかせのアプローチでは、極端に稀な事象を捉えるために天文学的な回数のシミュレーションを実行する必要がある。特に、物理法則に基づく従来の複雑な気候モデルは1回の実行コストが大きいため、十分なサンプル数を得ることは現実的ではなかった。

これに対し、近年では特定の状態へ生成結果を導く「誘導型拡散モデル（guided diffusion models）」を利用して、稀なイベント（台風やハリケーンなど）を優先的に発生させるアプローチが注目されている。しかし、この誘導技術には大きな課題が存在していた。特定のイベントが発生しやすいようにAIモデルの生成方向を意図的に偏らせて過剰サンプリングすると、その出力結果から「元の自然な気候条件下において、そのイベントが実際にどの程度の確率で発生するのか」という真の発生確率（真の尤度）を正しく計算できなくなってしまう点である。

「オッズ比」を用いた尤度の補正メカニズム

研究チームはこの課題を解決するため、誘導された確率と非誘導（通常）の確率の比率である「オッズ比（odds ratio）」を用いた補正メカニズムを開発した。

このアプローチでは、拡散ベースの気候エミュレータ（気候条件に応じた大気状態を生成するAI）である「NVIDIA cBottle」を使用する。モデルを意図的に熱帯低気圧（TC）が発生する状態へと誘導してサンプリングを行うと同時に、生成された大気状態 $x$ について、「誘導時（guided）の確率 $p_{guided}(x)$」と「非誘導時（unguided）の確率 $p_{unguided}(x)$」の2つから、ログオッズ比 $\log o(x) = \log p_{unguided}(x) - \log p_{guided}(x)$ を算出する。

このオッズ比 $o(x) = \exp(\log o(x))$ は、重要度サンプリング（特定の注目領域のサンプルを増やしつつ、元の分布の確率を正しく推定するために重み付けを調整する手法）における「重要度重み（importance weight）」として機能する。この重みを用いて誘導されたサンプル群を再重み付けすることにより、意図的に偏らせて生成したデータから、元の気候分布における正確な発生確率を逆算することが可能になる。

このオッズ比の算出には、生成モデルを通過する「2階微分（関数の傾きの変化率を示す数学的処理）」が必要となるが、これにより誘導がもたらした尤度のシフトを正確に捉えられるという。熱帯低気圧のリスク推定におけるデモンストレーションでは、この手法により従来のモンテカルロ法と比較して標準誤差が大幅に削減され、より少ないサンプル数で高精度な確率推定が行えることを実証した。

NVIDIA Earth2Studioでの実装と利用

今回提案されたワークフローは、NVIDIAが提供するAI気候および気象ワークフロー向けのオープンソース推論プラットフォーム「NVIDIA Earth2Studio」において、すでにサンプルコード「CBottleTCGuidance」として実装が公開されている。

この熱帯低気圧誘導ワークフローの例では、ユーザーが指定した熱帯低気圧の活動度合いに基づいて、モデルが大気サンプルを生成する。具体的には、誘導テンソルが米国フロリダ近郊の特定の領域を指定し、ハリケーンシーズン中の大気サンプルの生成をモデルに要求する。システムはデフォルトのcBottle熱帯低気圧誘導（TC guidance）診断モデルをロードし、上記の重要度サンプリングを実行する仕組みとなっている。

今後の展望と広範な応用

研究チームは、この手法が気候科学にとどまらず、稀なイベントの発生確率がクリティカルな影響を及ぼす幅広いドメインへの応用が可能であると言及している。

今後の研究開発においては、計算効率のさらなる最適化や、密度推定（確率分布の形状を推定するプロセス）の安定性向上が目指される。また、誘導サンプリングの手法をより多様な極端現象や、特定の気象イベントが温暖化などの要因に起因するかを調べる「イベント帰属研究（attribution studies）」へ拡張していくことを予定している。