Galaxy morphology classification based on Convolutional vision Transformer (CvT)

Jie Cao,Tingting Xu,Yuhe Deng,Linhua Deng,Mingcun Yang,Zhijing Liu,Weihong Zhou
DOI: https://doi.org/10.1051/0004-6361/202348544
2024-01-10
Abstract:The classification of galaxy morphology is among the most active fields in astronomical research today. With the development of artificial intelligence technology, deep learning is a useful tool in the classification of black the morphology of galaxies and significant progress has been made in this domain. However, there is still some room for improvement in terms of classification accuracy, automation, and related issues. Convolutional vision Transformer (CvT) is an improved version of the Vision Transformer (ViT) model. It improves the performance of the ViT model by introducing a convolutional neural network (CNN). This study explores the performance of the CvT model in the area of galaxy morphology classification. In this work, the CvT model was applied, for the first time, in a black five-class classification task of galaxy morphology. We black added different types and degrees of noise to the original galaxy images to verify that the CvT model achieves good classification performance, even in galaxy images with low signal-to-noise ratios (S/Ns). Then, we also validated the classification performance of the CvT model for galaxy images at different redshifts based on the low-redshift dataset GZ2 and the high-redshift dataset Galaxy Zoo CANDELS. In addition, we black visualized and analyzed the classification results of the CvT model based on the t-distributed stochastic black neighborhood -embedding (t-SNE) algorithm. We find that (1) compared with other black five-class classification models of galaxy morphology based on CNN models, the average accuracy, precision, recall, and F1\_score evaluation metrics of the CvT classification model are all higher than 98 , which is an improvement of at least 1 compared with those based on CNNs; (2) the classification black visualization results show that different categories of galaxies are separated from each other in multi-dimensional space. The application of the CvT model to the classification study of galaxy morphology is a novel undertaking that carries important implications for future studies.
astronomy & astrophysics
What problem does this paper attempt to address?