Galaxy morphology classification with deep convolutional neural networks
Abstract
We propose a variant of residual networks (ResNets) for galaxy morphology classification. The variant, together with other popular convolutional neural networks (CNNs), is applied to a sample of 28790 galaxy images from the Galaxy Zoo 2 dataset, to classify galaxies into five classes, i.e., completely round smooth, in-between smooth (between completely round and cigar-shaped), cigar-shaped smooth, edge-on and spiral. Various metrics, such as accuracy, precision, recall, F1 value and AUC, show that the proposed network achieves state-of-the-art classification performance among other networks, namely, Dieleman, AlexNet, VGG, Inception and ResNets. The overall classification accuracy of our network on the testing set is 95.2083% and the accuracy of each type is given as follows: completely round, 96.6785%; in-between, 94.4238%; cigar-shaped, 58.6207%; edge-on, 94.3590% and spiral, 97.6953%. Our model algorithm can be applied to large-scale galaxy classification in forthcoming surveys, such as the Large Synoptic Survey Telescope (LSST) survey.
- Publication:
-
Astrophysics and Space Science
- Pub Date:
- April 2019
- DOI:
- 10.1007/s10509-019-3540-1
- arXiv:
- arXiv:1807.10406
- Bibcode:
- 2019Ap&SS.364...55Z
- Keywords:
-
- Galaxy morphology classification;
- Deep learning;
- Convolutional neural networks;
- Astrophysics - Astrophysics of Galaxies;
- Astrophysics - Instrumentation and Methods for Astrophysics
- E-Print:
- 12 pages, 13 figures