On the Acceleration of Pulsar Timing computations using Normalising Flows and Parallelisation

宇宙が広大な宇宙のオーケストラであり、その交響曲の中に、重力波背景放射の微かな、リズムを刻む鼓動が隠されていると想像してみてください。これらは音波ではなく、数十億年前に巨大なブラックホールが共に踊りながら作り出した、時空そのもののさざ波です。この宇宙のハミングを聴くために、科学者たちは「パルサー・タイミング・アレイ」を使用します。パルサーを、宇宙で最も信頼できるメトロノームだと考えてください。それらは、非常に速く、かつ一定に回転し、灯台のように光のビームを放つ死んだ星です。これらのフラッシュのタイミングを極めて精密に測定することで、天文学者は、パルサーと私たちの間の時空が、通過する重力波によって引き伸ばされたり、押しつぶされたりしたかどうかを検知することができます。

しかし、この宇宙の音楽を聴くことは非常に困難です。データは乱雑であり、パルサー自身のふらつきによる「ノイズ」、霧のようなレンズとして機能する星間物質、そして私たちの時計の限界などによって満たされています。それは、ハリケーンの中でささやき声を聴こうとするようなものです。データを理解するために、科学者たちは複雑な数学を用いて信号をノイズから分離しますが、その数学は非常に重厚で、データも膨大であるため、単一のコンピュータが数字を計算するのに数週間、時には数ヶ月かかることも珍しくありません。この論文は、このボトルネックに取り組み、「これらの宇宙のパズルを解くために、永遠に待つことなく、より速い方法を見つけることはできるだろうか？」という単純かつ極めて重要な問いを投げかけています。

この研究の著者である Churchil Dwivedi、Hiya Shah、および Hemanga Tahbildar は、「ノーマライジング・フロー（正規化流）」と呼ばれる新しいハイテク・ツールを、天文学界で信頼されている従来の手法と比較テストすることに決めました。彼らはこの問題をレースのように扱いました。トラックの片側には、伝統的なランナーたちがいました。一つは、「パラレル・テンパリング（並列温度変化）」（多くのハイカーを異なる温度で送り出し、高い頂上を見つけるために場所を入れ替えさせる様子を想像してください）という手法を用いる、働き者の PTMCMCSAMPLER です。もう一方は、「ネステッド・サンプリング（入れ子状サンプリング）」（探索範囲を系統的に絞り込んでいく探検チームのようなもの）を用いる、新しいランナーの PARALLEL BILBY です。そして反対側のトラックには、「ノーマライジング・フロー」という巧妙なトリックを用いるスプリンター、POCOMC が登場しました。

このトリックを理解するために、データの風景を、ねじれ、絡まった毛糸玉だと想像してみてください。従来の手法は、端を一つずつ引っ張ることで、長い時間をかけて解こうとします。しかし、POCOMC は、数学的な「魔法の杖」（ノーマライジング・フロー）を使って、毛糸を瞬時に滑らかで単純な一本の線へと真っ直に伸ばし、解決策を見つけやすくします。研究者たちは、実際のパルサー観測を模したシミュレーションデータを用いて、ノイズが単純なシナリオと、相関関係が複雑に絡み合ったシナリオの両方で、これらのツールをテストしました。

結果は、「単一のマシンで最速」と「巨大なチームで最速」という混合したものでした。単一のコンピュータ・ノードで実行した場合、POCOMC は圧倒的なチャンピオンでした。それは、旧来の手法（PTMCMCSAMPLer）が10日以上かかった複雑な探索を、わずか約10時間で解決しました。さらに印象的なことに、特定のタイプの単一パルサー解析において、それは約10時間で完了しましたが、古い手法では数日を要したはずです。著者らは、POCOMC が非常に効率的であったため、驚異的な速さを出すためにスーパーコンピュータや特別なグラフィックスカードを必要とせず、単に賢く動作する標準的なコンピュータがあれば十分であったことを見出しました。

しかし、コンピュータのチームを投入すると、物語は変わります。研究者たちが多くのコンピューティング・ノード（最大16ノード、各48スレッド）に作業を分散させたとき、PARALLEL BILBY アーキテクチャがリードを取りました。それは、チームの中で労働を分担することにおいて最も優れていました。パルサー・アレイ全体にわたる複雑な相関関係を含む探索（「ヘリングス・アンド・ダウンズ」探索と呼ばれます）において、PARALLEL BILBY は16ノードを使用して時間を約100分にまで短縮しましたが、単一ノード上の POCOMC ではもっと長い時間がかかっていたでしょう。興味深いことに、POCOMC は、あまり多くのコンピュータを追加しようとすると壁に突き当たりました。4ノードを超えると、追加しても速くならず、場合によっては逆に遅くなることがありました。著者らは、あまりに多くのプロセッサが同時に作業しようとすると、「魔法の杖」のステップが混雑してしまうためではないかと推測しています。

また、論文では ECORR と呼ばれる特定の種類のノイズを扱う異なる方法についても比較しています。彼らは、「ファスト・シャーマン・モリソン（fast-sherman-morrison）」法がこのノザイズを計算する上で最も効率的な方法である一方、標準的な「シャーマン・モリソン（sherman-morrison）」法が最も遅いことを発見しました。また、PTMCMCSAMPLER が最も非効率的であり、他の手法よりも大幅に時間がかかり、場合によってはコンピュータを増やしても速くなるどころか、むしろ遅くなることも指摘しました。

結局のところ、著者らは、将来のパルサー・タイミングは「適切な道具を適切な仕事に選ぶこと」にかかっていると示唆しています。もしあなたが単一のコンピュータで作業しているか、あるいは一度に一つのパルサーを分析しているのであれば、POCOMC が新しいスターであり、スーパーコンピュータを必要とせずに劇的なスピードアップを提供します。しかし、もしあなたが数十のノードを持つ巨大なスーパーコンピュータへのアクセス権を持っており、パルサー・アレイ全体を一度に分析する必要があるならば、PARALLEL BILBY が最良の選択肢となります。これらの知見は、より速い手法を用いることで、科学者たちがこれまで計算不可能であった、より複雑な宇宙のモデルを間もなく探索できるようになり、それが重力波の宇宙のハミングをようやく聴き取ることに近づけてくれる可能性を示唆しています。