Spectroscopy and Spectral Analysis, Volume. 45, Issue 6, 1670(2025)

RN-SM:Stellar Spectral Classification Algorithm Based on ResNet Feature Extraction

YANG Jia-ming1, TU Liang-ping1,2、*, LI Jian-xi2, and MIAO Jia-wei1
Author Affiliations
  • 1School of Electronic and Information Engineering, University of Science and Technology, Anshan 114051,China
  • 2School of Mathematics and Statistics, Minnan Normal University, Zhangzhou 363000, China
  • show less

    The advent of the Large Survey Telescope made it possible to build spectral databases of stellar. To study the massive stellar spectrum data in the database more efficiently and effectively, it is necessary to develop a fast and efficient automatic stellar spectrum processing algorithm. Based on the deep learning model ResNet, a hybrid deep learning algorithm named RN-SM is built in this paper. The algorithm consists of five steps: ① Normalization processing: A linear normalization function is used to normalize the stellar spectrum, ensuring it has a uniform scale. ② Denoising processing: The Ces algorithm is used to denoise the stellar spectrum, removing noise from the data. ③ Composite RGB image: Three channels of an RGB image, corresponding to the gray image generated by the same spectrum. The superposition of the same spectrum makes the main features of the stellar spectrum more pronounced and easier to work with in subsequent analysis. Here, we normalize the continuous spectrum of the stellar spectrum so that the content shown in the RGB image is the spectral line information of the stellar spectrum. At the same time, we analyze the feasibility of data conversion (synthetic RGB image) by using the main spectral line information of the stellar spectrum as a reference and investigating whether the relevant pixel position of the synthetic RGB image contains these features. It is proven that the method of data conversion (synthesizing an RGB image) proposed in this paper is feasible. ④ Feature extraction: To facilitate the connection of the SoftMax algorithm, the ResNet algorithm was used to extract features from stellar spectra. The 1×2 048 feature vector from the 64×64 RGB image was extracted. The ResNet algorithm contains 49 convolution layers in total. Automatic classification: The feature vector is transferred to the SoftMax module for automatic classification. The loss function used by SoftMax is the sum of the dataset loss and the regularization term loss. After 10000 iterations, the loss function becomes stable. When the RN-SM algorithm uses the spectra of A, B, dM, F, G, gM, and K-type stars with R-band signal-to-noise ratio greater than 30 for classification, the classification accuracy is 0.91. This classification accuracy is also higher than that of the CNN+Bayes, CNN+Knn, CNN+SVM, CNN+AdaBoost, and CNN+RF algorithms, at 0.862, 0.876, 0.894, 0.868, and 0.889, respectively.

    Tools

    Get Citation

    Copy Citation Text

    YANG Jia-ming, TU Liang-ping, LI Jian-xi, MIAO Jia-wei. RN-SM:Stellar Spectral Classification Algorithm Based on ResNet Feature Extraction[J]. Spectroscopy and Spectral Analysis, 2025, 45(6): 1670

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Sep. 22, 2024

    Accepted: Jun. 27, 2025

    Published Online: Jun. 27, 2025

    The Author Email: TU Liang-ping (tuliangping@ustl.edu.cn)

    DOI:10.3964/j.issn.1000-0593(2025)06-1670-10

    Topics