[論文レビュー] PyTransport: A Python package for the calculation of inflationary correlation functions

David J. Mulryne, John W. Ronayne|arXiv (Cornell University)|Sep 1, 2016

Cosmology and Gravitation Theories被引用数 12

ひとこと要約

PyTransport 2.0 は、曲がった場の幾何構造を持つ多場理論インフレーションモデルにおけるインフレーション相関関数（パワー スペクトルやバイスペクトルなど）を計算する、Python-C++ハイブリッドパッケージである。このパッケージは、連立微分方程式を数値的に解くためにトランスポート法を用い、C++の高速性とPythonの使いやすさを組み合わせることで、宇宙論的観測量の効率的で正確かつ拡張可能な計算を実現する。

ABSTRACT

PyTransport constitutes a straightforward code written in C++ together with Python scripts which automatically edit, compile and run the C++ code as a Python module. It has been written for Unix-like systems (OS X and Linux). Primarily the module employs the transport approach to inflationary cosmology to calculate the tree-level power-spectrum and bispectrum of user specified models of multi-field inflation, accounting for all sub and super-horizon effects. The transport method we utilise means only coupled differential equations need to be solved, and the implementation presented here combines the speed of C++ with the functionality and convenience of Python. This document details the code and illustrates how to use it with a worked example. It has been updated to be a companion to the second version of the code, PyTransport 2.0, which includes functionality to deal with models of inflation with a curved field space metric.

研究の動機と目的

• 多場理論インフレーションモデルにおけるインフレーション相関関数を計算する、ユーザーフレンドリーで高性能なツールを提供すること。
• 元々の PyTransport フレームワークを拡張し、曲がった場の幾何構造を持つモデルを扱えるようにすることで、より現実的な宇宙論的モデル解析を可能にすること。
• C++の数値的効率性とPythonのスクリプトの柔軟性を組み合わせ、迅速なプロトタイピングおよび大規模計算を可能にすること。
• MPI を用いた並列化をサポートすることで、複数プロセッサにわたるスケーラブルな計算を実現すること。
• 再利用可能で拡張可能なフレームワークを提供し、研究者がカスタムのインフレーションモデルに容易にコードを適応・拡張できるようにすること。

提案手法

• コードはトランスポート法を採用し、インフレーション相関関数を数値積分可能な連立常微分方程式（ODE）系に再定式化する。
• 計算負荷の高い ODE の数値解法は、コンパイル済み C++ コードで実行され、Python モジュールとしてラップされて高速性を確保する。
• Python スクリプトがコンパイル、設定、実行を管理することで、シームレスな統合と迅速なプロトタイピングを実現する。
• 変化するステップサイズを用いた ODE ソルバーと誤差制御を採用し、数値的精度と収束テストを保証する。
• Mpi4Py を用いた並列計算をサポートし、複数プロセスにわたり k モードの計算を分散することでスケーラビリティを向上させる。
• パワー スペクトルやバイスペクトルの評価など、一般的なタスクを補助するユーティリティ関数をフレームワークに統合する。

実験結果

リサーチクエスチョン

• RQ1 曲がった場の幾何構造を持つ多場理論モデルにおけるインフレーション相関関数を、どのように効率的かつ正確に計算できるか？
• RQ2 宇宙論的シミュレーションにおいて、Python の使いやすさと C++ の計算速度を組み合わせた際の、パフォーマンスと精度のトレードオフは何か？
• RQ3 トランスポート法を、モジュラーで再利用可能かつ拡張可能なソフトウェアフレームワークとして効果的に実装できるか？
• RQ4 MPI を用いた並列化は、バイスペクトルおよびパワー スペクトル計算における大規模な k モードサンプリングのスケーラビリティをどのように向上させるか？
• RQ5 Python 2.7 と Python 3 の両方をサポートする際の実用的課題は何か？また、メジャーバージョンのアップグレード時に後方互換性をどのように管理できるか？

主な発見

• PyTransport 2.0 は、トランスポート法を用いて、曲がった場の幾何構造を持つ多場理論インフレーションモデルの木構造のパワー スペクトルおよびバイスペクトルを正常に計算した。
• Python-C++ハイブリッド実装により、ネイティブ C++ に近いパフォーマンスを達成しており、大規模な数値的スタディに適している。
• Mpi4Py を用いた並列計算をサポートしており、複数プロセッサにわたり k モードの計算を効率的に分散できる。
• PyTransScripts.pSpectra や PyTransScripts.eqSpectra などの関数により、k モードの配列に対してパワー スペクトルおよび等価バイスペクトルを、完全な誤差制御のもとで計算可能である。
• alpBetSpectra および alpBetMpi 関数により、複数の時間および k モード設定にわたるバイスペクトルの時間的追跡が可能となり、非ガウス性の時間発展を動的解析する支援が可能である。
• コードは拡張性を考慮して設計されており、ユーザーが下位の C++ コードを変更したり、Python スクリプトをカスタムインフレーションモデルに合わせて調整・再利用できる。