Galaxy Classification Using Transfer Learning and Ensemble of CNNs With Multiple Colour Spaces
Not on the map. Why?
Abstract (arXiv)
Big data has become the norm in astronomy, making it an ideal domain for computer science research. Astronomers typically classify galaxies based on their morphologies, a practice that dates back to Hubble (1936). With small datasets, classification could be performed by individuals or small teams, but the exponential growth of data from modern telescopes necessitates automated classification methods. In December 2013, Winton Capital, Galaxy Zoo, and the Kaggle team created the Galaxy Challenge, which tasked participants with developing models to classify galaxies. The Kaggle Galaxy Zoo dataset has since been widely used by researchers. This study investigates the impact of colour space transformation on classification accuracy and explores the effect of CNN architecture on this relationship. Multiple colour spaces (RGB, XYZ, LAB, etc.) and CNN architectures (VGG, ResNet, DenseNet, Xception, etc.) are considered, utilizing pre-trained models and weights. However, as most pre-trained models are designed for natural RGB images, we examine their performance with transformed, non-natural astronomical images. We test our hypothesis by evaluating individual networks with RGB and transformed colour spaces and examining various ensemble configurations. A minimal hyperparameter search ensures optimal results. Our findings indicate that using transformed colour spaces in individual networks yields higher validation accuracy, and ensembles of networks and colour spaces further improve accuracy. This research aims to validate the utility of colour space transformation for astronomical image classification and serve as a benchmark for future studies.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 18 of its 82 references that are in the map. First posted on arXiv 26 Mar 2023.
Builds on
18 of 82 references.
References · 18
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2014Going Deeper with Convolutionscited by 4,241
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,332
- 2014Deep Learning in Neural Networks: An Overviewcited by 770
- 2014CNN Features off-the-shelf: an Astounding Baseline for Recognitioncited by 316
- 2021Ensemble deep learning: A reviewcited by 284
- 2015Rotation-invariant convolutional neural networks for galaxy morphology predictioncited by 83
- 2017The Relative Performance of Ensemble Methods with Deep Convolutional Neural Networks for Image Classificationcited by 50
- 2019ColorNet: Investigating the importance of color spaces for image classificationcited by 31
6 more
- 2018Using transfer learning to detect galaxy mergerscited by 11
- 2017Big Universe, Big Data: Machine Learning and Image Analysis for Astronomycited by 7
- 2015Vehicle Color Recognition using Convolutional Neural Networkcited by 4
- 2020DeepShadows: Separating Low Surface Brightness Galaxies from Artifacts using Deep Learningcited by 4
- 2018A CNN and LSTM-Based Approach to Classifying Transient Radio Frequency Interferencecited by 3
- 2020StegColNet: Steganalysis based on an ensemble colorspace approachcited by 3
Built on by
Citing papers per year: 2024: 1.
1 paper.