StellarGAN: Classifying Stellar Spectra with Generative Adversarial Networks in SDSS and APOGEE Sky Surveys

Abstract

Extracting precise stellar labels is crucial for large spectroscopic surveys like the Sloan Digital Sky Survey (SDSS) and APOGEE. In this paper, we report the newest implementation of StellarGAN, a data-driven method based on generative adversarial networks (GANs). Using 1D operators like convolution, the 2D GAN is modified into StellarGAN. This allows it to learn the relevant features of 1D stellar spectra without needing labels for specific stellar types. We test the performance of StellarGAN on different stellar spectra trained on SDSS and APOGEE data sets. Our result reveals that StellarGAN attains the highest overall F1-score on SDSS data sets (F1-score = 0.82, 0.77, 0.74, 0.53, 0.51, 0.61, and 0.55, for O-type, B-type, A-type, F-type, G-type, K-type, and M-type stars) when the signal-to-noise ratio (S/N) is low (90% of the spectra have an S/N < 50), with 1% of labeled spectra used for training. Using 50% of the labeled spectral data for training, StellarGAN consistently demonstrates performance that surpasses or is comparable to that of other data-driven models, as evidenced by the F1-scores of 0.92, 0.77, 0.77, 0.84, 0.84, 0.80, and 0.67. In the case of APOGEE (90% of the spectra have an S/N < 500), our method is also superior regarding its comprehensive performance (F1-score = 0.53, 0.60, 0.56, 0.56, and 0.78 for A-type, F-type, G-type, K-type, and M-type stars) with 1% of labeled spectra for training, manifesting its learning ability out of a limited number of labeled spectra. Our proposed method is also applicable to other types of data that need to be classified (such as gravitational-wave signals, light curves, etc.).