Parallelized Real-time Physics Codes for Plasma Control on DIII-D

本論文は、DIII-Dプラズマ制御システム向けに開発された、TORBEAMおよびSTRIDE物理コードの実行をそれぞれ20ミリ秒および100ミリ秒未満へと最適化することに成功した、将来の核融合発電所における極めて重要な電子サイクロトロン波の伝搬および安定限界計算を可能にするリアルタイム安全なマルチスレッドライブラリを提示するものである。

要約

DIII-Dトカマクのような核融合炉を、巨大で超高温な、電気の渦巻く嵐（プラズマ）であり、磁気ボトルの中に完璧に静止させておく必要があるものだと想像してください。もし嵐が荒れすぎると、装置の壁に衝突して機械を破壊してしまいます。これを安全に保つために、「プラズマ制御システム（PCS）」がパイロットの役割を果たし、絶えず微調整を行います。

しかし、嵐は人間が反応できるよりも速く変化します。パイロットには、瞬きをする間にプラズマの挙هم動を予測し、調整を提案できる超高速のコンピューター・ブレイン（脳）が必要です。ここで、この論文が登場します。

問題点：「一人の作業員」によるボトルネック

あなたが大規模な宴会の料理を作ろうとしているシェフだと想像してください。あなたには、料理の作り方を教えてくれるレシピ（物理コード）があります。しかし、あなたには一人のシェフ（単一のコンピューター・プロセッサー・コア）しかおらず、その一人がすべての刻む、混ぜる、焼くといった作業を行っています。もしレシピが複雑すぎると、シェフは圧倒されてしまい、料理は焦げて、宴会は失敗に終わります。

核融合の世界において、これらの「レシピ」とは、プラズマを加熱する方法や、不安定になりかけていないかを確認するための複雑な物理シミュレーション（TORBEAMやSTRIDEなど）のことです。伝統的に、これらの計算は単一の「シェフ」だけで全てを行おうとしていたため、リアルタイムで実行するには遅すぎました。

解決策：「リアルタイム・セーフ」なシェフのチーム

著者たちは、その単独のシェフを、完璧に同期して働くチームのシェフへと変える新しいシステムを構築しました。

1. マネージャーとワーカー： 彼らは、マネージャーとして機能する特別なライブラリ（一連のルール）を作成しました。マネージャーは、グループ内のワーカースレッド（他のコンピューター・コア）に、小さく独立したタスクを配分します。
2. 混沌ではなく、秩序を： 通常のコンピューター・プログラムでは、ワーカーを増やしすぎると、混乱したり、互いに待ちすぎたり、あるいは一人がミスをした際にシステムをクラッシュさせたりすることがあります。著者たちのシステムは「リアルタイム・セーフ」です。それは、すべての兵士がいつ動き、いつ止まるかを正確に知っている軍隊ユニットのようなものです。彼らは、「準備完了」「完了」「次のラウンドを開始」と伝えるための、特別な「ハンドシェイク（握手）」（アトミック変数）を使用しています。
3. 決定論的なタイミング： 最も重要な部分は、このシステムが厳格な時間制限内に作業を完了することを保証している点です。コンピューターが他の作業で忙しくても関係ありません。このチームは隔離されており、常に時間通りに終了します。これは極めて重要です。なぜなら、コンピューターの処理が遅れると、プラズマはすでに崩壊してしまっている可能性があるからです。

彼らが調理した2つの主要なレシピ

チームはこの新しい「マルチシェフ」システムを使用して、2つの特定の物理コードを高速化しました。

1. TORBEAM：レーザービームのガイド

• 何をするのか： 暗い部屋の中で、動く小さな標的にレーザーで狙いを定めることを想像してください。プラズマはその「部屋」であり、「レーザー」はプラズマの安定性を制御するために使用されるエネルギーのビーム（電子サイクロトロン加熱）です。
• 課題： コンピューターは、レーザービームが正しい場所に当たるように、プラズマ内を通過する正確な経路を計算しなければなりません。
• 結果： 各レーザービーム（ジャイロトロンと呼ばれる異なる装置からのもの）は独立して移動するため、新しいシステムにより、「ワーカー」たちがすべてのビームの経路を同時に計算できるようになりました。
• 速度： 計算を20ミリ秒未満で完了させました。これは、レーザーをリアルタイムで操り、プラズマを安定させるのに十分な速さです。

