現世代の干渉計検出器の重力波データから連星中性子星(BNS)の合体イベントを検出するタスクで、汎用の畳み込みベースのシングルショットオブジェクト検出モデルであるYOLOv5モデルのアプリケーションを示します。また、モデルのトレーニング、検証、およびテストの手順で使用される近似波形モデルに基づいた合成データの生成および準備タスクについても詳しく説明します。このアプローチを使用すると、単一クラスの検証データセットで0.945、テストデータセットで0.978の平均精度(mAP_ [0.50])値を達成できます。さらに、トレーニングされたモデルは、LIGOH1検出器データでGW170817イベントを識別することに成功しています。このイベントの識別は、インスピレーションの最終段階で大きなグリッチを除去する必要なしに、追加の前処理ステップでLIGOL1検出器データに対しても可能です。 GW190425イベントの検出はあまり成功していません。これは、信号対雑音比によるパフォーマンスの低下を証明しています。私たちの研究は、YOLOv5モデルが、第1段階の検出アラームパイプラインにとって興味深いアプローチであり、より複雑なパイプラインに統合された場合、物理的なソースパラメータのリアルタイムの推論にとって興味深いアプローチであることを示しています。
We demonstrate the application of the YOLOv5 model, a general purpose convolution-based single-shot object detection model, in the task of detecting binary neutron star (BNS) coalescence events from gravitational-wave data of current generation interferometer detectors. We also present a thorough explanation of the synthetic data generation and preparation tasks based on approximant waveform models used for the model training, validation and testing steps. Using this approach, we achieve mean average precision (mAP_[0.50]) values of 0.945 for a single class validation dataset and as high as 0.978 for test datasets. Moreover, the trained model is successful in identifying the GW170817 event in the LIGO H1 detector data. The identification of this event is also possible for the LIGO L1 detector data with an additional pre-processing step, without the need of removing the large glitch in the final stages of the inspiral. The detection of the GW190425 event is less successful, which attests to performance degradation with the signal-to-noise ratio. Our study indicates that the YOLOv5 model is an interesting approach for first-stage detection alarm pipelines and, when integrated in more complex pipelines, for real-time inference of physical source parameters.