exa-PD: A scalable high-performance workflow for multi-element phase diagram construction

あなたは、完璧な新しい合金を発明しようとしているシェフだと想像してください。それは鋼鉄よりも強く、銅よりも導電性の高い材料です。コンロに火をつける前に、あなたには「地図」が必要です。材料科学の世界では、この地図は「相図（フェーズダイアグラム）」と呼ばれます。これは、特定の温度とレシピにおいて、元素の混合物がどのような状態（固体、液体、あるいは奇妙な結晶構造など）になるかを正確に教えてくれるものです。この地図がなければ、新しい材料を作ろうとする試みは、オーブンが必要なのか冷凍庫が必要なのかを知らずにケーキを焼こうとするようなものです。結局、焦げた塊を作ってしまうだけかもしれません。これらの地図を正確に描くために、科学者たちはあらゆる可能な混合物の「自由エネルギー」を算出するという、非常に複雑な数学を実行しなければなりません。自由エネルギーとは、材料の内部的な「気分」のようなものだと考えてください。つまり、その材料がどれほど固体のままでいたいか、あるいは溶けたいと思っているか、ということです。単一の混合物のこの「気分」を計算するだけでも大変な作業ですが、何千もの異なるレシピを同時に計算することは、通常、スーパーコンピュータがその重荷を担わなければならないほどの膨大なタスクです。

ここで、「exa-PD」と呼ばれる新しいツールが登場します。これは、これら膨大な計算のための、非常に効率的なキッチンマネージャーとして機能します。この論文は、exa-PDを、以前よりもはるかに速く、かつ大規模に多元素相図を構築するために設計された、高度に組織化されたワークフローとして紹介しています。一つずつシミュレーションを実行する代わりに、exa-PDはParslというスマートなシステムを使用して、数百の異なるコンピュータ・タスクを同時に調整します。これは、コンピュータを賑やかなシェフのチームのように扱う仕組みです。あるタスクは、数値を高速で処理できる強力なグラフィックスカード（GPU）に割り当てられ、別のタスクは、グラフィックスカードがまだこなせない特定のトリッキーな仕事を処理する標準的なプロセッサ（CPU）に割り当てられます。このシステムは非常に整理されているため、単一のコンピュータから大規模なスーパーコンピュータへとスケールアップさせることができ、ほぼすべてのプロセッサを稼働させ、完璧に同期させておくことができます。

研究者たちは、Perlmutterという名前のスーパーコンピュータを用いてこのシステムをテストし、それが極めてうまく機能することを確認しました。銅・ジルコニウム合金のシミュレーションを実行した際、システムは「ニアリニア・スケーリング（ほぼ線形なスケーリング）」を示しました。これは、コンピュータの数を2倍にすれば、作業時間が半分になることを意味します。彼らは32個のGPUと32個のCPUにおいてこの効率性を測定し、それぞれ約89%と90%の並列効率を達成しました。これは、システムが待ちぼうけを食らって時間を無駄にすることなく、チーム全体を常に動かし続けていることを示しています。

ワークフロー自体は、3つのステップからなるプロセスです。第一に、新しい果物の正確な重さを量るために標準的なリンゴと比較するように、既知の「参照用」結晶と比較することで、固体のエネルギーを計算します。第二に、異なる材料を混ぜたときにエネルギーがどのように変化するかを算出する特別な手法を用いて、液体の混合物についても同様の計算を行います。第三に、数値が正しいことを確認するために、オプションとして融点をチェックします。これらすべての重い計算が終わると、システムは自動的にデータをデータベースに整理し、固体と液体がどこに存在するのかを示す最終的な相図を描くために使用できるようになります。

この論文は、従来の手法が個々の計算の精度を高めることに焦点を当てているのに対し、exa-PDは異なる問題、すなわち「いかにしてコンピュータを圧倒させることなく、何千もの計算を同時に実行するか」という問題を解決していることを強調しています。多くの工程がグラフィックスカードによって加速できる一方で、不可欠なステップの中には依然として標準的なプロセッサを必要とするものがあるため、異なる種類のコンピュータ・パワーを組み合わせる能力が重要であると明記しています。この柔軟なアプローチを用いることで、著者らは、複雑な材料の信頼性の高い高性能な相図を生成することが可能であることを実証しており、適切な合成経路を推測することなく、新しい合金の発見を加速させる道を開いています。