# Bayesian and Convolutional Networks for Hierarchical Morphological Classification of Galaxies

arXiv: https://arxiv.org/abs/2405.02366 · HTML: https://mlsanity.com/paper/2405.02366/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 12 of its 49 references that are in the map.
- First posted on arXiv 3 May 2024.

## Builds on (most cited first, 12 of 12)

- [Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning](https://mlsanity.com/paper/1602.07261/) (2016)
- [EfficientNetV2: Smaller Models and Faster Training](https://mlsanity.com/paper/2104.00298/) (2021)
- [Big Transfer (BiT): General Visual Representation Learning](https://mlsanity.com/paper/1912.11370/) (2019)
- [A Survey of Convolutional Neural Networks: Analysis, Applications, and Prospects](https://mlsanity.com/paper/2004.02806/) (2020)
- [A Survey of the Recent Architectures of Deep Convolutional Neural Networks](https://mlsanity.com/paper/1901.06032/) (2019)
- [Bayesian Neural Networks: An Introduction and Survey](https://mlsanity.com/paper/2006.12024/) (2020)
- [Rotation-invariant convolutional neural networks for galaxy morphology prediction](https://mlsanity.com/paper/1503.07077/) (2015)
- [Galaxy Zoo: Probabilistic Morphology through Bayesian CNNs and Active Learning](https://mlsanity.com/paper/1905.07424/) (2019)
- [A catalog of broad morphology of Pan-STARRS galaxies based on deep learning](https://mlsanity.com/paper/2010.06073/) (2020)
- [What do Deep Networks Like to See?](https://mlsanity.com/paper/1803.08337/) (2018)
- [A Non-Technical Survey on Deep Convolutional Neural Network Architectures](https://mlsanity.com/paper/1803.02129/) (2018)
- [A Modular Deep Learning Pipeline for Galaxy-Scale Strong Gravitational Lens Detection and Modeling](https://mlsanity.com/paper/1911.03867/) (2019)

## Abstract (arXiv)

This work is focused on the morphological classification of galaxies following the Hubble sequence in which the different classes are arranged in a hierarchy. The proposed method, BCNN, is composed of two main modules. First, a convolutional neural network (CNN) is trained with images of the different classes of galaxies (image augmentation is carried out to balance some classes); the CNN outputs the probability for each class of the hierarchy, and its outputs/predictions feed the second module. The second module consists of a Bayesian network that represents the hierarchy and helps to improve the prediction accuracy by combining the predictions of the first phase while maintaining the hierarchical constraint (in a hierarchy, an instance associated with a node must be associated to all its ancestors), through probabilistic inference over the Bayesian network so that a consistent prediction is obtained. Different images from the Hubble telescope have been collected and labeled by experts, which are used to perform the experiments. The results show that BCNN performed better than several CNNs in multiple evaluation measures, reaching the next scores: 67% in exact match, 78% in accuracy, and 83% in hierarchical F-measure.
