Exploiting Convolutional Neural Networks for Galaxy Classification
DOI:
https://doi.org/10.54097/5kzwfy50Keywords:
Convolutional neural network; galaxy classification; deep learning.Abstract
Galaxy classification is an important work in the field of astronomy. The efficiency of manually solving this problem is too low, and deep learning provides many methods for image classification. Using deep learning methods to automatically classify galaxies can greatly improve the efficiency of this task. Therefore, this paper uses mature Convolutional Neural Networks (CNN) models and self-designed simple CNN models, and compares their performance on the Galaxy10 DECals Dataset. In the experiment, the Inception network performed best, with an accuracy rate of 0.63 on the test set, but the training time was as long as about 2 hours. The simple CNN model ranked second, with an accuracy rate of 0.54, and the training time was only 20 minutes. The accuracy of other models is about 0.5, and the training time also takes 2 hours. The result performance of the model is worse than expected, which may be due to the less data used. The result of Inception is much ahead, and its solution to the problem provides a solution for the task of dealing with small data sets.
Downloads
References
Baqui, P. O., V. Marra, L. Casarini, R. Angulo, L. A. Diaz-Garcia, C. Hernández-Monteagudo, P. A. A. Lopes et al. The miniJPAS survey: star-galaxy classification using machine learning. Astronomy & Astrophysics, 2021, 645: A87.
Simonyan, Karen, and Andrew Zisserman. Very deep convolutional networks for large-scale image recognition, 2014: arXiv preprint arXiv: 1409. 1556.
He, Kaiming, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition, 2016: 770-778.
Szegedy, Christian, Wei Liu, Yangqing Jia, Pierre Sermanet, Scott Reed, Dragomir Anguelov, Dumitru Erhan, Vincent Vanhoucke, and Andrew Rabinovich. Going deeper with convolutions. In Proceedings of the IEEE conference on computer vision and pattern recognition, 2015, 1-9.
Huang, GAO, Zhuang Liu, Laurens Van Der Maaten, and Kilian Q. Weinberger. Densely connected convolutional networks. In Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, 4700-4708.
Howard, Andrew G., Menglong Zhu, Bo Chen, Dmitry Kalenichenko, Weijun Wang, Tobias Weyand, Marco Andreetto, and Hartwig Adam. Mobilenets: Efficient convolutional neural networks for mobile vision applications. 2017, arXiv preprint arXiv: 1704.04861.
Deng, Jia, Wei Dong, Richard Socher, Li-Jia Li, Kai Li, and Li Fei-Fei. Imagenet: A large-scale hierarchical image database. In 2009 IEEE conference on computer vision and pattern recognition, 2019, 248-255.
Galaxy10 DECals Dataset, URL: https://github.com/henrysky/Galaxy10, Last Accessed: 2023/11/16.
Gu, Jiuxiang, Zhenhua Wang, Jason Kuen, Lianyang Ma, Amir Shahroudy, Bing Shuai, Ting Liu et al. Recent advances in convolutional neural networks. Pattern recognition, 2018, 77: 354-377.
Yamashita, Rikiya, Mizuho Nishio, Richard Kinh Gian Do, and Kaori Togashi. Convolutional neural networks: an overview and application in radiology. Insights into imaging, 2018, 9: 611-629.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Highlights in Science, Engineering and Technology

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.







