SFNet: Stellar Feature Network with CWT for Stellar Spectra Recognition

Stellar spectral classification is crucial in astronomical data analysis. However, existing studies are often limited by the uneven distribution of stellar samples, posing challenges in practical applications. Even when balancing stellar categories and their numbers, there is room for improvement in classification accuracy. This study introduces a Continuous Wavelet Transform using the Super Morlet wavelet to convert stellar spectra into wavelet images. A novel neural network, the Stellar Feature Network, is proposed for classifying these images. Stellar spectra from Large Sky Area Multi-Object Fiber Spectroscopic Telescope DR9, encompassing five equal categories (B, A, F, G, K), were used. Comparative experiments validate the effectiveness of the proposed methods and network, achieving significant improvements in classification accuracy.