2. STRIDE：安定性チェッカー

• 何をするのか： 綱渡りを想像してください。STRIDEは、綱渡りの人が落ちそうになっていないかを常にチェックする安全検査官です。プラズマが不安定になって崩壊しそうかどうかを判断するために、「安定性スコア」を計算します。
• 課題： この計算は非常に負荷が高く、通常はリアルタイムでの使用には時間がかかりすぎます。
• トリック： 著者たちは、この安全チェックを多くの小さく独立した断片（例えば、綱の異なるセクションをチェックすること）に分解できることに気づきました。彼らはこれらの断片を「ワーカー」に送り、同時に解決させ、その後で答えを結合しました。
• 速度： 計算時間を約100ミリ秒に短縮しました。これは、惨事が起こる前に制御システムに警告を発するのに十分な速さです。

まとめ

この論文は、専門化された、高度に規律あるコンピューター・プロセッサーのチーム（「リアルタイム・セーフなマルチスレッディング・ライブラリ」）を構築することで、核融合炉が稼働している間に、実際に制御できる速度で複雑な物理シミュレーションを実行できることを証明しています。

• TORBEAM（レーザー制御）は、約20msで動作します。
• STRIDE（安定性チェック）は、約100msで動作します。

この新しい「チームワーク」システムがなければ、これらの計算は核融合炉を安全かつ安定して維持するために有用なほど速くはありませんでした。この研究は、核融合炉の「パイロット」を、プラズマの嵐の極端なスピードに対処できるほど賢くすることができることを証明しています。

技術概要

技術要約：DIII-Dにおけるプラズマ制御のための並列化されたリアルタイム物理コード

問題提起
ITERやSPARCのような将来の核融合炉に向けたリアルタイム・プラズマ制御システム（PCS）には、決定論的なレイテンシと高い堅牢性を持って動作する物理ベースのコードが求められる。電子サイクロトロン加熱（ECH）用のレイトレーシング（TORBEAM）や、理想MHD（磁気流体力学）安定性解析（STRIDE）のためのオフライン物理コードは存在するが、それらのリアルタイム実装は、シングルCPUコア上での性能ボトルネックに直面することが多い。OpenMPのような既存のマルチスレッド・ソリューションは、ハードリアルタイムの保証を欠き、実装依存の挙動に依存し、カーネルとの相互作用による非決定論的なレイテンシ・ジッターを導入するため、核融合PCS環境には不適切である。また、GPU加速は、これらの特定のコードにおいては、スタートアップコストが高く、並列化の粒度が不十分であるため、実用的ではないことが多い。したがって、マイクロ秒規模の決定論性を確保できる、堅牢なCPUベースの並列化戦略が必要となる。

手法
著者らは、TORBEAMおよびSTRIDEコードの並行実行を可能にするため、DIII-DのPCS上にカスタムのリアルタイム安全なマルチスレッディング・ライブラリを開発・実装した。

