Enhanced Comparative Analysis of Pretrained and Custom Deep Convolutional Neural Networks for Galaxy Morphology Classification
Galaxy morphology classification is a crucial task in astronomy and astrophysics, providing information on galaxy formation and evolution. Traditionally, this classification has been a manual and labor-intensive process requiring significant astronomical expertise. However, advancements in artificial intelligence, particularly deep learning, offer more efficient and accurate solutions. We investigated the application of convolutional neural networks (CNNs) for galaxy morphology classification using the Galaxy10 DECals dataset. We developed and compared three models: a custom-built CNN using TensorFlow 2.18, a ResNet50 model initialized with random weights, and a pre-trained EfficientNetB5 model utilizing transfer learning, both implemented in PyTorch 2.6. Our results indicate that the custom model achieves an accuracy of 67%, while the ResNet50 and EfficientNetB5 models achieve 80 and 87% accuracy, respectively. The superior performance of the pre-trained Efficient-NetB5 model underscores the efficacy of transfer learning in astronomical image classification. These findings have significant implications for the application of deep learning techniques in astrophysical research.
Authors
- Deepshikha Bhati (ORCID: https://orcid.org/0009-0002-0115-6026)
- Norazlin Ibrahim (ORCID: https://orcid.org/0000-0001-6459-1290)
- Thu Nguyen (ORCID: https://orcid.org/0009-0007-6252-1249)
- Jungyoon Kim (ORCID: https://orcid.org/0000-0001-5013-446X)
- Tram Le
- Thanyaporn Noiplab
Institutions
- Kent State University (US)
Publication Details
- Published
- 2025-03-11
- DOI
- https://doi.org/10.3390/engproc2025089036
- Citations
- 1
- Primary Topic
- Geochemistry and Geologic Mapping
- Type
- article
- Field-Weighted Citation Impact
- 2.48