Abstract:Telescopes such as the Large Sky Area Multi-Object Spectroscopic Telescope and the Sloan Digital Sky Survey have produced an extensive collection of spectra, challenging the feasibility of manual classification in terms of accuracy and efficiency. To overcome these limitations, machine learning techniques are increasingly being utilized for automated spectral classification. However, these approaches primarily treat spectra as frequency domain signals, and lack robustness in low signal-to-noise ratio (S/N) scenarios and for small datasets of rare celestial objects. Moreover, they frequently neglect nuanced expert astronomical understanding. In this study, we draw inspiration from the human spectral discrimination process and propose a new model called the Image-EFficientNetV2-Spectrum Convolutional Neural Network (IEF-SCNN). IEF-SCNN combines spectral images using EfficientNetV2 with one-dimensional (1D) spectra through a 1DCNN. This integration effectively incorporates astronomical expertise into the classification process. Specifically, we plot the spectrum as an image and then classify it in a way that incorporates an attention mechanism. This attention mechanism mimics human observation of images for classification, selectively emphasizing relevant information while ignoring irrelevant details. Experimental data demonstrate that IEF-SCNN outperforms existing models in terms of the F1-score and accuracy metrics, particularly for low S/N (<6) data. Using progressive learning and an attention mechanism, the model trained on 12,000 M-class stars with an S/N below 6 achieved an accuracy of 87.38% on a 4000-sample test set. This surpasses traditional models (support vector machine with 83.15% accuracy, random forest with 65.40%, and artificial neural network with 84.40%) and the 1D stellar spectral CNN (85.65% accuracy). This research offers a foundation for the development of innovative methods for the automated identification of specific celestial objects, and can promote the creation of user-friendly software for astronomers who may not have computational expertise.

Automatic Classification Method for Astronomical Spectra Based on Wavelet Packet and SVM

Application Of Partial Least Squares Support Vector Machines (Pls-Svm) In Spectroscopy Quantitative Analysis

Stellar Data Classification Using SVM with Wavelet Transformation.

[Stellar Spectral Recognition Based on Wavelet De-Noising and SVM].

Classification of Stellar Spectral Data Using SVM

Spectral Classification and Particular Spectra Identification Based on Data Mining

RVM Supervised Feature Extraction and Seyfert Spectra Classification

Automated Spectral Classification Using Template Matching

Automatic Identification and Extraction of Clouds from Astronomical Images Based on Support Vector Machinetwo

Automatic Remote Sensing Image Classification Method Based On Spectral Angle And Spectral Distance

Classification of Astronomical Spectra Based on Multiscale Partial Convolution

Classification of quasars and stars by supervised and unsupervised methods

A Method for Auto-Recognizing the Stars Based on Spectral Feature

An Automated Galaxy Spectra Recognition Method Basing on Spectral Lines Information

Hyperspectral Image Classification Algorithm Based on Spectral-Spatial Hybrid Features and SVM

Spectral Classification of Galaxy Based on Wavelet Feature

Classification Technique for Hyperspectral Image Based on Subspace of Bands Feature Extraction and Ls-Svm

A New Automated Spectral Feature Extraction Method and Its Application in Spectral Classification and Defective Spectra Recovery

Data mining techniques on astronomical spectra data. II : Classification Analysis

Automatic Classification of All-Sky Nighttime Cloud Images Based on Machine Learning

Automatic Classification of Spectra with IEF-SCNN