Accelerating real-time hybrid-functional calculations with learned ACE and symmetry-based FFT reuse

私たちの世界を構成する固体材料の内部では、電子は静止しているわけではありません。それらは光のリズムに合わせて踊り、周囲の電場に応じてその位置やエネルギーを変化させています。これらの微小な粒子が、特に強力なレーザーに照射されたときにどのように振る舞うかを理解するために、科学者たちは密度汎関数理論と呼ばれる一連の規則を使用しています。この枠組みは地図のような役割を果たし、研究者が電子がどのように動き、相互作用するかを予測することを可能にします。しかし、光が十分に強く、材料を奇妙な非線形状態へと押しやる場合、単純な地図はしばしば失敗します。それらは、電子が互いにどのように反発し合うか、また、半導体や絶縁体の仕組みを理解する上で不可欠な「エキシトン（励起子）」と呼ばれる一時的なペアをどのように形成するかという極めて重要な詳細を見落としてしまうのです。真の姿を捉えるためには、科学者はハイブリッド汎関数と呼ばれる、より複雑で正確な一連の規則を使用しなければなりません。これらの規則には、すべての電子が材料全体にわたって他のすべての電子の存在をどのように感じるかという、困難な計算が含まれています。これは非常に明快な視点を提供しますが、代償は高く、計算は非常に遅く、要求される負荷が大きいため、かつてはこのレベルの精度で電子のリアルタイムの挙動を研究することはほぼ不可能でした。

ある研究者が、これらの重い計算を実用的な速さで実行する方法を見出し、高精度で電子の動きをリアルタイムで観察する道を開きました。彼らは、時間に着目したものと空間に着目したものの、2つの異なるテクニックを開発しました。最初のテクニックは、過去から学ぶことです。これらのシミュレーションにおいて、電子は微小なステップで移動し、各ステップでコンピュータは電子がどのように相互作用するかを再計算しなければなりません。この再計算が、作業の中で最もコストのかかる部分です。研究者は、これらの相互作用が次の瞬間にどのように変化するかというパターンを認識するようにコンピュータに学習させました。毎ステップごとに完全な相互作用を再計算する代わりに、コンピュータはシステムの最近の履歴に基づいて次の変化を予測します。コンピュータは、自身の作業を確認し、ズレを修正するために、時折フルでの高価な計算を行うだけです。このアプローチは、道の状況を熟知しているドライバーが、常に地図を見なくても次のカーブを予測でき、自分がまだ正しい経路を進んでいることを確認するために時折地図をちらりと見るようなものです。

2番目のテクニックは、結晶の中に隠された秩序を利用するものです。シリコンのようなほとんどの固体材料は、繰り返されるパターンから構築されています。この対称性により、材料内の多くの点は、回転したりシフトしたりしているだけで、本質的に互いに同一です。研究者は、電子がどのように相互作用するかを計算する際、これらの同一の点に対して同じ数学的作業を何度も繰り返していることに気づきました。彼らは、これらの繰り返されるパターンを特定し、ある地点からの結果をその「双子」である他のすべての地点に再利用するシステムを構築しました。つまり、コンピュータは一連の等価な場所に対して一度だけ重い作業を行い、その後は単に答えを他の場所へコピーするのです。過去に基づいて未来を予測することと、同一の場所で行った作業を再利用することという、これら2つの手法を組み合わせることで、研究者はシミュレーションに要する時間を劇的に短縮することに成功しました。

彼らが標準的なシリコンブロックでこれらの手法をテストしたところ、驚くべき結果が得られました。これらの計算を行う従来の方法では、シミュレーションの1ステップにつき約37秒かかっていましたが、彼らの新しい結合手法は、1ステップあたり0.5秒未満にまで時間を短縮しました。これは60倍以上の高速化を意味します。研究者は、この大幅なスピードアップにもかかわらず、結果の精度が損なわれていないことを検証しました。シミュレートされた電子は、レーザーから正しい量のエネルギーを吸収し、放出される光（高次高調波発生として知られる）も、より低速で精密な計算で見られる詳細なパターンと一致していました。彼らは、8つの時間間隔ごとに次のステップを予測することが、速度と精度の間で最良のバランスを提供し、電流の変化における誤差をわずか数パーセントに抑えられることを見出しました。

研究者はまた、この手法が世界最速のスーパーコンピュータ、具体的には日本の「富岳」システムでどの程度うまく機能するかをテストしました。彼らは、この手法が効率的にスケールすること、つまり、より多くのコンピュータ・プロセッサをジョブに追加しても、ボトルネックが発生することなく処理が速くなることを示しました。電子を追跡するためのより細かいグリッドを使用してシミュレーションの複雑さを増しても、1ステップあたりの時間は驚くほど低く保たれ、非常に大規模なシステムであっても1秒未満に留まりました。この効率性により、科学者は同じシミュレーションを異なる設定で何度も実行できるようになりました。これは、以前はコストがかかりすぎたことです。例えば、材料をサンプリングするために使用する点の数を変化させることで、電子の応答がどのように変化するかを簡単にテストでき、その結果が粗い近似によるアーティファクトではなく、真に信頼できるものであることを確認できるようになりました。

この研究は、単に時間を節約するだけではありません。それは「何が可能か」を変えるものです。計算コストが管理可能なレベルに抑えられたことで、研究者は今や、これらの高度に正確なハイブリッド汎関数を用いて、非線形な電子ダイナミクスを日常的に探求できるようになりました。彼らは、エキシトン（電子と正孔の結合ペア）が、強力なレーザーパルスの激しい圧力の下でどのように振る舞うかを調査することができます。これは、電子デバイスの根本的な限界を理解し、強い光に耐える、あるいは光を利用できる新しい材料を設計するために極めて重要です。これほどの速度と忠実度でこれらのプロセスをシミュレートできる能力は、科学者が理論モデルと、極限状態における材料の複雑な現実世界の挙動との間の溝を埋めることを可能にし、光と物質が最も基礎的なレベルでどのように相互作用するかについての完全な理解へと私たちを近づけています。