技術要約: COSMA – モジュール型量子アーキテクチャのためのフェルミオン・シミュレーション・カーネルの通信認識型最適化

1. 問題提起

量子シミュレーションは、量子コンピューティングにおける主要なアプリケーションであり、特に計算化学や材料科学において重要である。しかし、化学的に関連のある問題（例：FeMoco活性部位のシミュレーション）において実用的な量子優位性を達成するには、量子誤り訂正を考慮すると数百万の物理量子ビットが必要となる。モノリシックなアーキテクチャは、制御の複雑さ、クロストーク、および熱管理の問題により、この規模へのスケーリングにおいて大きな障壁に直面している。その結果、相互接続された量子プロセッシングユニット（QPU）からなるモジュール型（マルチコア）量子アーキテクチャが提案されている。

モジュール型システムでは、コア間量子通信（テレポーテーションまたは分散エンタングルメントを介したもの）が支配的な性能ボトルネックとなる。状態転送などの操作はコストが高く、ノイズを導入し、最小化されない場合は実行時間を支配する可能性がある。既存のコンパイルフレームワークは、通常、フェルミオンから量子ビットへのマッピング、パウリ・スケジューリング、および量子ビットの割り当てを個別に、あるいは単一コアのトポロジ用に最適化しており、モジュール型システムに固有の通信オーバーヘッドに対処できていない。

対処される核心的な問題は、フェルミオン・シミュレーション・カーネルのコア間状態転送を最小化するための、以下の3つの密結合したステージの結合最適化である：

1. フェルミオン・ツー・量子ビット (F2Q) マッピング: フェルミオン演算子を量子ビット演算子へと符号化する。
2. パウリ・スケジューリング: トロッター化された時間発展回路におけるパウリ項の順序付けを行う。
3. 量子ビットの割り当てとルーティング: 論理量子ビットを物理コアに割り当て、パウリ・ガジェットのためのパリティツリーを合成する。

2. 手法: COSMA フレームワーク

著者らは、クロスレイヤーの共同設計（co-design）を実行する通信認識型コンパイルフレームワークであるCOSMAを提案する。このフレームワークは、コンパイルフローを独立したステップの連続としてではなく、統一された最適化問題として扱う。

A. 問題定式化

入力は、マヨラナ形式のフェルミオン・ハミルトニアンと、特定のコア容量およびコア間距離を持つマルチコア・ハードウェア・グラフである。最適化は、連続する回路スライス間で論理量子ビットが移動する距離の総和として定義される、総コア間転送コストを最小化することを目指す。コスト関数は、4つの結合変数に依存する：

• マッピング ($M$): 各パウリストリングのサポート（関与する量子ビット）を決定する、積保存三項木（Product-Preserving Ternary-Tree: PPTT）マッピング。
• 順序 ($\pi$): トロッター・シーケンスにおけるパウリ項の置換。
• パリティツリー ($\{T_t\}$): 各パウリ・ガジェットのCNOT構造を定義する生成木。
• 割り当て ($\{\ell_l\}$): 各回路スライスにおける論理量子ビットのコアへの割り当て。

B. 主要なアルゴリズム構成要素

1. フェルミオン・ツー・量子ビット・マッピングの最適化

COSMAは、F2Qマッピングを最適化するために**遺伝的アルゴリズム (GA)**を採用している。

• ゲノム符号化: 候補となるマッピングは、2つのコンポーネントからなるゲノムとして符号化される：ツリー形状染色体（三項木の事前順序次数列）と、モード割り当て染色体（フェルミオン・モードの置換）。
• 進化: GAは、トーナメント選択、部分ツリー・スワップ交叉、および部分ツリー置換突然変異を使用する。
• 適応度: 適応度は、特定のマッピングを固定した状態でダウンストリームのスケジューリングおよび割り当てステージを実行した結果得られる、総コア間転送コストである。これにより、マッピングをモジュール型アーキテクチャの通信制約に特化して最適化することが可能となる。

2. パウリ・スケジューリング

マッピングが固定された後、連続するガジェット間のサポートの変化を最小化するようにパウリ項が順序付けられる。COSMAは3つの戦略を評価する：

• 大きさによる順序付け (Magnitude Ordering): 誤差低減のために係数の絶対値によってソートする（標準的な手法）。
• 辞書式順序付け (Lexicographic Ordering): ゲートのキャンセルを最大化するためにアルファベット順でソートする。
• グレイ・インスパイア型順序付け (Gray-Inspired Ordering): サポート・マスクのグレイ符号変換に基づいてソートする。これは、隣接する項が類似したサポートを持つように配置することで、連続するサポート間のハミング距離を最小化し、それによって量子ビットの移動を削減することを目的としている。

3. パリティツリー合成とマルチコア割り当て

このステージでは、各パウリ・ガジェットのCNOTパリティツリーを合成すると同時に、量子ビットの配置を決定する。これにはルックアヘッド・ヒューリスティックが用いられる：

