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).

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

References in corpus (18)

• Heating Hot Atmospheres with Active Galactic Nuclei
• The LOFAR Two-metre Sky Survey -- V. Second data release
• Revisiting the Fanaroff-Riley dichotomy and radio-galaxy morphology with the LOFAR Two-Metre Sky Survey (LoTSS)
• Classifying Radio Galaxies with Convolutional Neural Network
• FR0CAT: a FIRST catalog of FR0 radio galaxies
• The DAWES review 10: The impact of deep learning for the analysis of galaxy surveys
• FRICAT: A FIRST catalog of FRI radio galaxies
• Morphological classification of radio galaxies: Capsule Networks versus Convolutional Neural Networks
• 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
• Deep Learning-based Imaging in Radio Interferometry
• 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

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