Abstract：With the implementation of more and more large-scale spectral surveys, many star spectral data are generated, which is of great significance to the study of star evolution theory, but also brings great challenges to the traditional spectral classification and processing. In the LAMOST DR7 (v2.0 version) spectral data set released in 2021, the total spectrum of stars has reached the order of millions, but the number of O-type is only 129, and the data set is seriously unbalanced. Traditional machine learning classification methods cannot achieve good results for such a large amount of data and a serious imbalance of data sets. Therefore, they are mostly used to classify the spectra of adjacent stars of two types, partial types or subclasses. An automatic star spectrum classification method based on semi-supervised mode of one dimensional convolutional neural network (CNN) and one dimensional generative adversarial network (GAN) is used to solve this problem. Firstly, each spectrum is tailored and denoised, and the wavelength range of the spectrum is 370.00~867.16 nm. Then, uniform sampling and normalization are carried out to generate 1×3 700 data set samples, which are sent to one-dimensional CNN for training. In order to avoid overfitting and improve model’s prediction ability for unknown data, dropout is applied between the full connection layer and the pooling layer. This one-dimensional CNN network is used to classify six kinds of spectra except for O-type, and the average classification accuracy is 98.08%. This experiment uses one-dimensional GAN network to solve the problem that the number of O-type is seriously small. The input of GAN is a noise signal whose size is 1×900.Through the fully connected three-layer convolution operation in the generator, the output size is 1×3 700. The separate alternating iterative training of the generator and discriminator finally outputs the required number of O-type star samples to expand the data set. Compared with oversampling, using data sets expanded by GAN to classify the star spectra can improve the classification accuracy of O-type stars from 72.92% to 97.92%, and the accuracy of the classifier can reach 96.28%. The experimental results show that the automatic star spectra classification method using this semi-supervised mode can realize the rapid and accurate classification of seven types of star spectra and can also be used to mine the unclassified spectra marked as “unknown”.
[1] Li Xiangru，Pan Ruyang，Duan Fuqing. Research in Astronomy and Astrophysics，2017，17（4）：36.
[2] Morgan W, Keenan P. Annual Review of Astronomy and Astrophysics, 1973, 11: 29.
[3] Du Bing，Luo A-li，Zhang Shuo，et al. Research in Astronomy and Astrophysics，2021，21（8）：202.
[4] Fan J，Han F，Liu H. Nationanl Science Review，2014，1（2）：293.
[5] Schierscher F，Paunzen E. Astronomische Nachrichten，2011，332（6）：597.
[6] Liu C，Cui W Y，Zhang B，et al. Research in Astronomy and Astrophysics，2015，15（8）：1137.
[7] Kheirdastan S，Bazarghan M. Astrophysics and Space Science，2016，361（9）：304.
[8] SHI Chao-jun，QIU Bo，ZHOU Ya-tong，et al（石超君，邱 波，周亚同，等）. Spectroscopy and Spectral Analysis （光谱学与光谱分析），2019，39（4）：1312.
[9] Wen Xiaoqing，Yang Jinmeng. Chinese Journal of Physics，2021，69：303.
[10] XU Ting-ting，MA Chen-ye，ZHANG Jing-min，et al（许婷婷，马晨晔，张静敏，等）. Acta Astronomica Sinica（天文学报），2019，60（2）：11.
[11] Du Changde，Luo Ali, Yang Haifeng. New Astronomy，2017，51：51.
[12] Zheng Zipeng，Qiu Bo，Luo A-li，et al. Publications of the Astronomical Society of the Pacific，2020，132：024504.
[13] Si Jianmin，Li Yinbi，Luo A-li，et al. Research in Astronomy and Astrophysics，2015，15（10）：1671.
