Stellar Spectral Classification Based on Capsule Network

The rapid development of large-scale sky survey project has produced a large amount of stellar spectral data, which make the automatic classification of stellar spectral data a challenging task. In this paper, we have proposed a stellar spectral classification method based on a capsule network. At first, by using the one-dimensional convolutional network and short-time Fourier transform (STFT), the one-dimensional spectra of the F5, G5, and K5 types selected from the LAMOST Data Release 5 (DR5) are converted into the two-dimensional Fourier spectrum images. Then, the two-dimensional Fourier spectrum images are classified automatically by the capsule network. Because the capsule network can preserve the hierarchical pose relationships among the entities in the image, and it does not need any pooling layers, the experimental results show that the capsule network has a better classification performance, for the classifications of the F5, G5, and K5-type stellar spectra, its classification accuracy is superior to other classification methods.