• ライブラリ・アーキテクチャ: ライブラリは、C11で実装された「マネージャー・ワーカー」パターンを利用している。単一のマネージャー・スレッド（メインのPCSスレッドと一致する）が、複数のワーカー・スレッドにワークユニットを割り当てる。通信は、セマフォのようなカーネルと相互作用する高レベルの抽象化を避け、C11のacquire-releaseメモリモデルを用いた低レベルのアトミック変数のみを通じて行われる。
• 同期メカニメント: システムは、Adaの「ランデブー（rendezvous）」機能に着想を得た双方向同期プロトコルを採用している。マネージャーとワーカーは、アトミックなフラグ（ReadyおよびComplete）を介して互いにステータスを通知し合う。これにより、すべてのスレッドがクリティカルセクションの開始時と終了時に、未定義の動作なしに同期ポイントに到達することを保証する。設計においては、最大のスループットよりも決定論性を優先しており、他のスレッドが追いつくのを待つ間、スレッドがアイドル状態で回転（スピン）することを許容している。
• ハードウェア環境: 実装は、他のプロセスやカーネル割り込みからの干渉を防ぐために隔離されたCPUコアを備えたシステム上で実行された。STRIDEの実装には、レイテンシ・ジッターを最小限に抑えるために商用のリアルタイムOS（RedHawk）を使用し、TORBEAMはコア隔離を備えた標準的なCentOS 7環境で動作させた。
• コード固有の並列化:
  • rt-TORBEAM: 独立したジャイロトロンのレイトレーシング計算を並列化した。コードは、ガウスビームモデルから単一のレイトレースへと簡略化されており、最大吸収位置 $(R, Z)$ を算出した後、ミラー操舵のために正規化磁束座標 ($\rho$) へと変換される。
  • rt-STRIDE: 理想安定性の問題を、Newcomb方程式ではなくリッカチ方程式として再定式化した。これにより、積分領域を分割する「シューティング法」を用いて状態遷移行列（STM）を計算することが可能になる。これらの区間は、「スティフネス（剛性）」指標（理合面付近でのステップ密度が高いこと）に基づいてワーカー・スレッドに分配され、得られたSTMはマネージャーによって結合され、最終的な安定性パラメータ ($\delta W$) が算出される。

主な貢献

1. リアルタイム安全なマルチスレッディング・ライブラリ: 核融合PCSハードウェア上で安全な並列化を可能にする、決定論的でカーネルフリーなマルチスレッディング・ライブラリの開発。これは、OpenMPのような標準ライブラリの限界に対処するものである。
2. 並列化されたrt-TORBEAM: 複数のジャイロトロンを同時に扱うことが可能な、ECHミラー操舵用の並列化されたレイトレーシング・コードの展開。
3. 並列化されたrt-STRIDE: リッカチ定式化と区間分割を用いた、理想MHD安定性計算の並列化実装。これにより、リアルタイムでの安定限界の計算が可能となった。
4. 実験的検証: これらのコードが、実際のプラズマショット中のDIII-D PCSの厳格なサイクルタイム制約内で動作することのデモンストレーション。

結果

• rt-TORBEAMの性能: コードは、4つの操舵されたジャイロトロンに対して一貫して20 ms未満で実行された。システムは、ターゲットとなる $\rho$ 値を合理的に追跡していることを示した（$|\rho_{target} - \rho_{torbeam}| \le 0.05$）。自由なワーカー・スレッドが存在する限り、追加のジャイロトロンが増えてもサイクルタイムは一定であった。特定のエッジケースにおいてサイクルタイムが約40 msにスパイクしたが、これはスレッディング・ライブラリではなく、物理コードに起因するものとされた。
• rt-STRIDEの性能: コードは、72コアのシステム上で、約100 msで $\delta W$ 安定性パラメータを一貫して計算した。プロファイリングにより、STMの積分自体にはわずか約20 msしかかかっておらず、時間の大部分はシリアルなデータ前処理（平衡から磁束座標への変換）によって消費されていることが明らかになった。
• 追跡と制御: 実験結果によれば、物理計算自体は制御に十分な速度であったが、物理的なミラー・ハードウェア（移動に約0.5秒を要する）およびリアルタイム平衡再構成の速度が、特にターゲット値の急速な変化時における全体的な追跡性能を制限していることが示された。

意義と主張
本論文は、開発されたマルチスレッディング・ライブラリが次世代核融合装置の重要なイネーブラー（実現手段）であり、物理ベースのコードが実行時間の要求を満たすことを可能にすると主張している。著者らは、機械学習によるサロゲートモデルが訓練データの範囲外で堅牢性に欠ける可能性があるのに対し、これらの物理ベースのコードは「運用初日から堅牢な性能保証を提供する」ことを強調している。本研究は、適切な並列化と決定論的なスレッディングを用いることで、TORBEAMやSTRIDEのような複雑な物理コードを、ECHの堆積量や理想安定限界に関する不可欠な情報を提供するために、PCSへ統合できることを示している。著者らは、このライブラリが垂直安定性解析やプロファイル予測を含む他のリアルタイム・コードにも適用可能であることを示唆しており、今後の取り組みとして、レイテンシをさらに削減するためにデータ前処理ステップの並列化に焦点を当てるべきであると述べている。