Spectra Recognition Model for O-type Stars Based on Data Augmentation

Frontiers in Astronomy and Space Sciences

Wen-Yu Yang¹  Zhi-Qiang Zou²  Ke-Fei Wu³  A-Li Luo³ 

[1] College of Computer, Nanjing University of Posts and Telecommunications, Nanjing, China;College of Computer, Nanjing University of Posts and Telecommunications, Nanjing, China;Jiangsu Key Laboratory of Big Data Security and Intelligent Processing, Nanjing University of Posts and Telecommunications, Nanjing, China;Key Laboratory of Optical Astronomy, National Astronomical Observatories, Chinese Academy of Sciences, Beijing, China;School of Astronomy and Space Science, University of Chinese Academy of Sciences, Beijing, China;

关键词: data augmentation;    generative adversarial network;    celestial spectra recognition;    residual network;    attention mechanism;   

DOI  :  10.3389/fspas.2021.634328

来源: Frontiers

【 摘 要 】

It is an ongoing issue in astronomy to recognize and classify O-type spectra comprehensively. The neural network is a popular recognition model based on data. The number of O-stars collected in LAMOST is <1% of AFGK stars, and there are only 127 O-type stars in the data release seven version. Therefore, there are not enough O-type samples available for recognition models. As a result, the existing neural network models are not effective in identifying such rare star spectra. This paper proposed a novel spectra recognition model (called LCGAN model) to solve this problem with data augmentation, which is based on Locally Connected Generative Adversarial Network (LCGAN). The LCGAN introduced the locally connected convolution and two timescale update rule to generate O-type stars' spectra. In addition, the LCGAN model adopted residual and attention mechanisms to recognize O-type spectra. To evaluate the performance of proposed models, we conducted a comparative experiment using a stellar spectral data set, which consists of more than 40,000 spectra, collected by the large sky area multi-object fiber spectroscopic telescope (LAMOST). The experimental results showed that the LCGAN model could generate meaningful O-type spectra. In our validation data set, the recognition accuracy of the data enhanced recognition model can reach 93.67%, 8.66% higher than that of the non-data enhanced identification model, which lays a good foundation for further analysis of astronomical spectra.

【 授权许可】

CC BY