Advancing Gamma-Ray Burst Identification through Transfer Learning with Convolutional Neural Networks

Autor: Zhang, Peng, Li, Bing, Gui, Ren-zhou, Xiong, Shao-lin, Wang, Yu, Zhang, Yan-qiu, Wang, Chen-wei, Liu, Jia-cong, Xue, Wang-chen, Zheng, Chao, Yu, Zheng-hang, Zhang, Wen-long
Rok vydání: 2024
Předmět:
Druh dokumentu: Working Paper
Popis: The Rapid and accurate identification of Gamma-Ray Bursts (GRBs) is crucial for unraveling their origins. However, current burst search algorithms frequently miss low-threshold signals or lack universality for observations. In this study, we propose a novel approach utilizing transfer learning experiment based on convolutional neural network (CNN) to establish a universal GRB identification method, which validated successfully using GECAM-B data. By employing data augmentation techniques, we enhance the diversity and quantity of the GRB sample. We develop a 1D CNN model with a multi-scale feature cross fusion module (MSCFM) to extract features from samples and perform classification. The comparative results demonstrated significant performance improvements following pre-training and transferring on a large-scale dataset. Our optimal model achieved an impressive accuracy of 96.41% on the source dataset of GECAM-B, and identified three previously undiscovered GRBs by contrast with manual analysis of GECAM-B observations. These innovative transfer learning and data augmentation methods presented in this work hold promise for applications in multi-satellite exploration scenarios characterized by limited data sets and a scarcity of labeled samples in high-energy astronomy.
Comment: 17 pages, 7 figures
Databáze: arXiv