A Comparison of Deep Learning Architectures for Optical Galaxy Morphology Classification
arXiv:2111.04353 · doi:10.1109/ICECET52533.2021.9698414
Abstract
The classification of galaxy morphology plays a crucial role in understanding galaxy formation and evolution. Traditionally, this process is done manually. The emergence of deep learning techniques has given room for the automation of this process. As such, this paper offers a comparison of deep learning architectures to determine which is best suited for optical galaxy morphology classification. Adapting the model training method proposed by Walmsley et al in 2021, the Zoobot Python library is used to train models to predict Galaxy Zoo DECaLS decision tree responses, made by volunteers, using EfficientNet B0, DenseNet121 and ResNet50 as core model architectures. The predicted results are then used to generate accuracy metrics per decision tree question to determine architecture performance. DenseNet121 was found to produce the best results, in terms of accuracy, with a reasonable training time. In future, further testing with more deep learning architectures could prove beneficial.
References in corpus (14)
- Galaxy Zoo : Morphologies derived from visual inspection of galaxies from the Sloan Digital Sky Survey
- Rotation-invariant convolutional neural networks for galaxy morphology prediction
- Improving galaxy morphologies for SDSS with Deep Learning
- Galaxy Zoo DECaLS: Detailed Visual Morphology Measurements from Volunteers and Deep Learning for 314,000 Galaxies
- Galaxy Zoo: Probabilistic Morphology through Bayesian CNNs and Active Learning
- Radio Galaxy Zoo: host galaxies and radio morphologies derived from visual inspection
- Surveying the reach and maturity of machine learning and artificial intelligence in astronomy
- Deblending and Classifying Astronomical Sources with Mask R-CNN Deep Learning
- Radio Galaxy Zoo: Compact and extended radio source classification with deep learning
- A Machine Learning Based Morphological Classification of 14,245 Radio AGNs Selected From The Best-Heckman Sample
- Cataloging the radio-sky with unsupervised machine learning: a new approach for the SKA era
- CNN Architecture Comparison for Radio Galaxy Classification
- The FIRST Classifier: Compact and Extended Radio Galaxy Classification using Deep Convolutional Neural Networks
- Twelve Years of Galaxy Zoo