Morphological Classification of Radio Galaxies with wGAN-supported Augmentation

Abstract

Machine learning techniques that perform morphological classification of astronomical sources often suffer from a scarcity of labelled training data. Here, we focus on the case of supervised deep learning models for the morphological classification of radio galaxies, which is particularly topical for the forthcoming large radio surveys. We demonstrate the use of generative models, specifically Wasserstein GANs (wGANs), to generate data for different classes of radio galaxies. Further, we study the impact of augmenting the training data with images from our wGAN on three different classification architectures. We find that this technique makes it possible to improve models for the morphological classification of radio galaxies. A simple Fully Connected Neural Network (FCN) benefits most from including generated images into the training set, with a considerable improvement of its classification accuracy. In addition, we find it is more difficult to improve complex classifiers. The classification performance of a Convolutional Neural Network (CNN) can be improved slightly. However, this is not the case for a Vision Transformer (ViT).

arXiv comment15 pages, 7+6 figures, 1+5 tables; v2: matches published version in RASTI

References in corpus (42)

• Transformers in Vision: A Survey
• LOFAR: The LOw-Frequency ARray
• Galaxy Zoo : Morphologies derived from visual inspection of galaxies from the Sloan Digital Sky Survey
• Heating Hot Atmospheres with Active Galactic Nuclei
• The LOFAR Two-metre Sky Survey - II. First data release
• The LOFAR Two-metre Sky Survey -- V. Second data release
• A Catalog of Detailed Visual Morphological Classifications for 14034 Galaxies in the Sloan Digital Sky Survey
• Generative Adversarial Networks recover features in astrophysical images of galaxies beyond the deconvolution limit
• Revisiting the Fanaroff-Riley dichotomy and radio-galaxy morphology with the LOFAR Two-Metre Sky Survey (LoTSS)
• Finding Strong Gravitational Lenses in the Kilo Degree Survey with Convolutional Neural Networks
• Galaxy Zoo: Probabilistic Morphology through Bayesian CNNs and Active Learning
• Getting High: High Fidelity Simulation of High Granularity Calorimeters with High Speed
• Classifying Radio Galaxies with Convolutional Neural Network
• Optimising Automatic Morphological Classification of Galaxies with Machine Learning and Deep Learning using Dark Energy Survey Imaging
• FRII{\sl{CAT}}: A FIRST catalog of FR~II radio galaxies
• FR0CAT: a FIRST catalog of FR0 radio galaxies
• The DAWES review 10: The impact of deep learning for the analysis of galaxy surveys
• Radio Galaxy Zoo: ClaRAN - A Deep Learning Classifier for Radio Morphologies
• FRICAT: A FIRST catalog of FRI radio galaxies
• The nuclear properties and extended morphologies of powerful radio galaxies: the roles of host galaxy and environment
• GANplifying Event Samples
• Morphological Annotations for Groups in the FIRST Database
• Probing the Dark Ages with the Square Kilometer Array
• Morphological classification of radio galaxies: Capsule Networks versus Convolutional Neural Networks
• A Machine Learning Based Morphological Classification of 14,245 Radio AGNs Selected From The Best-Heckman Sample
• Square Kilometre Array Science Data Challenge 1: analysis and results
• The FIRST Classifier: Compact and Extended Radio Galaxy Classification using Deep Convolutional Neural Networks
• The CoNFIG Catalogue - II. Comparison of Space Densities in the FR Dichotomy
• Machine learning technique for morphological classification of galaxies from the SDSS. I. Photometry-based approach
• Unveiling the rarest morphologies of the LOFAR Two-metre Sky Survey radio source population with self-organised maps
• Fanaroff-Riley classification of radio galaxies using group-equivariant convolutional neural networks
• Attention-gating for improved radio galaxy classification
• Transfer learning for radio galaxy classification
• Deep Learning-based Imaging in Radio Interferometry
• Pulsar Candidate Identification Using Semi-Supervised Generative Adversarial Networks
• Morphological classification of compact and extended radio galaxies using convolutional neural networks and data augmentation techniques
• Deep Learning Assisted Data Inspection for Radio Astronomy
• The Combined NVSS-FIRST Galaxies (CoNFIG) Sample - I. Sample Definition, Classification and Evolution
• Life cycle of cosmic-ray electrons in the intracluster medium
• Rapid sorting of radio galaxy morphology using Haralick features
• Structured Variational Inference for Simulating Populations of Radio Galaxies
• Data-Efficient Classification of Radio Galaxies

Cited by in corpus (5)

• Deep learning-based radiointerferometric imaging with GAN-aided training
• Classification of Radio Sources Through Self-Supervised Learning
• Simulating images of radio galaxies with diffusion models
• Radio Halo Detection in MWA Data using Deep Neural Networks and Generative Data Augmentation
• Very-Long Baseline Interferometry Imaging with Closure Invariants using Conditional Image Diffusion