見出し画像

DeepMindが天気予報AIである降水のDeep Generativeモデルをリリース

(投稿日:2021/12/21)へのリンク

DeepMindDeep Generative Models of Rainfall(DGMR)のデータセットとトレーニング済みモデルスナップショットをオープンソース化した。短期の降水量予測用のAIシステムだ。58人の専門気象学者が、他の既存の方法との比較で実施した評価で、DGMRはテストケースの89%で精度と有用性で1位にランクインした。

モデルといくつかの実験Natureに掲載された記事に記載されている。DeepMindは、ナウキャストを行うために英国気象庁と共同でDGMRを開発した。ナウキャストは、降水量の短期的な高解像度の予測である。DGMRは生成モデルと呼ばれる深層学習手法を使って「レーダームービー」を生成するための学習をする。短期の一連の降雨のレーダー画像が与えられると、将来のレーダー画像の予測を学習する。そうすると、将来の降水量と場所を予測する。DeepMindは次のように述べている。

これはエキサイティングな研究分野であると考えています。私たちの論文が、競争力のある検証と運用上の実用性の両方を提供できるデータと検証方法を提供することで、新しい研究の基盤として役立つことを願っています。また、英国気象庁とのコラボレーションにより、機械学習と環境科学の統合が促進され、変化していく気候における意思決定がより適切にサポートされることを願っています。

ナウキャストは多くの分野で意思決定を支援するためによく使われている。航空交通管制やエネルギー管理などである。そのため、その精度は経済的および安全性に影響を及ぼす。STEPSPySTEPSなどの現在の方法では、気象の振る舞いを表す物理方程式を解く数理アプローチをよく使っている。これらのシステムは、予測のアンサンブルを生成することにより、予測の不確実性をモデル化している。さらに最近では、研究者はレーダー観測のデータセットでトレーニングされた深層学習モデルを開発した。ただし、DeepMindチームは、このモデルは「複数の空間的・時間的集約にわたった一貫した予測を提供することができない」ため、運用上の有用性が限られていることを述べている。
DGMRモデルは、条件付き敵対的生成ネットワーク(GAN)をベースとしている。ジェネレータネットワークは、観測された4つのレーダーフレームをコンテキストとして取り込み、次の18フレームの出力予測を生成する。ジェネレータは2つの弁別器ネットワークとともにトレーニングされる。弁別器ネットワークは、実際のレーダーデータと生成されたデータの違いを区別することを学習する。1つの弁別子はフレーム内の空間的一貫性にフォーカスしており、もう1つは一連のフレームを通した時間的一貫性にフォーカスしている。システム全体は、2016年から2019年までの、英国でのレーダー観測からの履歴データでトレーニングされている。トレーニングされたモデルは、単一のNVIDIA V100 GPUを使って「1秒強」で予測を生成できる。
DGMRのパフォーマンスを評価するために、DeepMindは3つのベースラインモデル(PySTEPS、UNetMetNet)と比較した。精度と価値の一般的なランク付けに加えて、専門の気象学者のグループも、唯一の「気象学的な挑戦イベント」に対してモデルの予測を評価した。このケーススタディでは、グループの93%がDGMRの結果を1番とした。DeepMindチームはさらに、いくつかのメトリックでモデルを評価した。例えば、クリティカルサクセスインデックス(CSI)、放射状に平均化されたパワースペクトル密度(PSD)、連続ランク確率スコア(CRPS)などである。これらの指標について、DGMRはベースラインと比較して「競争力」があった。
天気予報用のAIモデルは活発な研究分野である。InfoQは以前、暴風雨によって引き起こされた停電を予測するためのAIモデルを記事にしている。また、気候のモデリングに使用できる偏微分方程式を解くためのモデルについても記事にしている。Googleは最近MetNet-2を発表した。これはMetNetの「パフォーマンスを大幅に向上させた」ものである。

ソースはコチラ


ちなみにこのデータは公開されている。


ちなみに少し古いデータだが紹介しとく。


Caltechが偏微分方程式を解くAIをオープンソースとして公開


CaltechのDOLCITグループの研究者たちが、偏微分方程式(partial differential equations、PDE)を解くディープラーニングメソッドのFouries Neural Operator(FNO)をオープンソースとして公開した。FNOは既存の他のディープラーニングテクニックを性能面で凌駕しており、従来のソルバよりも3桁高速に処理することができる。

