CP2K on the road to exascale

水滴が単なる静止した液体としてではなく、あらゆる原子が絶えず動き、衝突し、反応している、生きている、呼吸しているシステムとしてどのように振る舞うのかを理解しようとしている場面を想像してみてください。これを行うために、科学者たちは、極めて微小な世界を支配する物理学である量子力学の厳格な法則に従って物質を扱う、強力なタイプのコンピュータ・シミュレーションを使用します。これらのシミュレーションは、電子と原子核の挙動を同時に追跡しなければならないため、非常に負荷が高く、膨大な計算能力を必要とします。数十年にわたり、研究者たちはこれらの計算を実行するためにスーパーコンピュータに頼ってきましたが、研究対象となるシステムが大きくなるにつれ（小さなクラスターから、数百万の原子を持つ広大な材料へと移行するにつれ）、答えを得るまでに要する時間がボトルネックとなっています。目標は常に、コンピュータがこれらの問題を非常に高速に解けるようにし、科学者が原子のわずかな集まりだけでなく、原子の世界全体をシミュレートしながら、複雑な化学プロセスが展開される様子をリアルタイムで観察できるようにすることでした。

これらの種の原子シミュレーションに広く使用されているツールであるCP2Kソフトウェア・パッケージの開発に携わる研究チームは、現在の能力と、次世代のスーパーコンピュータとして知られるエクサスケール・マシンとの間の溝を埋めるべく取り組んできました。これらの新しいコンピュータは、1秒間に100京回の計算を行うように設計されており、速度の劇的な飛躍をもたらします。しかし、単にハードウェアを速くするだけでは不十分であり、ソフトウェアはこれらのマシン特有の動作方法を処理できるように書き換えられなければなりません。チームの作業は、シミュレーションを駆動するコア・アルゴリズムの改良に焦点を当てており、膨大な数のプロセッサ・コアを同時に使用しても、処理が滞ることなくスケールアップできるようにすることを目指しています。彼らは特に、第2世代のカー・パリネロ分子動力学と呼ばれる特定の手法に関心を持っており、これは、電子を最も安定した状態に保ちながら、原子の動きをナノ秒といった非常に長い期間にわたってシミュレートすることを可能にします。このアプローチは、凍結された物質のスナップショットではなく、液体や固体のような現実的な有限温度のシステムを研究するために極めて重要です。

研究者たちは、現代のコンピューティングにおける3つの主要なトレンドを特定しており、ソフトウェアはこれらに適応する必要があります。それは、大規模な並列処理の必要性、計算のためのグラフィックス・プロセッシング・ユニット（GPU）の広範な利用、そして様々な精度レベルで計算を実行するハードウェアの可用性です。これらの課題に対処するため、チームはコードのデータ処理方法を再構築しました。彼らは、大規模な数学的問題を、コンピュータの異なる部分が互いに頻繁に確認を取ることなく同時に解決できる、小さく独立した塊へと分割する新しい方法を開発しました。これは、データを特定のスレッドに割り当てられた固定ブロックとして整理することで実現され、多くのプロセッサが同時に同じ情報にアクセスしようとする際に通常発生する遅延を排除します。また、ビデオゲーム用に設計されたものの、現在では科学計算に不可欠となっている強力なチップであるGPUのサポートを統合し、最も重い計算をこれらの特化型アクセラレータにオフロードできるようにしました。

彼らの努力の大部分は、電子間の相互作用という数学的な複雑さに取り組むことに費やされています。これは、シミュレーションの中で最もコストがかかる部分です。従来、大規模なシステムにおける電子同士の反発を計算するには、システムが大きくなるにつれて爆発的に増大するステップ数が必要であり、大規模なシミュレーションをほぼ不可能にしていました。チームは、無視できる相互作用を排除する新しい技術を導入し、重要ではない微小な力を効果的に無視することで、計算コストを劇的に削減しました。また、電子密度をより小さく計算しやすい行列を用いて近似し、その結果を後でより精密だが安価な計算で補正する手法も開発しました。これにより、高い精度を維持しながら、計算時間を大幅に短縮することが可能になります。さらに、彼らは「混合精度」の使用についても探求しています。これは、コンピュータが計算の大部分をより高速に計算できる低い数値精度で行い、最終的な結果が正確であることを保証するために、特別な数学的補正を用いる手法です。

これらの開発による成果は、すでに目覚ましいものです。バルク水のシミュレーションにおいて、研究者たちは最適化されたコードが、数万個のコアを持つスーパーコンピュータ上で最大100万個の原子を扱えることを実証しました。より最近では、新しいサブマトリックス法と電子密度の純化スキームを組み合わせることで、1億個を超える原子を含むシステムの記録的なシミュレーションを達成しました。この大規模な計算は、324ペタフロップスの持続性能を発揮してスーパーコンピュータ上で実行され、マシンの全理論的性能の67パーセント以上を活用しました。この効率性は、ソフトウェアが現代のハードウェアの生のスピードを効果的に活用し、以前は手の届かなかった問題を解決できることを証明しています。

将来を見据えて、著者らは、これらの古典的なシミュレーションが他の新興技術によって拡張される未来を描いています。彼らは、人工知能、特にディープニューラルネットワークが、複雑な計算をゼロから行う必要があるのか、それとも訓練されたネットワークが十分に正確な回答を提供できるのかといった判断を下すために使用される可能性があると示唆しています。また、量子力学の原理に基づいて動作する量子コンピュータが、古典的なスーパーコンピュータと連携して、現在の古典的なマシンでは困難すぎる電子の挙動の計算などの特定の課題を解決するという、ハイブリッドなアプローチも構想しています。これらの将来的な応用はまだ計画段階にありますが、現在のCP2Kに関する研究は、ソフトウェアがハードウェアと共に進化し、エクサスケール・コンピューティングの約束を、物質界を理解するための実用的な現実へと変えるための不可欠な基礎を築いています。