• 配置ミス・スコア (Misplacement Score): ルックアヘッド・ウィンドウ内の将来の相互作用パートナーに対する各量子ビットの距離に基づき、スコアを算出する。
• パリティ・フォレスト構築: サポート量子ビットを現在のコア割り当てごとにグループ化し、コア内でのローカルなCNOTチェーンを形成する。
• ミーティング・コアの選択: すべてのサポート量子ビットへの重み付き距離を最小化する「ミーティング・コア」を特定し、空間的なアトラクター（引き込み点）として機能させる。
• 前方および後方マージ: 異なるコアからの非連結な部分ツリーを、単一のパリティツリーへと反復的にマージする。アルゴリズムは、将来の相互作用距離とコア容量の制約に基づき、代表的な量子ビットを移動させるか、居住中の量子ビットとスワップするかを決定する。このプロセスにより、必要な回路スライスのシーケンスとコア間転送が生成される。

3. 主な貢献

1. 初の共同設計研究: 著者らの知る限り、モジュール型アーキテクチャ上の量子シミュレーション・カーネルに対して、F2Qマッピング、パウリ・スケジューリング、および量子ビット割り当てを共同で最適化する最初の研究である。
2. マッピングのための遺伝的アルゴリズム: 固定されたエンコーディング（Jordan-WignerやBravyi-Kitaevなど）を超えて、コア間通信コストに関して三項木F2Qマッピングを最適化するための新しいGAアプローチ。
3. ヒューリスティック合成および割り当て: 将来の通信圧力を最小化するためにルックアヘッド・ウィンドウを利用した、パリティツリー合成および量子ビット割り当てのための高速なヒューリスティック。
4. オープンソース・フレームワーク: 提供されたリポジトリで利用可能な、提供された戦略を評価するためのGPU加速されたC++実装のオープンソース・フレームワーク（COSMA）。

4. 実験結果

フレーム本フレームワークは、2Dグリッド・マルチコア・アーキテクチャ上で、14個の分子ワークロード（14から90のフェルミオン・モードの範囲）に対して評価された。

• ベースラインとの比較: 最先端のベースライン（固定マッピングであるJW, BK, PE, JKMNと、ハンガリアン割り当ておよび標準的なスケジューリングの組み合わせ）と比較して、COSMAは以下を達成した：
  • グレイ・インスパイア型スケジューリングを使用した場合、通信コストの中央値1.7倍の削減（59.7%削減）。
  • 大きさによる順序付けを使用した場合、中央値1.4倍の削減（42.8%削減）。
  • 特定のワークロードでは、通信コストを最大2.5倍削減。
• アブレーション研究:
  • マッピングの影響: マッピングのみを最適化した場合（固定のスケジューリング/割り当てを使用）、最良の固定マッピングに対して中央値11.4%の改善が見られたが、GAの探索予算が固定されているため、大規模なシステムでは利得が減少した。
  • スケジューリングの影響: グレイ・インスパイア型スケジューリングは、大きさによる順序付け（中央値46.0%削減）および辞書式順序付け（中央値10.5%削減）を大幅に上回り、モジュール型システムにおいてはサポートの遷移を滑らかにすることが極めて重要であることを裏付けた。
• 実行時間: フレームワークはCUDA加速を備えたC++で実装されている。合成および割り当てのステージがボトルネックとなるが、モード数に対しておおよそ4次関数的にスケールする（パウリ項の数に起因）。マルチスレッド化およびCUDAにより、シングルスレッド実行に対してそれぞれ5.0倍および9.4倍の高速化が提供された。

5. 意義と主張

本論文は、マルチコア・ハードウェア上での効率的かつスケーラブルな量子シミュレーションには、クロスレイヤーの共同設計が不可欠であると主張している。結果は、マッピング、スケジューリング、および割り当てを個別に最適化することは、これらのステージが密結合しているため、最適ではないことを示している。

著者らは、統一されたフレームワークを通じてコア間通信を明示的にターゲットにすることで、モジュール型量子コンピュータにおける実行時間を支配するであろう通信オーバーヘッドを大幅に削減できると主張している。このアプローチは、モジュール型アーキテクチャの主要なボトルネックを緩和することにより、化学的に関連のある問題への量子シミュレーションのスケーリングへの道筋を提供するものである。

6. 限界と今後の課題

著者らはいくつかの限界を認めている：

• コストモデル: 転送コストモデルは最短経路距離に基づいており、リンク容量、混雑、またはレイテンシの非対称性は考慮していない。
• トロッター誤差: 本フレームワークは、非可換な項の再順序付けによって導入されるトロッター誤差を明示的に最適化しておらず、通信のみに焦点を当てている。
• マッピング・ファミリー: 現在のアプローチはPPTTマッピング（Nモードに対してN量子ビット）に限定されている。異なるスケーリング比を持つマッピング（例：LDPCコードに基づくもの）はまだ探索されていない。
• パイプラインのオーバーラップ: 目的は転送回数を最小化することであり、通信を局所的な計算とオーバーラップさせる可能性については考慮していない。

今後の課題には、ハードウェア・コストモデルの精緻化、トロッター誤差のトレードオフの組み込み、より表現力豊かな進化オペレータの探索、および他のモジュール型トポロジや変分アルゴリズムへのフレームワークの拡張が含まれる。