チームは自分たちのモデルと一連の試験について、arXivに発表された論文で説明している。このニューラルネットワークは、ひとつの関数から別の関数へのマッピングを学習することによって、メッシュ解像度の不変なPDEファミリに対する解を提示することができる。モデルでは、グローバル畳み込み(global convolution)を効率的に計算するためにフーリエ変換を採用する。PDEを解く他のディープラーニングモデルと比較した場合、FNOはNavier-Stokesで30パーセント、Darcy Flowで60パーセント低いエラー率を達成している。FNOの適用可能な分野のひとつが、天候パターン(weather patterns)の予測計算の速度向上である — 実験では、従来のソルバで2.2秒を要していた予測を、同モデルによって0.005秒で実行することができた。気象モデルの研究における有望な選択肢のひとつだ。

物理やエンジニアリングの分野では、熱伝導、流体力学、量子メカニズムなど、多くの領域のさまざまな事象を説明する手段としてPDEが使用されている。PDEの解の多くは空間と時間の関数である。例えば、流体の動作を説明する非線形PDEのセットであるNavier-Stokes方程式の解は、特定の場所と瞬間における流体の流れを示すベクタとして出力される。

しかしながら、現実的な状況においては、PDEの閉じた解が存在しない場合が大部分を占めるため、有限要素法(FEM)や有限差分法(FDM)を使用した数学的近似が行われる。これらのテクニックは、対象とする離散的なポイントの詳細なメッシュを作成し、各メッシュ点を囲む小近傍における短時間のPDEの動作を分析することによって機能する。これらの制約範囲内でのPDEは、反復的な数学的処理によって解析可能な、より単純な連立方程式として近似されることが多い。このような方法は、多くの技術分野で主力的ではあるが、デメリットもある — プロセスに時間が掛かる上、さまざまな翼の設計を検証する場合など、グリッド定義や問題のパラメータに少しでも変更があれば再実行が必要になる。

ニューラルネットワークとディープラーニングは、科学的シミュレーションを高速化する手段として有望視されている。PDEを解く場合には、統計サンプル用のデータを短時間で生成可能なモデル開発に目標が置かれることが多い。これは、最終的な結果からシステムの初期状態を決定しようという逆問題(inverse problem)でしばしば適用される。この目的には、有限次元演算子ニューラルFEMという、2つのディープラーニングアプローチが従来より存在する。後者は、畳み込みニューラルネットワーク(CNN)を使用して、解のパラメータ化された近似を求めるものだ。ニューラルFEMで使用されるモデルはメッシュに依存しない反面、表現可能なものがPDEの特定のインスタンスに限られており、PDEのパラメータが変更された場合には再トレーニングが必要になる。

これに対して、Caltechのチームが採用したアプローチでは、解演算子(solution operator)を学習可能なニューラルネットワークを構築する。すなわち、PDEと解の間のマッピングを学習するのだ。論文の筆頭執筆者であるZongi Li氏によれば、この問題は画像間モデルに似たものだ。氏らのネットワークはCNN層ではなく、一連のフーリエ層(Fourier layer)によって構成される。各層では入力に対して、高速フーリエ変換(FFT)、線形変換、逆FFTを順に適用する。FFTによって準線形計算複雑性(quasi-linear computational complexity)が実現するため、モデルはデータの空間的分解能に対して不変なものになる。ただしメッシュは均一であることが必要だ。

FNOの開発は、先頃のNeurIPSカンファレンスで同チームが発表したグラフカーネルネットワーク(GKN)の論文を足場としている。Twitterでの議論でLi氏は、均一なメッシュが適応不能な状況では、速度的に劣るGKNが使用可能であることを指摘した。さらに氏は、次のようにも述べている。

特定のジオメトリでトレーニングされたモデルを別のジオメトリに汎化することはできませんが、異なるジオメトリの解を得るためのトレーニング済モデル、あるいは前処理行列(preconditioner)として使用することは可能です。

ソースはコチラ


コードも公開されているので是非★



画像引用



この記事が参加している募集

この記事が気に入ったらサポートをしてみませんか?